Add unit test

sashabaranov · Sep 2, 2024 · 39fa8a2 · 39fa8a2
1 parent 3058f9e
commit 39fa8a2
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/chat_test.go b/chat_test.go
@@ -425,6 +425,28 @@ func TestMultipartChatMessageSerialization(t *testing.T) {
 	}
 }
 
+func TestToolCallChatMessageSerialization(t *testing.T) {
+	jsonText := `{"role":"assistant","tool_calls":[{"id":"123","type":"function","function":{"name":"my_func","arguments":"{}"}}]}`
+
+	toolCallMsg := openai.ChatCompletionMessage{
+		Role: openai.ChatMessageRoleAssistant,
+		ToolCalls: []openai.ToolCall{{
+			ID:       "123",
+			Type:     openai.ToolTypeFunction,
+			Function: openai.FunctionCall{Name: "my_func", Arguments: "{}"},
+		}},
+	}
+
+	s, err := json.Marshal(toolCallMsg)
+	if err != nil {
+		t.Fatalf("Expected no error: %s", err)
+	}
+	res := strings.ReplaceAll(string(s), " ", "")
+	if res != jsonText {
+		t.Fatalf("invalid message: %s", string(s))
+	}
+}
+
 // handleChatCompletionEndpoint Handles the ChatGPT completion endpoint by the test server.
 func handleChatCompletionEndpoint(w http.ResponseWriter, r *http.Request) {
 	var err error