python sdk for superagi

Author: jagtarcontlo

.gitignore

@@ -0,0 +1,9 @@
+.idea
+**/.env
+**/.venv
+__pycache__
+**/__gitpycache__/
+venv
+.DS_Store
+dist
+superagi_client.egg-info

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 TransformerOptimus
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

__init__.py

@@ -0,0 +1,13 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="superagi_client",
+    version="0.0.1",
+    packages=find_packages(exclude=[".idea", ".git", "tests", "venv"]),
+    url="https://app.superagi.com",
+    license="MIT",
+    author="superagi",
+    author_email="mukunda@superagi.com",
+    description="Python package for Superagi",
+    install_requires=["pydantic==1.10.8", "requests==2.31.0", "pytest==7.3.2"],
+)

setup.py

@@ -0,0 +1 @@
+from superagi_client.superagi_client import SuperagiClient

superagi_client/__init__.py

@@ -0,0 +1,25 @@
+from typing import Optional, List
+
+from pydantic import BaseModel
+
+from superagi_client.dto.agent_schedule import AgentSchedule
+
+
+class AgentConfig(BaseModel):
+    name: str
+    description: str
+    project_id: Optional[int]
+    goal: List[str]
+    instruction: List[str]
+    agent_workflow: str
+    constraints: List[str]
+    tools: List[dict]
+    LTM_DB: Optional[str]
+    exit: Optional[str]
+    permission_type: Optional[str]
+    iteration_interval: int
+    model: str
+    schedule: Optional[AgentSchedule]
+    max_iterations: int
+    user_timezone: Optional[str]
+    knowledge: Optional[int]

superagi_client/dto/__init__.py

@@ -0,0 +1,9 @@
+from typing import Optional, List
+
+from pydantic import BaseModel
+
+
+class AgentExecution(BaseModel):
+    name: Optional[str]
+    goal: Optional[List[str]]
+    instruction: Optional[List[str]]

superagi_client/dto/agent_config.py

@@ -0,0 +1,8 @@
+from typing import Optional, List
+
+from pydantic import BaseModel
+
+
+class AgentRunFilter(BaseModel):
+    run_ids: Optional[List[int]]
+    run_status_filter: Optional[str]

superagi_client/dto/agent_execution.py

@@ -0,0 +1,12 @@
+from datetime import datetime
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class AgentSchedule(BaseModel):
+    agent_id: Optional[int]
+    start_time: datetime
+    recurrence_interval: Optional[str] = None
+    expiry_date: Optional[datetime] = None
+    expiry_runs: Optional[int] = -1

superagi_client/dto/agent_run_filter.py

@@ -0,0 +1,24 @@
+from pydantic import BaseModel
+from typing import Optional, List
+
+from superagi_client.dto.agent_schedule import AgentSchedule
+
+
+class AgentUpdateConfig(BaseModel):
+    name: Optional[str]
+    description: Optional[str]
+    project_id: Optional[int]
+    goal: Optional[List[str]]
+    instruction: Optional[List[str]]
+    agent_workflow: Optional[str]
+    constraints: Optional[List[str]]
+    tools: Optional[List[dict]]
+    LTM_DB: Optional[str]
+    exit: Optional[str]
+    permission_type: Optional[str]
+    iteration_interval: Optional[int]
+    model: Optional[str]
+    schedule: Optional[AgentSchedule]
+    max_iterations: Optional[int]
+    user_timezone: Optional[str]
+    knowledge: Optional[int]

superagi_client/dto/agent_schedule.py

@@ -0,0 +1,18 @@
+from superagi_client.lib.logger import logger
+
+
+class BadRequestException(Exception):
+    def __init__(self, message="Bad Request", status_code=400, additional_info=None):
+        self.message = message
+        self.status_code = status_code
+        self.additional_info = additional_info
+        super().__init__(
+            f"{self.message} => {self.additional_info}"
+            if self.additional_info
+            else self.message
+        )
+
+        logger.error(
+            f"BadRequestException: {self.message}, Status Code: {self.status_code}, "
+            f"Additional Info: {self.additional_info}"
+        )

superagi_client/dto/agent_update_config.py

@@ -0,0 +1,18 @@
+from superagi_client.lib.logger import logger
+
+
+class ConflictException(Exception):
+    def __init__(self, message="Unauthorized", status_code=409, additional_info=None):
+        self.message = message
+        self.status_code = status_code
+        self.additional_info = additional_info
+        super().__init__(
+            f"{self.message} => {self.additional_info}"
+            if self.additional_info
+            else self.message
+        )
+
+        logger.error(
+            f"BadRequestException: {self.message}, Status Code: {self.status_code}, "
+            f"Additional Info: {self.additional_info}"
+        )

superagi_client/exceptions/__init__.py

@@ -0,0 +1,25 @@
+from superagi_client.exceptions.bad_request_exception import BadRequestException
+from superagi_client.exceptions.conflict_exception import ConflictException
+from superagi_client.exceptions.internal_server_error_exception import (
+    InternalServerErrorException,
+)
+from superagi_client.exceptions.not_found_exception import NotFoundException
+from superagi_client.exceptions.unauthorized_exception import UnauthorizedException
+from superagi_client.lib.logger import logger
+
+
+def http_status_code_to_exception(status_code, additional_info):
+    status_code_to_exception_map = {
+        400: BadRequestException,
+        401: UnauthorizedException,
+        404: NotFoundException,
+        409: ConflictException,
+        500: InternalServerErrorException,
+    }
+
+    exception = status_code_to_exception_map.get(status_code)
+    if exception is None:
+        logger.error(f"Status Code:{status_code}, Additional Info:{additional_info}")
+    raise Exception() if exception is None else exception(
+        additional_info=additional_info
+    )

