From c4f2225576493fdf6af7e85953aaf5d5f5cc2963 Mon Sep 17 00:00:00 2001 From: "C.C" Date: Fri, 22 Mar 2024 11:23:41 +0800 Subject: [PATCH] update test --- pkg/bridge/ai/provider/cfopenai/provider_test.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pkg/bridge/ai/provider/cfopenai/provider_test.go b/pkg/bridge/ai/provider/cfopenai/provider_test.go index 547b1b342..35ecb2a97 100644 --- a/pkg/bridge/ai/provider/cfopenai/provider_test.go +++ b/pkg/bridge/ai/provider/cfopenai/provider_test.go @@ -48,3 +48,12 @@ func TestNewProvider(t *testing.T) { }) }) } + +func TestChatCompletions(t *testing.T) { + provider := NewProvider("test_endpoint", "test_api_key", "test_model") + + resp, err := provider.GetChatCompletions("test_instruction", "test_base", ai.ChainMessage{}, metadata.M{}, false) + + assert.NotNil(t, resp) + assert.Errorf(t, err, "no_function_call") +}