blob: ef21f1ee0d88e7ff71cc061acec5834703d5574e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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)
}
}
|
