Update app.py

app.py

@@ -104,7 +104,7 @@ class BasicModel:
         # This serialization might need adjustment based on how CodeAgent sends prompts.
         # It needs to handle both initial strings and potential chat histories (as a list of ChatMessage objects).
         if isinstance(messages, str):
-            return messages
+            return f"<|start_header_id|>user<|end_header_id|>\n\n{messages}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
             
         prompt = []
         for m in messages: ## <- For each ChatMessage object
@@ -125,7 +125,6 @@ class BasicModel:
         return "".join(prompt)
         
     def generate(self, prompt: str | list, stop_sequences=None, **kwargs) -> str: ## <- 'prompt' is either a string or a list of ChatMessages
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
         # 1. Build the HF kwargs
         allowed = {"max_new_tokens", "temperature", "top_k", "top_p"}
         gen_kwargs = {k: v for k, v in kwargs.items() if k in allowed}