AI butler Jarvis

agent_jarvis/jarvis/utils/asynchttp.py

@@ -0,0 +1,29 @@
+import asyncio
+import json
+import time
+from typing import List, Dict
+import aiohttp
+
+
+async def do_post(url, body, params=None) -> Dict | List:
+    if not isinstance(body, str):
+        body = json.dumps(body)
+    headers = {
+        'accept': 'application/json',
+        'Content-Type': 'application/json',
+    }
+
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=headers, data=body, params=params) as response:
+            return await response.json()
+
+
+async def do_get(url, params=None) -> Dict | List:
+    headers = {
+        'accept': 'application/json',
+        'Content-Type': 'application/json',
+    }
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, headers=headers, params=params) as response:
+            return await response.json()

agent_jarvis/jarvis/utils/function_error.py

@@ -0,0 +1,13 @@
+EC_SUCCESS = 0
+
+EC_UNKNOWN_ERROR = -1
+
+EC_RESET = 1
+
+EC_DECODE_JSON_ERROR = 100
+
+
+class FunctionError(Exception):
+    def __init__(self, code, msg):
+        self.code = code
+        self.msg = msg

agent_jarvis/jarvis/utils/incoming_chat_message_parser.py

@@ -0,0 +1,65 @@
+import json
+
+from jarvis.logger import logger
+
+
+class IncomingChatMessage:
+    user_id: str = None
+    chat_id: str = None
+    message_type: str = None
+    message_content: str = None
+    message_id: str = None
+
+    def __init__(self):
+        pass
+
+
+def parse_incoming_chat_message(data: str | dict):
+    """
+    The expected format of data is
+    {
+      user: {
+        id: string
+      }
+      chat: {
+        id: string
+      }
+      message: {
+        type: 'text' | 'voice' | ...
+        content: string
+        id: string
+      }
+    }
+    """
+    result = IncomingChatMessage()
+    try:
+        if isinstance(data, dict):
+            obj = data
+        else:
+            obj = json.loads(data)
+        result.user_id = obj["user"]["id"]
+        result.chat_id = obj["chat"]["id"]
+        result.message_type = obj["message"]["type"]
+        result.message_id = obj["message"]["id"]
+        result.message_content = obj["message"]["content"]
+        # TODO: Check if they are str
+        return result
+    except Exception as e:
+        logger.debug(f"An invalid message from session: {data}")
+        return None
+
+
+def assemble_json_message(msg_type: str, msg: str, user_id: str, session_id: str, message_id: str):
+    return {
+        "user": {
+            "id": user_id
+        },
+        "chat": {
+            "id": session_id
+        },
+        "message": {
+            "type": msg_type,
+            "content": msg,
+            "id": message_id
+        }
+    }