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)
	}
}