await for /search-assistant api

utils/ModelCallingFunctions.py

@@ -1,6 +1,7 @@
 import os
 from typing import List
 from PIL import Image
+from dotenv import load_dotenv
 import json
 import pickle
 
@@ -16,6 +17,7 @@ from .HelperFunctions import web_search_result_processor
 
 from .prompts import PROMPTS
 
+load_dotenv("utils/.env")
 
 TOGETHER_API = os.getenv("TOGETHER_API")
 GEMINI_API = os.getenv("GEMINI_API")
@@ -71,15 +73,23 @@ def industry_finder(collection_id):
 
 async def web_search(session, question):
     data = {"query": question, "model_id": "openai/gpt-4o-mini"}
-    async with session.post(
-        "https://general-chat.elevatics.cloud/search-assistant",
-        json=data,
-        headers={"X-API-KEY": X_API_KEY, "Content-Type": "application/json"},
-    ) as response:
-        print(f"Status: {response.status}")
-        print(f"Content: {response.content}")
-        content = await response.text()
-        return content
+    try:
+        async with session.post(
+            "https://general-chat.elevatics.cloud/search-assistant",
+            json=data,
+            headers={"X-API-KEY": X_API_KEY, "Content-Type": "application/json"},
+            timeout=aiohttp.ClientTimeout(total=60),  # Increase timeout to 60 seconds
+        ) as response:
+            print(f"Status: {response.status}")
+            if response.status == 200:
+                content = await response.text()
+                return content
+            else:
+                return f"Error: HTTP {response.status}"
+    except asyncio.TimeoutError:
+        return "Error: Request timed out"
+    except aiohttp.ClientError as e:
+        return f"Error: {str(e)}"
 
 
 async def other_info(company_data):
@@ -98,14 +108,9 @@ async def other_info(company_data):
     results = {}
     async with aiohttp.ClientSession() as session:
         tasks = [web_search(session, question) for question in questions.values()]
-        responses = await asyncio.gather(*tasks, return_exceptions=True)
-
+        responses = await asyncio.gather(*tasks)
     for type_, response in zip(questions, responses):
-        if isinstance(response, Exception):
-            results[type_] = {"error": str(response)}
-        else:
-            results[type_] = response
-
+        results[type_] = response
     return results