Update main.py

main.py

@@ -27,7 +27,7 @@ else:
     logger.info("LLM API Key loaded successfully.")
 
 # --- Constants & Headers ---
-SEARCH_API_URL = "https://search.privateinstance.com/api/text" # The new search provider
+SEARCH_API_URL = "https://search.privateinstance.com/api/text"
 LLM_API_URL = "https://api.typegpt.net/v1/chat/completions"
 LLM_MODEL = "gpt-4.1-mini"
 MAX_SOURCES_TO_PROCESS = 15
@@ -47,7 +47,7 @@ class DeepResearchRequest(BaseModel):
 app = FastAPI(
     title="AI Deep Research API",
     description="Provides robust, long-form, streaming deep research completions using the PrivateInstance Search API.",
-    version="8.0.0"  # Final Production Version with PrivateInstance API
+    version="9.0.0"  # Definitive Production Version
 )
 
 # Enable CORS for all origins
@@ -70,13 +70,16 @@ async def call_privateinstance_search(session: aiohttp.ClientSession, query: str
         async with session.get(SEARCH_API_URL, params=params, timeout=15) as response:
             response.raise_for_status()
             data = await response.json()
-            # The API might return results in a list directly or under a 'results' key.
             raw_results = data if isinstance(data, list) else data.get('results', [])
             
-            # Map the API's response keys to our internal format
+            # ***** THE DEFINITIVE FIX: Correctly map the API's response keys *****
             results = [
-                {'title': r.get('title'), 'link': r.get('url'), 'snippet': r.get('description')}
-                for r in raw_results if r.get('url') and r.get('title') and r.get('description')
+                {
+                    'title': r.get('title'),
+                    'link': r.get('href'),      # Use 'href' instead of 'url'
+                    'snippet': r.get('body')    # Use 'body' instead of 'description'
+                }
+                for r in raw_results if r.get('href') and r.get('title') and r.get('body')
             ]
             logger.info(f"Found {len(results)} sources from PrivateInstance for: '{query}'")
             return results