diff --git a/agent/assistant/hook/create_test.go b/agent/assistant/hook/create_test.go index 20a33957..3a11fcfd 100644 --- a/agent/assistant/hook/create_test.go +++ b/agent/assistant/hook/create_test.go @@ -1,8 +1,11 @@ package hook_test import ( + "context" "testing" + "github.com/yaoapp/yao/agent/assistant" + agentContext "github.com/yaoapp/yao/agent/context" "github.com/yaoapp/yao/agent/testutils" ) @@ -10,4 +13,22 @@ import ( func TestCreate(t *testing.T) { testutils.Prepare(t) defer testutils.Clean(t) + + agent, err := assistant.Get("tests.create") + if err != nil { + t.Fatalf("Failed to get the tests.create assistant: %s", err.Error()) + } + + if agent.Script == nil { + t.Fatalf("The tests.create assistant has no script") + } + + ctx := &agentContext.Context{ + Context: context.Background(), + ChatID: "chat-test-create-hook", + AssistantID: "tests.create", + Sid: "test-session-create-hook", + } + + agent.Script.Create(ctx, []agentContext.Message{{Role: "user", Content: "Hello, how are you?"}}) }