Update main.py

main.py

@@ -5,7 +5,7 @@ from duckai import DuckAI
 import uvicorn
 
 app = FastAPI()
-
+proxy= "socks4://98.178.72.21:10919"
 API_PREFIX = "/"
 
 # Middleware for logging request time
@@ -48,7 +48,9 @@ async def chat_completions(request: ChatCompletionRequest):
     try:
         # Only using DuckAI directly
         content = " ".join([msg.get("content", "") for msg in request.messages])
-        duck = DuckAI()
+
+
+        duck = DuckAI(proxy=proxy, timeout=20)
         results = duck.chat(content, model=request.model)
         response = create_complete_response(results, request.model)
         return response