admin/cyclop
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fixed openai

Browse Source
This commit is contained in:
Jonathan Lucas
2024-08-19 13:41:12 -05:00
parent f920da8537
commit fcd558b4c6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
chatgpt_telegram_inference_bot.py
+3 -2
View File
@@ -16,7 +16,8 @@ class ChatGPTTelegramInferenceBot(BaseTelegramInferenceBot):
response = self.client.chat.completions.create( response = self.client.chat.completions.create(
model=self.model, model=self.model,
messages=[{"role": "system", "content": self.system_prompt}] + messages, messages=[{"role": "system", "content": self.system_prompt}] + messages,
tools=self.functions, functions=self.functions,
function_call="auto",
max_tokens=self.max_tokens max_tokens=self.max_tokens
) )
return response return response
@@ -32,7 +33,7 @@ class ChatGPTTelegramInferenceBot(BaseTelegramInferenceBot):
tool_calls = [] tool_calls = []
assistant_message = {} assistant_message = {}
for message_part in response.choices: for message_part in response.choices:
if message_part.finish_reason == "function_call": if message_part.finish_reason == "function_call":
tool_calls.append(message_part.message.function_call) tool_calls.append(message_part.message.function_call)