Final

Dockerfile

@@ -6,7 +6,7 @@ ENV HOME=/app
 
 # Add ollama user and ensure home + ollama directories
 RUN groupadd -r ollama && useradd -r -g ollama ollama \
-    && mkdir -p /app/.ollama /app/.streamlit \
+    && mkdir -p /app/.ollama /app/.local \
     && chown -R ollama:ollama /app
 
 # Install packages

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

@@ -1,5 +1,5 @@
 from typing import List
-from crewai import Agent, Crew, Process, Task, LLM
+from crewai import Agent, Crew, Process, Task, LLM, TaskOutput
 from crewai.project import CrewBase, agent, crew, task
 from crewai.agents.agent_builder.base_agent import BaseAgent
 import sys
@@ -16,6 +16,16 @@ 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]
@@ -119,6 +129,7 @@ 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