Update main.py

main.py

@@ -3,10 +3,13 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from datetime import date
 from ollama import Client
+import os
 
 app = FastAPI()
 today_date = date.today()
 
+JINA_PASS = os.environ("JINA_AUTH")
+
 class PromptRequest(BaseModel):
     prompt: str
     temperature: float = 0.5
@@ -30,7 +33,7 @@ def search_tool(query: str) -> str:
         r = requests.get(
             "https://s.jina.ai/",
             params={"q": query},
-            headers={"Accept": "application/json", "X-Retain-Images": "none"},
+            headers={"Accept": "application/json", "Authorization":{JINA_PASS}},
             timeout=10
         )
         data = r.json().get("data", [])[:5]