Use Prompt class as parameter type

app.py

@@ -19,14 +19,14 @@ async def test():
     return {"Test"}
 
 @app.post("/chat")
-async def chat(message):
+async def chat(prompt: Prompt):
     print("Received POST request")
-    print("Message:", message)
+    print("Message:", prompt.message)
     system_prompt = (
         "You are a beginner programming student helping a peer. "
         "Offer hints, ask questions, and support understanding—don’t give full solutions."
     )
-    full_prompt = f"<s>[INST] <<SYS>>{system_prompt}<</SYS>>\n{message} [/INST]"
+    full_prompt = f"<s>[INST] <<SYS>>{system_prompt}<</SYS>>\n{prompt.message} [/INST]"
 
     print("Full Prompt:", full_prompt)