diff --git a/api/__init__.py b/api/__init__.py
index dd79308..83313a2 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -7,14 +7,14 @@
     - IM, OM, IncomingMessage, OutgoingMessage: Pydantic модели контракта
 """
 
-from agent_api import AgentApi, AgentException
-from models import IM, OM, IncomingMessage, OutgoingMessage
+from .agent_api import AgentApi, AgentException
+from .models import IM, OM, ClientMessage, ServerMessage
 
 __all__ = [
     "AgentApi",
     "AgentException",
     "IM",
     "OM",
-    "IncomingMessage",
-    "OutgoingMessage",
+    "ClientMessage",
+    "ServerMessage",
 ]