Update main.py

main.py

@@ -26,10 +26,10 @@ def chat_with_model(query: str, model: str) -> JSONResponse:
 async def chat(query: str) -> JSONResponse:
     results = None
     try:
-        return chat_with_model(query, model='gpt-4o-mini')
+        return  DDGS().chat(query, model='gpt-4o-mini')
     except Exception as e:
         try:
-            return chat_with_model(query, model='claude-3-haiku')
+            return  DDGS().chat(query, model='claude-3-haiku')
         except Exception as e:
             return JSONResponse(content={"error": str(e)})