feat: exploring tools

umbertogriffo · Jan 11, 2025 · 5948b17 · 5948b17
1 parent 36a8a9a
commit 5948b17
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/chatbot/bot/client/lama_cpp_client.py b/chatbot/bot/client/lama_cpp_client.py
@@ -168,7 +168,10 @@ def retrieve_tools(
         )
 
         tool_calls = output["choices"][0]["message"].get("tool_calls", None)
-
+        # first_choice_function = output["choices"][0]['message'].get('function_call', None)
+        # if first_choice_function:
+        #     function_name = first_choice_function.get('name', None)
+        #     function_args = first_choice_function.get('arguments', None)
         return tool_calls
 
     def stream_answer(self, prompt: str, max_new_tokens: int = 512) -> str: