Parcourir la source

add custom tool (#2157)

Seefs il y a 1 mois
Parent
commit
96a2b81aaa
1 fichiers modifiés avec 4 ajouts et 1 suppressions
  1. 4 1
      dto/openai_request.go

+ 4 - 1
dto/openai_request.go

@@ -232,10 +232,13 @@ func (r *GeneralOpenAIRequest) GetSystemRoleName() string {
 	return "system"
 }
 
+const CustomType = "custom"
+
 type ToolCallRequest struct {
 	ID       string          `json:"id,omitempty"`
 	Type     string          `json:"type"`
-	Function FunctionRequest `json:"function"`
+	Function FunctionRequest `json:"function,omitempty"`
+	Custom   json.RawMessage `json:"custom,omitempty"`
 }
 
 type FunctionRequest struct {