Просмотр исходного кода

feat: claude response return model name

CalciumIon 1 год назад
Родитель
Сommit
2650ec9b59
2 измененных файлов с 2 добавлено и 0 удалено
  1. 1 0
      dto/text_response.go
  2. 1 0
      relay/channel/claude/relay-claude.go

+ 1 - 0
dto/text_response.go

@@ -34,6 +34,7 @@ type OpenAITextResponseChoice struct {
 
 type OpenAITextResponse struct {
 	Id      string                     `json:"id"`
+	Model   string                     `json:"model"`
 	Object  string                     `json:"object"`
 	Created int64                      `json:"created"`
 	Choices []OpenAITextResponseChoice `json:"choices"`

+ 1 - 0
relay/channel/claude/relay-claude.go

@@ -341,6 +341,7 @@ func ResponseClaude2OpenAI(reqMode int, claudeResponse *ClaudeResponse) *dto.Ope
 	if len(tools) > 0 {
 		choice.Message.ToolCalls = tools
 	}
+	fullTextResponse.Model = claudeResponse.Model
 	choices = append(choices, choice)
 	fullTextResponse.Choices = choices
 	return &fullTextResponse