Final

Dockerfile

@@ -32,7 +32,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 RUN chmod +x /usr/local/bin/entrypoint.sh
 
-RUN chmod -R 777 /app/.streamlit /app/.ollama
+RUN chmod -R 777 /app/.local /app/.ollama
 
 USER ollama
 

src/codeagent/src/codeagent/crews/SqlCrew/sqlcrew.py

@@ -16,16 +16,6 @@ llm = LLM(model="ollama/llama3.1:8b-instruct-q4_0", base_url="http://localhost:1
 def send_chat_message(output: str) -> None:
     socketio.emit("event", {'kind':'agent_message', 'output': output})
 
-def callback_function(output: TaskOutput):
-    # Do something after the task is completed
-    # Example: Send an email to the manager
-    socketio.emit("event", {'kind':'agent_message', 'output': output})
-    print(f"""
-        Task completed!
-        Task: {output.description}
-        Output: {output.raw}
-    """)
-
 @CrewBase
 class SqlCrew:
     agents: List[BaseAgent]
@@ -59,7 +49,6 @@ class SqlCrew:
             llm=llm,
             tools=[get_files_and_schema],
             allow_delegation=True,
-            step_callback = send_chat_message,
             verbose=True
         )
 
@@ -76,7 +65,6 @@ class SqlCrew:
             llm=llm,
             tools=[propose_code_changes, get_files_and_schema],
             allow_delegation=True,
-            step_callback = send_chat_message,
             verbose=True,
         )
 
@@ -93,7 +81,6 @@ class SqlCrew:
             llm=llm,
             tools=[get_recent_query_result, get_files_and_schema],
             allow_delegation=True,
-            step_callback = send_chat_message,
             verbose=True,
         )
 
@@ -111,7 +98,6 @@ class SqlCrew:
             llm=llm,
             tools=[execute_sql_file, get_files_and_schema],
             allow_delegation=True,
-            step_callback = send_chat_message,
             verbose=True,
         )
 
@@ -129,7 +115,6 @@ class SqlCrew:
             description="Propose schema (tables, columns, PKs, FKs) if needed meaning if user specifically said so to create database or to change database.",
             expected_output="Schema design in text format.",
             agent=self.schema_agent(),
-            callable = callback_function
         )
 
     @task
@@ -164,8 +149,6 @@ class SqlCrew:
             tasks=self.tasks,
             process=Process.hierarchical,
             manager_llm = llm,
-            verbose=True,
-            step_callback=send_chat_message,
-            task_callback=send_chat_message,
+            verbose=True
 
         )