vital-agent-container-sdk 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

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.
@@ -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
- print(f"process_ws_message: Processing: {data}")
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
- print(f"Processing: {data}")
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
- raise
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
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vital-agent-container-sdk
3
- Version: 0.1.4
3
+ Version: 0.1.5
4
4
  Summary: Vital Agent Container SDK
5
5
  Home-page: https://github.com/vital-ai/vital-agent-container-python
6
6
  Author: Marc Hadfield
@@ -1,12 +1,12 @@
1
1
  vital_agent_container/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- vital_agent_container/agent_container_app.py,sha256=5lbge8OC8tEkjJ_TsZ9-zkRPPKA_Jr8gwwIFxmo7z6U,5879
2
+ vital_agent_container/agent_container_app.py,sha256=u6nKDlAJiRHGsRa9Fe7avv04s7gCZeRTF6sdxbYfk-c,5886
3
3
  vital_agent_container/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  vital_agent_container/handler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  vital_agent_container/handler/aimp_message_handler_inf.py,sha256=H9Bey5fAAh13LscsmwlCkK6i-rMKe5iN_vwk279p57s,315
6
6
  vital_agent_container/handler/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
7
  vital_agent_container/handler/impl/aimp_echo_message_handler.py,sha256=QQl3MbAWONno408wXOXzlubhFRvp9Vjwk5iCiF12Sek,764
8
8
  vital_agent_container/processor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
9
- vital_agent_container/processor/aimp_message_processor.py,sha256=4aaMEKxpzpXuH-kjLyqGY8t5PkqFif7Dr5ZaFHKEr2o,532
9
+ vital_agent_container/processor/aimp_message_processor.py,sha256=1w2uzgWKZz9tr1Y05aJMKrOuTYnyVtrqTeQw7yi-lmA,798
10
10
  vital_agent_container/streaming/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  vital_agent_container/streaming/custom_streaming_response.py,sha256=KDuhl4Z5opb0pxCymkjFvyBPKvhf64TZatLY-qK0l_w,1059
12
12
  vital_agent_container/tasks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -15,8 +15,8 @@ vital_agent_container/tasks/task_manager_async_client.py,sha256=hOkhGYgtdrpOkZyg
15
15
  vital_agent_container/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
16
  vital_agent_container/utils/aws_utils.py,sha256=HuwdOs58VU9UDnytx2QlKjJNoPE9k2KIQbKJwgi-9zk,580
17
17
  vital_agent_container/utils/config_utils.py,sha256=0iqBXct1O0nMTTBkYSdqIp0WIuVM24fbSpw5uoZQi8k,426
18
- vital_agent_container_sdk-0.1.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
19
- vital_agent_container_sdk-0.1.4.dist-info/METADATA,sha256=p-MB6aRnKr2GKMGqGG2qVFWZ-aWv1_eS36fY8WZplPo,1187
20
- vital_agent_container_sdk-0.1.4.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
21
- vital_agent_container_sdk-0.1.4.dist-info/top_level.txt,sha256=acZrRgjFIj0mWxdLPWONKwMgxnFReQmyjttb8Z76fsc,22
22
- vital_agent_container_sdk-0.1.4.dist-info/RECORD,,
18
+ vital_agent_container_sdk-0.1.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
19
+ vital_agent_container_sdk-0.1.5.dist-info/METADATA,sha256=lVsRv1qBzoq2q65-KReawpHRCOmRw0uHzaAFZ4itjzs,1187
20
+ vital_agent_container_sdk-0.1.5.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
21
+ vital_agent_container_sdk-0.1.5.dist-info/top_level.txt,sha256=acZrRgjFIj0mWxdLPWONKwMgxnFReQmyjttb8Z76fsc,22
22
+ vital_agent_container_sdk-0.1.5.dist-info/RECORD,,