1 files changed, 34 insertions, 0 deletions
diff --git a/libpanto-go/panto_test.go b/libpanto-go/panto_test.go
new file mode 100644
index 0000000..ef21f1e
--- /dev/null
+++ b/libpanto-go/panto_test.go
@@ -0,0 +1,34 @@
+package panto
+
+import "testing"
+
+func testConfig() Config {
+	return Config{
+		Provider: ProviderConfig{
+			OpenAIChat: &OpenAIChatConfig{
+				APIKey:    "test-key",
+				BaseURL:   "http://localhost",
+				Model:     "test-model",
+				MaxTokens: 16,
+			},
+		},
+	}
+}
+
+func TestNewAgentSetSystemPromptDoesNotCrash(t *testing.T) {
+	Init()
+	defer Deinit()
+
+	a, err := NewAgent(testConfig())
+	if err != nil {
+		t.Fatalf("NewAgent: %v", err)
+	}
+	defer a.Close()
+
+	if err := a.SetSystemPrompt("be brief"); err != nil {
+		t.Fatalf("SetSystemPrompt: %v", err)
+	}
+	if err := a.AddSystemMessage("also be accurate"); err != nil {
+		t.Fatalf("AddSystemMessage: %v", err)
+	}
+}