Merge pull request voxos-ai#369 from bolna-ai/docker-cleanup

adding logs for transfer_call
big-jobs · Aug 9, 2024 · 2976300 · 2976300
2 parents eba30b5 + 7379f70
commit 2976300
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bolna/agent_manager/task_manager.py b/bolna/agent_manager/task_manager.py
@@ -843,9 +843,12 @@ async def __execute_function_call(self, url, method, param, api_token, model_arg
 
             async with aiohttp.ClientSession() as session:
                 logger.info(f"Sending the payload to stop the conversation {payload} url {url}")
+                convert_to_request_log(str(payload), meta_info, None, "function_call", direction="request", is_cached=False,
+                                       run_id=self.run_id)
                 async with session.post(url, json = payload) as response:
                     response_text = await response.text()
                     logger.info(f"Response from the server after call transfer: {response_text}")
+                    convert_to_request_log(str(response_text), meta_info, None, "function_call", direction="response", is_cached=False, run_id=self.run_id)
                     return
 
         response = await trigger_api(url= url, method=method.lower(), param= param, api_token= api_token, meta_info = meta_info, run_id = self.run_id, **resp)