Update api.py

api.py

@@ -6,7 +6,8 @@ app = FastAPI()
 
 class PromptRequest(BaseModel):
     text: str
-    persona: str = "You are a helpful assistant."
+    persona: str = "You are a knowledgable and helpful assistant."
+    max_tokens: int = 500
 
 # --- NEW: Add a Root Route (The Health Check) ---
 @app.get("/")