Update

app.py

@@ -11,7 +11,7 @@ from dotenv import load_dotenv
 # SmolaGents imports
 from smolagents import CodeAgent, LiteLLMModel, tool
 from smolagents.agent_types import AgentText
-
+from smolagents.memory import ActionStep, TaskStep
 # Local imports
 from init_db import Project
 
@@ -102,11 +102,11 @@ def chat_with_sql(message, history):
     try:
       input_message = "You are a helpful assistant that presents information about real estate projects in database for investors. Use 'IS NOT NULL' when necessary. Translate any city names from input into Czech language. Present useful fields such as url. Do not make up information or hallucinate. If final result rows are empty respond that you could not find records matching criteria. Use only Czech city names. Note that there is limit 10 records for database result. Reformat final answer in markdown. User Input:\n\n {message}"
       prompt = input_message.format(message=message)
-      result = codeagent.run(prompt)
+      result = codeagent.run(prompt, reset=False)
       yield  gr.ChatMessage(role="assistant", content=result)
 
     except Exception as e:
-        yield f"Error: {str(e)}"
+        yield gr.ChatMessage(role="assistant", content=f"Error: {str(e)}")
 
 # Create the Gradio interface
 count_projects = Project.select().where(Project.structure.is_null(False)).count()