Sfoglia il codice sorgente

feat: support logprobs

1808837298@qq.com 2 anni fa
parent
commit
6fa1837c24
1 ha cambiato i file con 2 aggiunte e 0 eliminazioni
  1. 2 0
      controller/relay.go

+ 2 - 0
controller/relay.go

@@ -140,6 +140,8 @@ type GeneralOpenAIRequest struct {
 	Tools            any             `json:"tools,omitempty"`
 	ToolChoice       any             `json:"tool_choice,omitempty"`
 	User             string          `json:"user,omitempty"`
+	LogProbs         bool            `json:"logprobs,omitempty"`
+	TopLogProbs      int             `json:"top_logprobs,omitempty"`
 }
 
 func (r GeneralOpenAIRequest) ParseInput() []string {