Update main.py

main.py

@@ -226,7 +226,7 @@ def fetch_news(query, num_results=20):
     else:
         return []
 
-
+@cache(expire=3600)
 def analyze_news(query):
     news_data = fetch_news(query)
     
@@ -245,7 +245,6 @@ def analyze_news(query):
     return messages
 
 @app.post("/news-assistant")
-@cache(expire=3600)
 async def news_assistant(query: NewsQueryModel, api_key: str = Depends(verify_api_key)):
     """
     News assistant endpoint that provides summaries and analysis of recent news based on user queries.
@@ -255,7 +254,8 @@ async def news_assistant(query: NewsQueryModel, api_key: str = Depends(verify_ap
     
     if not messages:
         raise HTTPException(status_code=500, detail="Failed to fetch news data")
-
+        
+    @cache(expire=3600)
     def process_response():
         for content in chat_with_llama_stream(messages, model="google/gemini-pro-1.5"):
             yield content