vital-agent-container-sdk 0.1.4__tar.gz → 0.1.6__tar.gz

{vital_agent_container_sdk-0.1.4/vital_agent_container_sdk.egg-info → vital_agent_container_sdk-0.1.6}/PKG-INFO

@@ -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 → vital_agent_container_sdk-0.1.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='vital-agent-container-sdk',
-    version='0.1.4',
+    version='0.1.6',
     author='Marc Hadfield',
     author_email='marc@vital.ai',
     description='Vital Agent Container SDK',

{vital_agent_container_sdk-0.1.4 → vital_agent_container_sdk-0.1.6}/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_sdk-0.1.6/vital_agent_container/processor/aimp_message_processor.py

@@ -0,0 +1,26 @@
+import asyncio
+import logging
+from vital_agent_container.handler.aimp_message_handler_inf import AIMPMessageHandlerInf
+
+
+class AIMPMessageProcessor:
+
+    def __init__(self):
+        pass
+
+    async def process_message(self, handler: AIMPMessageHandlerInf, app_config, client, websocket, data, started_event):
+
+        logger = logging.getLogger("AIMPMessageProcessor")
+        logger.info(f"Processing: {data}")
+
+        try:
+            await handler.process_message(app_config, client, websocket, data, started_event)
+        except asyncio.CancelledError as e:
+            # log canceling
+            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 → vital_agent_container_sdk-0.1.6/vital_agent_container_sdk.egg-info}/PKG-INFO

@@ -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/vital_agent_container/processor/aimp_message_processor.py

@@ -1,18 +0,0 @@
-import asyncio
-
-from vital_agent_container.handler.aimp_message_handler_inf import AIMPMessageHandlerInf
-
-
-class AIMPMessageProcessor:
-
-    def __init__(self):
-        pass
-
-    async def process_message(self, handler: AIMPMessageHandlerInf, app_config, client, websocket, data, started_event):
-        print(f"Processing: {data}")
-        try:
-            return await handler.process_message(app_config, client, websocket, data, started_event)
-        except asyncio.CancelledError:
-            # log canceling
-            raise
-