vital-agent-container-sdk 0.1.4__tar.gz → 0.1.5__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {vital_agent_container_sdk-0.1.4/vital_agent_container_sdk.egg-info → vital_agent_container_sdk-0.1.5}/PKG-INFO +1 -1
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/setup.py +1 -1
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/agent_container_app.py +2 -1
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/processor/aimp_message_processor.py +14 -3
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5/vital_agent_container_sdk.egg-info}/PKG-INFO +1 -1
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/LICENSE +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/README.md +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/setup.cfg +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/test/test_app.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/config/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/handler/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/handler/aimp_message_handler_inf.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/handler/impl/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/handler/impl/aimp_echo_message_handler.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/processor/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/streaming/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/streaming/custom_streaming_response.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/tasks/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/tasks/request_task.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/tasks/task_manager_async_client.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/utils/__init__.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/utils/aws_utils.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container/utils/config_utils.py +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container_sdk.egg-info/SOURCES.txt +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container_sdk.egg-info/dependency_links.txt +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container_sdk.egg-info/requires.txt +0 -0
- {vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.5}/vital_agent_container_sdk.egg-info/top_level.txt +0 -0
@@ -47,7 +47,8 @@ class AgentContainerApp(FastAPI):
|
|
47
47
|
|
48
48
|
async def process_ws_message(self, client: httpx.AsyncClient, websocket: WebSocket, data: str,
|
49
49
|
started_event: asyncio.Event):
|
50
|
-
|
50
|
+
|
51
|
+
logger.info(f"process_ws_message: Processing: {data}")
|
51
52
|
await self.message_processor.process_message(self.handler, self.config, client, websocket, data, started_event)
|
52
53
|
|
53
54
|
def add_routes(self):
|
@@ -1,4 +1,5 @@
|
|
1
1
|
import asyncio
|
2
|
+
import logging
|
2
3
|
|
3
4
|
from vital_agent_container.handler.aimp_message_handler_inf import AIMPMessageHandlerInf
|
4
5
|
|
@@ -9,10 +10,20 @@ class AIMPMessageProcessor:
|
|
9
10
|
pass
|
10
11
|
|
11
12
|
async def process_message(self, handler: AIMPMessageHandlerInf, app_config, client, websocket, data, started_event):
|
12
|
-
|
13
|
+
|
14
|
+
logger = logging.getLogger("AIMPMessageProcessor")
|
15
|
+
logger.setLevel(logging.INFO)
|
16
|
+
|
17
|
+
logger.info(f"Processing: {data}")
|
18
|
+
|
13
19
|
try:
|
14
20
|
return await handler.process_message(app_config, client, websocket, data, started_event)
|
15
|
-
except asyncio.CancelledError:
|
21
|
+
except asyncio.CancelledError as e:
|
16
22
|
# log canceling
|
17
|
-
|
23
|
+
logger.error(f"Canceling {e}")
|
24
|
+
raise e
|
25
|
+
except Exception as e:
|
26
|
+
logger.error(f"Exception {e}")
|
27
|
+
raise e
|
28
|
+
|
18
29
|
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|