superagi_client/exceptions/bad_request_exception.py

@@ -0,0 +1,20 @@
+from superagi_client.lib.logger import logger
+
+
+class InternalServerErrorException(Exception):
+    def __init__(
+        self, message="Internal Server Error", status_code=500, additional_info=None
+    ):
+        self.message = message
+        self.status_code = status_code
+        self.additional_info = additional_info
+        super().__init__(
+            f"{self.message} => {self.additional_info}"
+            if self.additional_info
+            else self.message
+        )
+
+        logger.error(
+            f"BadRequestException: {self.message}, Status Code: {self.status_code}, "
+            f"Additional Info: {self.additional_info}"
+        )

superagi_client/exceptions/conflict_exception.py

@@ -0,0 +1,18 @@
+from superagi_client.lib.logger import logger
+
+
+class NotFoundException(Exception):
+    def __init__(self, message="Not Found", status_code=404, additional_info=None):
+        self.message = message
+        self.status_code = status_code
+        self.additional_info = additional_info
+        super().__init__(
+            f"{self.message} => {self.additional_info}"
+            if self.additional_info
+            else self.message
+        )
+
+        logger.error(
+            f"BadRequestException: {self.message}, Status Code: {self.status_code}, "
+            f"Additional Info: {self.additional_info}"
+        )

superagi_client/exceptions/http_exception.py

@@ -0,0 +1,18 @@
+from superagi_client.lib.logger import logger
+
+
+class UnauthorizedException(Exception):
+    def __init__(self, message="Unauthorized", status_code=401, additional_info=None):
+        self.message = message
+        self.status_code = status_code
+        self.additional_info = additional_info
+        super().__init__(
+            f"{self.message} => {self.additional_info}"
+            if self.additional_info
+            else self.message
+        )
+
+        logger.error(
+            f"BadRequestException: {self.message}, Status Code: {self.status_code}, "
+            f"Additional Info: {self.additional_info}"
+        )

superagi_client/exceptions/internal_server_error_exception.py

@@ -0,0 +1,12 @@
+import requests as requests
+
+from superagi_client.exceptions.unauthorized_exception import UnauthorizedException
+
+
+def validate_api_key(base_url: str, api_key: str):
+    response = requests.get(
+        f"{base_url}/api/api-keys/validate", headers={"X-API-Key": api_key}
+    )
+
+    if response.status_code == 401:
+        raise UnauthorizedException()

superagi_client/exceptions/not_found_exception.py

@@ -0,0 +1,108 @@
+import logging
+import inspect
+
+
+class CustomLogRecord(logging.LogRecord):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        frame = inspect.currentframe().f_back
+        while frame:
+            if (
+                frame.f_globals["__name__"] != __name__
+                and frame.f_globals["__name__"] != "logging"
+            ):
+                break
+            frame = frame.f_back
+
+        if frame:
+            self.filename = frame.f_code.co_filename
+            self.lineno = frame.f_lineno
+        else:
+            self.filename = "unknown"
+            self.lineno = 0
+
+
+class SingletonMeta(type):
+    _instances = {}
+
+    def __call__(cls, *args, **kwargs):
+        if cls not in cls._instances:
+            instance = super().__call__(*args, **kwargs)
+            cls._instances[cls] = instance
+        return cls._instances[cls]
+
+
+class Logger(metaclass=SingletonMeta):
+    def __init__(
+        self, logger_name="Super AGI superagi_client", log_level=logging.DEBUG
+    ):
+        if not hasattr(self, "logger"):
+            self.logger = logging.getLogger(logger_name)
+            self.logger.setLevel(log_level)
+            self.logger.makeRecord = self._make_custom_log_record
+
+            console_handler = logging.StreamHandler()
+            console_handler.setLevel(log_level)
+
+            formatter = logging.Formatter(
+                "%(asctime)s - %(name)s - %(levelname)s - [%(filename)s:%(lineno)d] - %(message)s",
+                datefmt="%Y-%m-%d %H:%M:%S %Z",
+            )
+
+            console_handler.setFormatter(formatter)
+            self.logger.addHandler(console_handler)
+
+    def _make_custom_log_record(
+        self,
+        name,
+        level,
+        fn,
+        lno,
+        msg,
+        args,
+        exc_info,
+        func=None,
+        extra=None,
+        sinfo=None,
+    ):
+        return CustomLogRecord(
+            name,
+            level,
+            fn,
+            lno,
+            msg,
+            args,
+            exc_info,
+            func=func,
+            extra=extra,
+            sinfo=sinfo,
+        )
+
+    def debug(self, message, *args):
+        self.logger.debug(message)
+        if args:
+            self.logger.debug(*args)
+
+    def info(self, message, *args):
+        self.logger.info(message)
+        if args:
+            self.logger.info(*args)
+
+    def warning(self, message, *args):
+        self.logger.warning(message)
+        if args:
+            self.logger.warning(*args)
+
+    def error(self, message, *args):
+        self.logger.error(message)
+        if args:
+            self.logger.error(*args)
+
+    def critical(self, message, *args):
+        self.logger.critical(message)
+        if args:
+            self.logger.critical(*args)
+
+
+logger = Logger("Super AGI superagi_client")