Multiple tools calls in a raw.

solovieff · Aug 2, 2024 · eebc59b · eebc59b
1 parent 97a8380
commit eebc59b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/kibernikto/interactors/openai_executor.py b/kibernikto/interactors/openai_executor.py
@@ -259,6 +259,7 @@ async def process_tool_calls(self, choice: Choice, original_request_text: str, s
         if not choice.message.tool_calls:
             raise ValueError("No tools provided!")
         message_dict = None
+        tool_call_messages = []
         for tool_call in choice.message.tool_calls:
             fn_name = tool_call.function.name
             function_impl = self._get_tool_implementation(fn_name)
@@ -269,8 +270,8 @@ async def process_tool_calls(self, choice: Choice, original_request_text: str, s
                                                                          additional_params=additional_params)
             message_dict = dict(content=f"{original_request_text}", role=OpenAIRoles.user.value)
             prompt.append(message_dict)
-            tool_call_messages = ai_tools.get_tool_call_serving_messages(tool_call, tool_call_result,
-                                                                         xml=self.xml_tools)
+            tool_call_messages += ai_tools.get_tool_call_serving_messages(tool_call, tool_call_result,
+                                                                          xml=self.xml_tools)
 
         choice, usage = await self._run_for_messages(full_prompt=prompt + tool_call_messages)
         response_message: ChatCompletionMessage = choice.message

diff --git a/kibernikto/telegram/telegram_bot.py b/kibernikto/telegram/telegram_bot.py
@@ -19,7 +19,6 @@ def __init__(self, aiogram_chat: Chat, aiogram_user: User = None):
         self.aiogram_user = aiogram_user
 
 
-
 class TelegramBot(OpenAIExecutor):
     def __init__(self, config: OpenAiExecutorConfig, master_id, username, key=NOT_GIVEN,
                  chat_info: KiberniktoChatInfo = None):