Update app.py

app.py

@@ -1,27 +1,25 @@
 import os
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
-from pydantic import BaseModel
 from openai import AsyncOpenAI
 
 app = FastAPI()
 
-class GenerateRequest(BaseModel):
-    prompt: str
-    model: str  # Model is required, no default
-
-async def generate_ai_response(prompt: str, model: str):
+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"
+    model = "openai/gpt-4.1-mini"  # Unofficial model name
+
     client = AsyncOpenAI(base_url=endpoint, api_key=token)
 
     try:
         stream = await client.chat.completions.create(
             messages=[
-                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "system", "content": "You are a helpful assistant named Orion and made by Abdullah Ali"},
                 {"role": "user", "content": prompt}
             ],
             model=model,
@@ -36,17 +34,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=f"AI generation failed: {str(err)}")
+        raise HTTPException(status_code=500, detail="AI generation failed")
 
 @app.post("/generate")
-async def generate_response(request: GenerateRequest):
-    if not request.prompt:
+async def generate_response(prompt: str , model: str):
+    if not prompt:
         raise HTTPException(status_code=400, detail="Prompt cannot be empty")
-    if not request.model:
-        raise HTTPException(status_code=400, detail="Model must be specified")
     
     return StreamingResponse(
-        generate_ai_response(request.prompt, request.model),
+        generate_ai_response(prompt , model),
         media_type="text/event-stream"
     )