Update app.py

app.py

@@ -6,13 +6,11 @@ from openai import AsyncOpenAI
 app = FastAPI()
 
 async def generate_ai_response(prompt: str, model: str):
-    # Configuration for unofficial GitHub AI endpoint
     token = os.getenv("GITHUB_TOKEN")
     if not token:
         raise HTTPException(status_code=500, detail="GitHub token not configured")
     
     endpoint = "https://models.github.ai/inference"
-
     client = AsyncOpenAI(base_url=endpoint, api_key=token)
 
     try:
@@ -33,12 +31,15 @@ async def generate_ai_response(prompt: str, model: str):
 
     except Exception as err:
         yield f"Error: {str(err)}"
-        raise HTTPException(status_code=500, detail="AI generation failed")
+        raise HTTPException(status_code=500, detail=f"AI generation failed: {str(err)}")
 
 @app.post("/generate")
-async def generate_response(prompt: str = Query(...), model: str = Query("openai/gpt-4.1-mini")):
-    if not prompt:
-        raise HTTPException(status_code=400, detail="Prompt cannot be empty")
+async def generate_response(
+    prompt: str = Query(...),
+    model: str = Query(...)
+):
+    if not prompt or not model:
+        raise HTTPException(status_code=400, detail="Prompt and model must be provided")
     
     return StreamingResponse(
         generate_ai_response(prompt, model),