amesa-api 0.1.0__tar.gz

amesa_api-0.1.0/PKG-INFO

@@ -0,0 +1,45 @@
+Metadata-Version: 2.4
+Name: amesa-api
+Version: 0.1.0
+Summary: the Amesa API
+Author-email: Hunter Park <hunter@amesa.com>
+Requires-Python: >=3.10, <3.13
+Description-Content-Type: text/markdown
+Requires-Dist: cython
+Requires-Dist: ruff
+Requires-Dist: simple-term-menu
+Requires-Dist: aiohttp
+Requires-Dist: httpretty==1.0.5
+Requires-Dist: pytest-asyncio
+Requires-Dist: pre-commit
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-python
+
+# Amesa
+
+Amesa helps you build Autonomous Agents! Through an easy SDK you get access to outscaled simulator training tools.
+
+## Licenses / Seats
+
+Amesa is an enterprise platform and requires an API key to use.
+
+We offer 3 types of licenses:
+
+- **Enterprise:** for enterprise projects and agent creations
+- **Personal:** for personal usage and testing (typically offered to System Integrators)
+- **Trial:** validate the Amesa platform (requires sales contact person)
+
+One key is needed per user (seat-based licensing)
+
+You can request an API Key through the following methods:
+
+- [Discord](https://discord.gg/EQ3BgJt9NC)
+- [Mail](mailto:sales@amesa.com?subject=REQUEST%20API%20KEY%20-%20COMPANY_NAME%20-%20NAME&body=Hi%2C%0D%0A%0D%0AI%20would%20like%20to%20request%20an%20API%20key%20for%20my%20company%20to%20get%20started%20with%20Amesa.%0D%0A%0D%0A*%20Company%20Name%3A%20COMPANY_NAME%0D%0A*%20Seats%3A%20NO_OF_SEATS%0D%0A*%20License%20Type%3A%20Enterprise%20%7C%20Personal%20%7C%20Trial%20%28keep%20what%20is%20required%29%0D%0A%0D%0AKind%20Regards%2C%0D%0ANAME%0D%0AFUNCTION)
+
+## Getting Started
+
+1. Download the Amesa SDK: `pip install amesa`
+2. Request an API Key
+3. Get a simulator or use one of the [prebuilt ones](https://hub.docker.com/u/composabl)
+4. Create an Agent or see our [examples](https://github.com/Amesa/examples.amesa.com)

amesa_api-0.1.0/README.md

@@ -0,0 +1,27 @@
+# Amesa
+
+Amesa helps you build Autonomous Agents! Through an easy SDK you get access to outscaled simulator training tools.
+
+## Licenses / Seats
+
+Amesa is an enterprise platform and requires an API key to use.
+
+We offer 3 types of licenses:
+
+- **Enterprise:** for enterprise projects and agent creations
+- **Personal:** for personal usage and testing (typically offered to System Integrators)
+- **Trial:** validate the Amesa platform (requires sales contact person)
+
+One key is needed per user (seat-based licensing)
+
+You can request an API Key through the following methods:
+
+- [Discord](https://discord.gg/EQ3BgJt9NC)
+- [Mail](mailto:sales@amesa.com?subject=REQUEST%20API%20KEY%20-%20COMPANY_NAME%20-%20NAME&body=Hi%2C%0D%0A%0D%0AI%20would%20like%20to%20request%20an%20API%20key%20for%20my%20company%20to%20get%20started%20with%20Amesa.%0D%0A%0D%0A*%20Company%20Name%3A%20COMPANY_NAME%0D%0A*%20Seats%3A%20NO_OF_SEATS%0D%0A*%20License%20Type%3A%20Enterprise%20%7C%20Personal%20%7C%20Trial%20%28keep%20what%20is%20required%29%0D%0A%0D%0AKind%20Regards%2C%0D%0ANAME%0D%0AFUNCTION)
+
+## Getting Started
+
+1. Download the Amesa SDK: `pip install amesa`
+2. Request an API Key
+3. Get a simulator or use one of the [prebuilt ones](https://hub.docker.com/u/composabl)
+4. Create an Agent or see our [examples](https://github.com/Amesa/examples.amesa.com)

amesa_api-0.1.0/amesa_api/__init__.py

@@ -0,0 +1,10 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+# We want to expose these classes to the user at the base of core
+# ex: `from amesa.core import Agent`
+
+# Continue with the imports
+from .nocode import APINoCode
+from .common import get_ssl_connector

amesa_api-0.1.0/amesa_api/common.py

@@ -0,0 +1,116 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+from typing import Union
+
+import aiohttp
+import json
+import os
+import ssl
+
+
+def get_ssl_connector():
+    ssl_cert_file = os.getenv("SSL_CERT_FILE")
+    ssl_cert_path = os.getenv("SSL_CERT_PATH")
+    ssl_ctx = ssl.create_default_context(
+        ssl.Purpose.SERVER_AUTH, cafile=ssl_cert_file, capath=ssl_cert_path
+    )
+    return aiohttp.TCPConnector(ssl_context=ssl_ctx)
+
+
+class APIBase:
+    def __init__(self, base_url):
+        self.base_url = base_url
+        self.token = None
+
+    def _get_headers(self):
+        headers = {"Content-Type": "application/json", "Accept": "application/json"}
+
+        if self.token is not None:
+            headers["Authorization"] = f"Bearer {self.token}"
+
+        return headers
+
+    async def get(self, path: str, headers={}):
+        """
+        Make a GET request to the API
+        """
+        headers_base = self._get_headers()
+        headers_base.update(headers)
+        connector = get_ssl_connector()
+        async with aiohttp.ClientSession(connector=connector) as session:
+            async with session.get(
+                self.base_url + path, headers=headers_base
+            ) as response:
+                return await self._process_response(response)
+
+    async def post(self, path: str, data: Union[dict, str], headers={}):
+        """
+        Make a POST request to the API
+        """
+        headers_base = self._get_headers()
+        headers_base.update(headers)
+
+        if isinstance(data, dict):
+            data = json.dumps(data)
+        else:
+            del headers_base["Content-Type"]
+
+        connector = get_ssl_connector()
+        async with aiohttp.ClientSession(connector=connector) as session:
+            async with session.post(
+                self.base_url + path, headers=headers_base, data=data
+            ) as response:
+                return await self._process_response(response)
+
+    async def put(self, path: str, data: Union[dict, str], headers={}):
+        """
+        Make a PUT request to the API
+        """
+        headers_base = self._get_headers()
+        headers_base.update(headers)
+
+        if isinstance(data, dict):
+            data = json.dumps(data)
+        else:
+            del headers_base["Content-Type"]
+
+        connector = get_ssl_connector()
+        async with aiohttp.ClientSession(connector=connector) as session:
+            async with session.put(
+                self.base_url + path, headers=headers_base, data=data
+            ) as response:
+                return await self._process_response(response)
+
+    async def delete(self, path: str, headers={}):
+        """
+        Make a DELETE request to the API
+        """
+        headers_base = self._get_headers()
+        headers_base.update(headers)
+
+        connector = get_ssl_connector()
+        async with aiohttp.ClientSession(connector=connector) as session:
+            async with session.delete(
+                self.base_url + path, headers=headers_base
+            ) as response:
+                return await self._process_response(response)
+
+    async def _process_response(self, response):
+        """
+        Process the response from the API
+        """
+        if response.status >= 200 and response.status < 202:
+            try:
+                return await response.json()
+            except Exception:
+                return await response.text()
+        elif response.status >= 202 and response.status < 300:
+            return None
+        else:
+            try:
+                res_json = await response.json()
+                raise Exception(f"Error: {response.status} - {res_json['message']}")
+            except Exception:
+                txt = await response.text()
+                raise Exception(f"Error: {response.status} - {txt}")

amesa_api-0.1.0/amesa_api/controller/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+from .main import APIController

amesa_api-0.1.0/amesa_api/controller/job_status_enum.py

@@ -0,0 +1,14 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+from enum import Enum
+
+
+class JobStatusEnum(str, Enum):
+    STARTING = "starting"
+    STARTED = "started"
+    STOPPING = "stopping"
+    STOPPED = "stopped"
+    FAILED = "failed"
+    DONE = "done"

amesa_api-0.1.0/amesa_api/controller/main.py

@@ -0,0 +1,60 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+from amesa_api.common import APIBase
+from amesa_api.controller.job_status_enum import JobStatusEnum
+
+
+class APIController(APIBase):
+    def __init__(self, base_url):
+        super().__init__(base_url)
+
+        self.admin = APIControllerAdmin(self)
+        self.sdk = APIControllerSdk(self)
+        self.direct_mode = APIControllerDirectMode(self)
+
+
+class APIControllerDirectMode:
+    def __init__(self, api):
+        self.api: APIController = api
+
+    async def submit(self, job_id: str, serialized_trainer: str):
+        body = {"id": job_id, "json": serialized_trainer}
+
+        await self.api.post("/jobs", body)
+
+
+class APIControllerAdmin:
+    def __init__(self, api):
+        self.api: APIController = api
+
+    async def list_jobs(self):
+        return await self.api.get("/api/admin/v1/jobs")
+
+    async def get_job(self, job_id: str):
+        return await self.api.get(f"/api/admin/v1/jobs/{job_id}")
+
+    async def cancel_job(self, job_id: str):
+        return await self.api.delete(f"/api/admin/v1/jobs/{job_id}")
+
+
+class APIControllerSdk:
+    def __init__(self, api):
+        self.api: APIController = api
+
+    async def heartbeat(self, job_id: str):
+        body = {
+            "health": "ok",
+        }
+
+        res = await self.api.post(f"/api/sdk/v1/jobs/{job_id}/heartbeat", body)
+
+        return res
+
+    async def status(self, status: JobStatusEnum, job_id: str):
+        status = {"status": status}
+
+        res = await self.api.post(f"/api/sdk/v1/jobs/{job_id}/status", status)
+
+        return res

amesa_api-0.1.0/amesa_api/controller/service.py

@@ -0,0 +1,87 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+import asyncio
+import threading
+import time
+from os import getenv
+
+import amesa_api.util.logger as logger_util
+from amesa_api.controller.job_status_enum import JobStatusEnum
+from amesa_api.controller.main import APIController
+
+logger = logger_util.get_logger(__name__)
+
+
+class ControllerService:
+    _instance = None
+    heartbeat_thread = None
+    healthy = True
+    job_id: str = None
+
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+
+        return cls._instance
+
+    def __init__(self, url=None):
+        if url is None:
+            controller_url = getenv("AMESA_CONTROLLER_URL", None)
+        else:
+            controller_url = url
+
+        if controller_url is None:
+            raise Exception("AMESA_CONTROLLER_URL could not be read!")
+
+        self.client = APIController(controller_url)
+
+    def set_job_id(self, job_id):
+        self.job_id = job_id
+
+    async def start_heartbeat_thread(self):
+        try:
+            await self.client.sdk.heartbeat(self.job_id)
+        except Exception as e:
+            logger.error(f"Failed to connect to controller: {e}")
+            return
+
+        self.heartbeat_thread = run_async_in_daemon_thread(self.send_heartbeat)
+
+    def mark_unhealthy(self):
+        self.healthy = False
+
+    async def send_heartbeat(self):
+        while self.healthy:
+            try:
+                await self.client.sdk.heartbeat(self.job_id)
+            except Exception as e:
+                logger.error(f"Failed to send heartbeat to controller: {e}")
+
+            time.sleep(10)
+
+        logger.info(
+            "ControllerService heartbeat task stopped because the job is marked as unhealthy"
+        )
+
+    async def send_status(self, status: JobStatusEnum):
+        await self.client.sdk.status(status, self.job_id)
+
+
+def run_async_in_daemon_thread(fn, *args, **kwargs):
+    """
+    Runs an asynchronous function in a separate daemon thread, and waits for it to complete.
+    Returns the result of the asynchronous function.
+    """
+
+    def run(loop):
+        asyncio.set_event_loop(loop)
+
+        loop.run_until_complete(fn(*args, **kwargs))
+
+    loop = asyncio.new_event_loop()
+    t = threading.Thread(target=run, args=(loop,), daemon=True)
+    t.start()
+
+    return t

amesa_api-0.1.0/amesa_api/nocode/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (C) Amesa, Inc - All Rights Reserved
+# Unauthorized copying of this file, via any medium is strictly prohibited
+# Proprietary and confidential
+
+from .main import APINoCode