Final setup: Ollama API with gemma3:1b and updated generate method

app/main.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, Header, HTTPException
+from pydantic import BaseModel
 import os
 import requests
 
@@ -6,17 +7,20 @@ app = FastAPI()
 
 API_KEY = os.getenv("API_KEY", "default_key")  # change this!
 
+class PromptRequest(BaseModel):
+    prompt: str
+
 @app.get("/")
 def root():
     return {"status": "Ollama API is running"}
 
 @app.post("/generate")
-def generate(prompt: str, authorization: str = Header(None)):
+def generate(request: PromptRequest, authorization: str = Header(None)):
     if authorization != f"Bearer {API_KEY}":
         raise HTTPException(status_code=401, detail="Unauthorized")
 
     r = requests.post(
         "http://localhost:11434/api/generate",
-        json={"model": "gemma3:1b", "prompt": prompt},
+        json={"model": "gemma3:1b", "prompt": request.prompt},
     )
     return r.json()