Merge branch 'main' into staging/deploy

cohere-ai · Aug 20, 2024 · 3be4afd · 3be4afd
2 parents 3263183 + a16fa2e
commit 3be4afd
Show file tree

Hide file tree

Showing 9 changed files with 1,351 additions and 96 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/backend/services/file.py b/src/backend/services/file.py
@@ -487,6 +487,7 @@ async def consolidate_agent_files_in_compass(
                     "index": agent_id,
                     "file_id": file_id,
                     "file_bytes": fetched_doc["text"],
+                    "filename": fetched_doc["file_name"],
                     "custom_context": {
                         "file_id": file_id,
                         "file_name": fetched_doc["file_name"],
@@ -569,6 +570,7 @@ async def insert_files_in_compass(
                     "index": new_file_id if index is None else index,
                     "file_id": new_file_id,
                     "file_bytes": file_bytes,
+                    "filename": filename,
                     "custom_context": {
                         "file_id": new_file_id,
                         "file_name": filename,

diff --git a/src/community/tools/clinicaltrials.py b/src/community/tools/clinicaltrials.py
@@ -21,11 +21,8 @@ def __init__(self, url="https://clinicaltrials.gov/api/v2/studies"):
     def is_available(cls) -> bool:
         return True
 
-    def call(
-        self,
-        parameters: Dict[str, Any],
-        ctx: Any,
-        n_max_studies: int = 10,
+    async def call(
+        self, parameters: Dict[str, Any], ctx: Any, n_max_studies: int = 10, **kwargs
     ) -> List[Dict[str, Any]]:
         query_params = {"sort": "LastUpdatePostDate"}
         if condition := parameters.get("condition", ""):