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

vital_agent_container/agent_container_app.py

@@ -24,7 +24,9 @@ formatter = jsonlogger.JsonFormatter('%(asctime)s %(name)s %(levelname)s %(messa
 stream_handler = logging.StreamHandler(sys.stdout)
 stream_handler.setFormatter(formatter)
 
-log_file = "/var/log/agentcontainer/app.log"
+# log_file = "/var/log/agentcontainer/app.log"
+
+log_file = "app.log"
 
 file_handler = RotatingFileHandler(log_file, maxBytes=5 * 1024 * 1024, backupCount=5)
 file_handler.setFormatter(formatter)
@@ -47,7 +49,9 @@ class AgentContainerApp(FastAPI):
 
     async def process_ws_message(self, client: httpx.AsyncClient, websocket: WebSocket, data: str,
                                  started_event: asyncio.Event):
-        print(f"process_ws_message: Processing: {data}")
+
+        logger.info(f"process_ws_message: Processing: {data}")
+
         await self.message_processor.process_message(self.handler, self.config, client, websocket, data, started_event)
 
     def add_routes(self):

vital_agent_container/processor/aimp_message_processor.py

@@ -1,5 +1,5 @@
 import asyncio
-
+import logging
 from vital_agent_container.handler.aimp_message_handler_inf import AIMPMessageHandlerInf
 
 
@@ -9,10 +9,18 @@ class AIMPMessageProcessor:
         pass
 
     async def process_message(self, handler: AIMPMessageHandlerInf, app_config, client, websocket, data, started_event):
-        print(f"Processing: {data}")
+
+        logger = logging.getLogger("AIMPMessageProcessor")
+        logger.info(f"Processing: {data}")
+
         try:
-            return await handler.process_message(app_config, client, websocket, data, started_event)
-        except asyncio.CancelledError:
+            await handler.process_message(app_config, client, websocket, data, started_event)
+        except asyncio.CancelledError as e:
             # log canceling
-            raise
+            logger.error(f"Canceling {e}")
+            raise e
+        except Exception as e:
+            logger.error(f"Exception {e}")
+            raise e
+
 

{vital_agent_container_sdk-0.1.4.dist-info → vital_agent_container_sdk-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vital-agent-container-sdk
-Version: 0.1.4
+Version: 0.1.6
 Summary: Vital Agent Container SDK
 Home-page: https://github.com/vital-ai/vital-agent-container-python
 Author: Marc Hadfield

{vital_agent_container_sdk-0.1.4.dist-info → vital_agent_container_sdk-0.1.6.dist-info}/RECORD

@@ -1,12 +1,12 @@
 vital_agent_container/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vital_agent_container/agent_container_app.py,sha256=5lbge8OC8tEkjJ_TsZ9-zkRPPKA_Jr8gwwIFxmo7z6U,5879
+vital_agent_container/agent_container_app.py,sha256=bA6TKJKAgbP6V4fbWJA80QHg_mM3QHF7ahz1p67yKag,5911
 vital_agent_container/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vital_agent_container/handler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vital_agent_container/handler/aimp_message_handler_inf.py,sha256=H9Bey5fAAh13LscsmwlCkK6i-rMKe5iN_vwk279p57s,315
 vital_agent_container/handler/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vital_agent_container/handler/impl/aimp_echo_message_handler.py,sha256=QQl3MbAWONno408wXOXzlubhFRvp9Vjwk5iCiF12Sek,764
 vital_agent_container/processor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vital_agent_container/processor/aimp_message_processor.py,sha256=4aaMEKxpzpXuH-kjLyqGY8t5PkqFif7Dr5ZaFHKEr2o,532
+vital_agent_container/processor/aimp_message_processor.py,sha256=BNbAUrI0MOfWfaEqTFgfwTCyK6-ZISaC6uNz2Dfy6vU,751
 vital_agent_container/streaming/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vital_agent_container/streaming/custom_streaming_response.py,sha256=KDuhl4Z5opb0pxCymkjFvyBPKvhf64TZatLY-qK0l_w,1059
 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
 vital_agent_container/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vital_agent_container/utils/aws_utils.py,sha256=HuwdOs58VU9UDnytx2QlKjJNoPE9k2KIQbKJwgi-9zk,580
 vital_agent_container/utils/config_utils.py,sha256=0iqBXct1O0nMTTBkYSdqIp0WIuVM24fbSpw5uoZQi8k,426
-vital_agent_container_sdk-0.1.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vital_agent_container_sdk-0.1.4.dist-info/METADATA,sha256=p-MB6aRnKr2GKMGqGG2qVFWZ-aWv1_eS36fY8WZplPo,1187
-vital_agent_container_sdk-0.1.4.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-vital_agent_container_sdk-0.1.4.dist-info/top_level.txt,sha256=acZrRgjFIj0mWxdLPWONKwMgxnFReQmyjttb8Z76fsc,22
-vital_agent_container_sdk-0.1.4.dist-info/RECORD,,
+vital_agent_container_sdk-0.1.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vital_agent_container_sdk-0.1.6.dist-info/METADATA,sha256=hmaoy9joueOCsw_3VL4I2A1ComFVMpq3ukMKFpcQSMA,1187
+vital_agent_container_sdk-0.1.6.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+vital_agent_container_sdk-0.1.6.dist-info/top_level.txt,sha256=acZrRgjFIj0mWxdLPWONKwMgxnFReQmyjttb8Z76fsc,22
+vital_agent_container_sdk-0.1.6.dist-info/RECORD,,