From 00d6cda9ed8c1fd41ab13258ef273fb6ea07e7ff Mon Sep 17 00:00:00 2001
From: wozulong <>
Date: Thu, 16 May 2024 18:54:21 +0800
Subject: [PATCH] refine completions req type

Signed-off-by: wozulong <>
---
 dto/text_request.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/dto/text_request.go b/dto/text_request.go
index 0f696fc..a73297c 100644
--- a/dto/text_request.go
+++ b/dto/text_request.go
@@ -10,7 +10,11 @@ type GeneralOpenAIRequest struct {
 	Model            string          `json:"model,omitempty"`
 	Messages         []Message       `json:"messages,omitempty"`
 	Prompt           any             `json:"prompt,omitempty"`
+	BestOf           int             `json:"best_of"`
+	Echo             bool            `json:"echo"`
 	Stream           bool            `json:"stream,omitempty"`
+	StreamOptions    any             `json:"stream_options"`
+	Suffix           string          `json:"suffix"`
 	MaxTokens        uint            `json:"max_tokens,omitempty"`
 	Temperature      float64         `json:"temperature,omitempty"`
 	TopP             float64         `json:"top_p,omitempty"`
@@ -28,6 +32,7 @@ type GeneralOpenAIRequest struct {
 	Tools            any             `json:"tools,omitempty"`
 	ToolChoice       any             `json:"tool_choice,omitempty"`
 	User             string          `json:"user,omitempty"`
+	LogitBias        any             `json:"logit_bias"`
 	LogProbs         bool            `json:"logprobs,omitempty"`
 	TopLogProbs      int             `json:"top_logprobs,omitempty"`
 }