fix

src/backend/chat/custom/custom.py

@@ -259,14 +259,14 @@ def call_tools(self, chat_history, deployment_model, **kwargs: Any):
             if not tool:
                 continue
 
-            outputs = asyncio.run(tool.implementation().call(
+            outputs = tool.implementation().call(
                 parameters=tool_call.get("parameters"),
                 session=kwargs.get("session"),
                 model_deployment=deployment_model,
                 user_id=kwargs.get("user_id"),
                 trace_id=kwargs.get("trace_id"),
                 agent_id=kwargs.get("agent_id"),
-            ))
+            )
 
             # If the tool returns a list of outputs, append each output to the tool_results list
             # Otherwise, append the single output to the tool_results list

src/backend/model_deployments/cohere_platform.py

@@ -43,7 +43,7 @@ def __init__(self, **kwargs: Any):
 
     @property
     def rerank_enabled(self) -> bool:
-        return False
+        return True
 
     @classmethod
     def list_models(cls) -> List[str]:

src/backend/tools/google_drive/tool.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 import time
 from typing import Any, Dict, List
@@ -61,6 +62,9 @@ def _handle_tool_specific_errors(cls, error: Exception, **kwargs: Any):
         raise Exception(message)
 
     async def call(self, parameters: dict, **kwargs: Any) -> List[Dict[str, Any]]:
+        return asyncio.run(self.call_async(parameters, **kwargs))
+
+    async def call_async(self, parameters: dict, **kwargs: Any) -> List[Dict[str, Any]]:
         """
         Google Drive logic
         """