vital-agent-container-sdk 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

vital_agent_container/agent_container_app.py

@@ -42,11 +42,13 @@ class AgentContainerApp(FastAPI):
         self.app_home = app_home
         load_dotenv()
         self.config = ConfigUtils.load_config(app_home)
+        self.message_processor = AIMPMessageProcessor()
         self.add_routes()
 
     async def process_ws_message(self, client: httpx.AsyncClient, websocket: WebSocket, data: str,
                                  started_event: asyncio.Event):
-        await AIMPMessageProcessor.process_message(self.handler, self.config, client, websocket, data, started_event)
+        print(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):
         @self.get("/health")
@@ -67,8 +69,9 @@ class AgentContainerApp(FastAPI):
 
                     logger.info(f"Received message from {websocket.client.host}:{websocket.client.port}: {data}")
                     message_obj = json.loads(data)
-                    message_type = message_obj[0].get("type")
-                    message_intent = message_obj[0].get("http://vital.ai/ontology/vital-aimp#hasIntent")
+                    logger.info(f"Received message: {message_obj}")
+                    message_type = message_obj[0].get("type", None)
+                    message_intent = message_obj[0].get("http://vital.ai/ontology/vital-aimp#hasIntent", None)
                     logger.info(f"message_type: {message_type}")
                     logger.info(f"message_intent: {message_intent}")
 
@@ -96,9 +99,12 @@ class AgentContainerApp(FastAPI):
                     else:
                         logger.info(f"Processing message: {data}")
                         started_event = asyncio.Event()
-                        task = asyncio.create_task(self.process_ws_message(self.handler, client, websocket, data, started_event))
+                        task = asyncio.create_task(self.process_ws_message(client, websocket, data, started_event))
                         background_tasks.append(task)
                         await started_event.wait()
+                        logger.info(f"Completed Processing message: {data}")
+                        # break out of infinite loop
+                        break
             except WebSocketDisconnect:
                 logger.info("WebSocket connection closed by the client.")
                 # still try to close it in finally?

vital_agent_container/handler/aimp_message_handler_inf.py

@@ -1,5 +1,14 @@
+import asyncio
+import httpx
+from starlette.websockets import WebSocket
+from abc import ABC, abstractmethod
+
+
+class AIMPMessageHandlerInf(ABC):
+
+    @abstractmethod
+    async def process_message(self, config, client: httpx.AsyncClient, websocket: WebSocket, data: str, started_event: asyncio.Event):
+        pass
 
 
-class AIMPMessageHandlerInf:
-    pass
 

vital_agent_container/handler/impl/aimp_echo_message_handler.py

@@ -1,7 +1,23 @@
+import asyncio
+import httpx
+from starlette.websockets import WebSocket
 from vital_agent_container.handler.aimp_message_handler_inf import AIMPMessageHandlerInf
 
 
 class AIMPEchoMessageHandler(AIMPMessageHandlerInf):
-    pass
+    async def process_message(self, config, client: httpx.AsyncClient, websocket: WebSocket, data: str, started_event: asyncio.Event):
+        try:
+            print(f"Received Message: {data}")
+            await websocket.send_text(data)
+            print(f"Sent Message: {data}")
+            # await websocket.close(1000, "Processing Complete")
+            # print(f"Websocket closed.")
+            started_event.set()
+            print(f"Completed Event.")
+        except asyncio.CancelledError:
+            # log canceling
+            raise
+
+
 
 

vital_agent_container/processor/aimp_message_processor.py

@@ -1,3 +1,6 @@
+import asyncio
+
+from vital_agent_container.handler.aimp_message_handler_inf import AIMPMessageHandlerInf
 
 
 class AIMPMessageProcessor:
@@ -5,5 +8,11 @@ class AIMPMessageProcessor:
     def __init__(self):
         pass
 
-    async def process_message(self, handler, app_config, client, websocket, data, started_event):
-        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
+

{vital_agent_container_sdk-0.1.1.dist-info → vital_agent_container_sdk-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vital-agent-container-sdk
-Version: 0.1.1
+Version: 0.1.2
 Summary: Vital Agent Container SDK
 Home-page: https://github.com/vital-ai/vital-agent-container-python
 Author: Marc Hadfield
@@ -12,8 +12,8 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: vital-ai-vitalsigns ==0.1.19
-Requires-Dist: vital-ai-aimp ==0.1.6
+Requires-Dist: vital-ai-vitalsigns ==0.1.20
+Requires-Dist: vital-ai-aimp ==0.1.7
 Requires-Dist: httpx ==0.26.0
 Requires-Dist: python-json-logger ==2.0.7
 Requires-Dist: python-dotenv ==1.0.1

{vital_agent_container_sdk-0.1.1.dist-info → vital_agent_container_sdk-0.1.2.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=H9b5dzW6FDZMdqb2oaf7Nqtj2GS2APdl5YslXAdZsJI,5538
+vital_agent_container/agent_container_app.py,sha256=5lbge8OC8tEkjJ_TsZ9-zkRPPKA_Jr8gwwIFxmo7z6U,5879
 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=p2Itiil2K1acZ6Js_rShL5mJyrciqH2B_pqwmPJp_zw,41
+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=Rt4yE-_wlyIUdLhuBQ-dKQ7bycRDHCA6KgtphA4fgFA,155
+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=KcyzRQNhEuPM8PCg7Oq0GycIjAfVPoPLvjlYzd2CYR8,180
+vital_agent_container/processor/aimp_message_processor.py,sha256=4aaMEKxpzpXuH-kjLyqGY8t5PkqFif7Dr5ZaFHKEr2o,532
 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.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vital_agent_container_sdk-0.1.1.dist-info/METADATA,sha256=Obf9jQzG8K0CKLcVr9bg4ayWF114Xi1rOP8aWj-rdW0,1187
-vital_agent_container_sdk-0.1.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-vital_agent_container_sdk-0.1.1.dist-info/top_level.txt,sha256=acZrRgjFIj0mWxdLPWONKwMgxnFReQmyjttb8Z76fsc,22
-vital_agent_container_sdk-0.1.1.dist-info/RECORD,,
+vital_agent_container_sdk-0.1.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vital_agent_container_sdk-0.1.2.dist-info/METADATA,sha256=UqRcZ3v3Q_jkm_2luMEthBtiSGKzeKVDR265f9uSgrY,1187
+vital_agent_container_sdk-0.1.2.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+vital_agent_container_sdk-0.1.2.dist-info/top_level.txt,sha256=acZrRgjFIj0mWxdLPWONKwMgxnFReQmyjttb8Z76fsc,22
+vital_agent_container_sdk-0.1.2.dist-info/RECORD,,

{vital_agent_container_sdk-0.1.1.dist-info → vital_agent_container_sdk-0.1.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (72.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any