Merge branch 'refs/heads/main' into eugene/tlk-2361-north-modify-prea…

…mble-or-instructions-to-generate-valid
cohere-ai · Dec 13, 2024 · 51412a5 · 51412a5
2 parents 6974443 + 839f258
commit 51412a5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/backend/tools/hybrid_search.py b/src/backend/tools/hybrid_search.py
@@ -141,7 +141,8 @@ async def rerank_results(
                 documents=[
                     f"{result.get('title', '')} {result.get('text')}"
                     for result in results_batch
-                    if "text" in result
+                    if isinstance(result, dict)
+                    and "text" in result
                 ],
                 ctx=ctx,
             )

diff --git a/src/backend/tools/web_scrape.py b/src/backend/tools/web_scrape.py
@@ -64,15 +64,15 @@ async def call(
                     return await self.handle_response(response, url)
 
             except aiohttp.ClientError as e:
-                return  {
+                return  [{
                     "text": f"Client error using web scrape: {str(e)}",
                     "url": url,
-                }
+                }]
             except Exception as e:
-                return {
+                return [{
                     "text": f"Request failed using web scrape: {str(e)}",
                     "url": url,
-                }
+                }]
 
     async def handle_response(self, response: aiohttp.ClientResponse, url: str):
         content_type = response.headers.get("content-type")