Update main.py

main.py

@@ -1,24 +1,27 @@
 from fastapi import FastAPI, HTTPException, Request, Query
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi_cache import FastAPICache
+from fastapi_cache.backends.inmemory import InMemoryBackend
+from fastapi_cache.decorator import cache
+
+app = FastAPI()
+
+@app.on_event("startup")
+async def startup():
+    FastAPICache.init(InMemoryBackend(), prefix="fastapi-cache")
+
 from pydantic import BaseModel
 from typing import List, Dict, Any
 from helper_functions_api import md_to_html, search_brave, fetch_and_extract_content, limit_tokens, together_response, insert_data
 import os
 from dotenv import load_dotenv, find_dotenv
 from datetime import datetime, timedelta
-from fastapi_cache import FastAPICache
-from fastapi_cache.backends.inmemory import InMemoryBackend
-from fastapi_cache.decorator import cache
 import asyncio
 import re
 # Load environment variables from .env file
 #load_dotenv("keys.env")
 
-app = FastAPI()
 
-@app.on_event("startup")
-async def startup():
-    FastAPICache.init(InMemoryBackend(), prefix="fastapi-cache")
     
 TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
 BRAVE_API_KEY = os.getenv('BRAVE_API_KEY')
@@ -83,7 +86,7 @@ async def generate_report(query: QueryModel):
     if query.internet:
         search_query = re.sub(r'[^\w\s]', '', description).strip()
         try:
-            urls, optimized_search_query = search_brave(search_query, num_results=8)
+            urls, optimized_search_query = await search_brave(search_query, num_results=8)
             all_text_with_urls = fetch_and_extract_content(data_format, urls, description)
             additional_context = limit_tokens(str(all_text_with_urls))
             prompt = f"Write a factual answer to the query:: {description} #### , refer the provided internet search results reference:{additional_context}"