Explorar el Código

feat: support logprobs

1808837298@qq.com hace 2 años
padre
commit
6fa1837c24
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  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 {