chore: refactor chat

Author: henomis

examples/cmd/chat/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"os"
 
@@ -19,24 +20,32 @@ func main() {
 	err := client.Chat(
 		context.Background(),
 		&request.Chat{
-			ChatHistory: []model.ChatMessage{
+			Message: "Calculate 12345 + 67890",
+			Tools: []model.Tool{
 				{
-					Role:    model.ChatMessageRoleUser,
-					Message: "Hello, how are you?",
-				},
-				{
-					Role:    model.ChatMessageRoleChatbot,
-					Message: "I'm good, how are you?",
+					Name:        "sum",
+					Description: "Use this tool to sum two numbers",
+					ParameterDefinitions: map[string]model.ToolParameterDefinition{
+						"num1": {
+							Description: "The first number to sum",
+							Type:        "int",
+						},
+						"num2": {
+							Description: "The second number to sum",
+							Type:        "int",
+						},
+					},
 				},
 			},
-			Message: "I'm very happy!",
 		},
 		resp,
 	)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Printf("resp: %#v\n", resp)
+	b, _ := json.MarshalIndent(resp, "", "  ")
+
+	fmt.Printf("%s\n", string(b))
 
 }

model/models.go

@@ -213,9 +213,8 @@ const (
 )
 
 type StreamedChat struct {
-	EventType    EventType       `json:"event_type"`
-	FinishReason FinishReason    `json:"finish_reason"`
-	Response     NonStreamedChat `json:"response"`
+	EventType EventType       `json:"event_type"`
+	Response  NonStreamedChat `json:"response"`
 }
 
 type FinishReason string
@@ -240,12 +239,17 @@ const (
 )
 
 type NonStreamedChat struct {
-	Text          string         `json:"text"`
-	GenerationID  string         `json:"generation_id"`
-	Citations     []Citation     `json:"citations"`
-	Documents     []Document     `json:"documents"`
-	SearchQueries []SearchQuery  `json:"search_queries"`
-	SearchResults []SearchResult `json:"search_results"`
+	Text             string         `json:"text"`
+	GenerationID     string         `json:"generation_id"`
+	Citations        []Citation     `json:"citations"`
+	Documents        []Document     `json:"documents"`
+	IsSearchRequired bool           `json:"is_search_required"`
+	SearchQueries    []SearchQuery  `json:"search_queries"`
+	SearchResults    []SearchResult `json:"search_results"`
+	FinishReason     FinishReason   `json:"finish_reason"`
+	ToolCalls        []ToolCall     `json:"tool_calls"`
+	ChatHistory      []ChatMessage  `json:"chat_history"`
+	ForceSingleStep  bool           `json:"force_single_step,omitempty"`
 }
 
 type SearchResult struct {