diff --git a/.gitignore b/.gitignore
index 274911b..974ad1b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,5 @@
 .idea/
+workspace/
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
diff --git a/docker-compose.yml b/docker-compose.yml
index 32ec3ae..95960f4 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -13,6 +13,8 @@ services:
     build:
       context: .
       target: development
+      tags:
+        - mrkan0/lambda-agent:dev
       additional_contexts:
         agent_api: ${AGENT_API_PATH}
     volumes:
diff --git a/src/api/external.py b/src/api/external.py
index 219d5f7..c93f2cb 100644
--- a/src/api/external.py
+++ b/src/api/external.py
@@ -40,5 +40,5 @@ async def process_message(ws: WebSocket, msg, agent_service: AgentService):
     match msg:
         case MsgUserMessage():
             async for chunk in agent_service.astream(msg.text):
-                await ws.send_text(MsgEventTextChunk(text=chunk).model_dump_json())
+                await ws.send_text(chunk.model_dump_json())
             await ws.send_text(MsgEventEnd(tokens_used=0).model_dump_json())