diff --git a/src/llama_stack_client/lib/agents/event_logger.py b/src/llama_stack_client/lib/agents/event_logger.py
index d894f59..39e4cce 100644
--- a/src/llama_stack_client/lib/agents/event_logger.py
+++ b/src/llama_stack_client/lib/agents/event_logger.py
@@ -6,8 +6,9 @@
 
 from typing import List, Optional, Union
 
-from llama_stack.types import ToolResponseMessage
-from llama_stack.types.agents import AgentsTurnStreamChunk
+from llama_stack_client.types import ToolResponseMessage
+
+from llama_stack_client.types.agents import AgentsTurnStreamChunk
 from termcolor import cprint
 
 
diff --git a/src/llama_stack_client/lib/inference/event_logger.py b/src/llama_stack_client/lib/inference/event_logger.py
index 8430dfc..3faa92d 100644
--- a/src/llama_stack_client/lib/inference/event_logger.py
+++ b/src/llama_stack_client/lib/inference/event_logger.py
@@ -6,7 +6,10 @@
 
 from typing import List, Optional, Union
 
-from llama_stack.types import ChatCompletionStreamChunk, InferenceChatCompletionResponse
+from llama_stack_client.types import (
+    ChatCompletionStreamChunk,
+    InferenceChatCompletionResponse,
+)
 from termcolor import cprint