update: xai增加max_tokens,max_words等参数

This commit is contained in:
taylor
2024-11-12 10:24:55 +08:00
parent 331709411a
commit 978bb9051d

View File

@@ -120,9 +120,13 @@ class Anthropic(Platform):
class XAi(Platform):
max_tokens: int
temperature: int
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http)
self.temperature = self.config['temperature']
self.max_tokens = self.config['max_tokens']
async def create_chat_completion(self, plugin: AbsExtraConfigPlugin, evt: MessageEvent) -> ChatCompletion:
full_context = []
@@ -139,6 +143,9 @@ class XAi(Platform):
"stream": False
}
if 'max_tokens' in self.config and self.max_tokens:
data["max_tokens"] = self.max_tokens
if 'temperature' in self.config and self.temperature:
request_body["temperature"] = self.temperature