Skip to content

Commit 4f7673d

Browse files
committed
chore: update models
1 parent c0fb0d6 commit 4f7673d

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

model/models.go

+7-6
Original file line numberDiff line numberDiff line change
@@ -220,11 +220,13 @@ type StreamedChat struct {
220220
type FinishReason string
221221

222222
const (
223-
FinishReasonComplete FinishReason = "COMPLETE"
224-
FinishReasonErrorLimit FinishReason = "ERROR_LIMIT"
225-
FinishReasonMaxTokens FinishReason = "MAX_TOKENS"
226-
FinishReasonError FinishReason = "ERROR"
227-
FinishReasonErrorToxic FinishReason = "ERROR_TOXIC"
223+
FinishReasonComplete FinishReason = "COMPLETE"
224+
FinishReasonStopSequence FinishReason = "STOP_SEQUENCE"
225+
FinishReasonError FinishReason = "ERROR"
226+
FinishReasonErrorToxic FinishReason = "ERROR_TOXIC"
227+
FinishReasonErrorLimit FinishReason = "ERROR_LIMIT"
228+
FinishReasonUserCancel FinishReason = "USER_CANCEL"
229+
FinishReasonMaxTokens FinishReason = "MAX_TOKENS"
228230
)
229231

230232
type EventType string
@@ -249,7 +251,6 @@ type NonStreamedChat struct {
249251
FinishReason FinishReason `json:"finish_reason"`
250252
ToolCalls []ToolCall `json:"tool_calls"`
251253
ChatHistory []ChatMessage `json:"chat_history"`
252-
ForceSingleStep bool `json:"force_single_step,omitempty"`
253254
}
254255

255256
type SearchResult struct {

request/chat.go

+1
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ type Chat struct {
3131
PresencePenalty *float64 `json:"presence_penalty,omitempty"`
3232
Tools []model.Tool `json:"tools,omitempty"`
3333
ToolResults []model.ToolResult `json:"tool_results,omitempty"`
34+
ForceSingleStep bool `json:"force_single_step,omitempty"`
3435
}
3536

3637
func (c *Chat) Path() (string, error) {

0 commit comments

Comments
 (0)