Update app.py

app.py

@@ -10,6 +10,7 @@ from typing import List, Dict, Any, Annotated
 from langgraph.graph import Graph, StateGraph
 from typing_extensions import TypedDict
 from openai import OpenAI
+from urllib.parse import quote_plus
 
 # -------------------------
 # Utility helpers
@@ -42,32 +43,37 @@ def jina_search_tool(query: str, api_key: str, max_results: int = 5) -> List[str
     print(f"Query: {query}")
     print(f"Max Results: {max_results}")
     
-    endpoint = f"https://s.jina.ai/{query.replace(' ', '+')}"
-    print(f"Endpoint: {endpoint}")
+    url = f"https://s.jina.ai/?q={quote_plus(query)}"
+    print(f"URL: {url}")
     
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Accept": "application/json",
-        "User-Agent": "Mozilla/5.0",
+        "User-Agent": "Mozilla/5.0"
     }
     print("Headers:", {k: v[:10] + "..." if k == "Authorization" else v for k, v in headers.items()})
     
     try:
         print("\nSending request to Jina API...")
-        resp = requests.get(endpoint, headers=headers, timeout=15)
-        print(f"Response Status: {resp.status_code}")
+        r = requests.get(url, headers=headers, timeout=15)
+        print(f"Response Status: {r.status_code}")
         
-        if resp.status_code != 200:
-            print(f"Error Response: {resp.text[:500]}")
-            raise RuntimeError(f"Jina search failed with status {resp.status_code}: {resp.text[:200]}")
+        r.raise_for_status()
+        data = r.json()
         
-        data = resp.json()
         print("\nResponse Data Structure:")
         print(f"Keys in response: {list(data.keys())}")
         print(f"Number of results: {len(data.get('results', []))}")
         
-        results = [item.get("content", "") for item in data.get("results", [])][:max_results]
+        results = [
+            f"{item['title']} – {item['url']}\n{item['content']}"
+            for item in data.get("results", [])[:max_results]
+        ]
+        
         print(f"\nRetrieved {len(results)} results")
+        if results:
+            print("\nFirst result preview:")
+            print(results[0][:200] + "..." if len(results[0]) > 200 else results[0])
         
         if not results:
             print("WARNING: No results found in the response")
@@ -83,7 +89,7 @@ def jina_search_tool(query: str, api_key: str, max_results: int = 5) -> List[str
         raise
     except json.JSONDecodeError as e:
         print(f"ERROR: Failed to parse JSON response: {str(e)}")
-        print(f"Raw response: {resp.text[:500]}")
+        print(f"Raw response: {r.text[:500]}")
         raise
     except Exception as e:
         print(f"ERROR: Unexpected error: {str(e)}")