update test

lfedgeai · Mar 22, 2024 · c4f2225 · c4f2225
1 parent 5d0f1c3
commit c4f2225
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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")
+}