core: remove typing

1 year ago · bbe27f208d
parent 8ed39cac53
commit bbe27f208d
2 changed files with 1 additions and 199 deletions
--- a/schemas/typings.py
+++ b/schemas/typings.py
@ -1,198 +0,0 @@
 """
 A module that contains all the types used in this project
 """
 import os
 import platform
 from enum import Enum
 from typing import Union
 python_version = list(platform.python_version_tuple())
 SUPPORT_ADD_NOTES = int(python_version[0]) >= 3 and int(python_version[1]) >= 11
 class ChatbotError(Exception):
    """
    Base class for all Chatbot errors in this Project
    """
    def __init__(self, *args: object) -> None:
        if SUPPORT_ADD_NOTES:
            super().add_note(
                "Please check that the input is correct, or you can resolve this issue by filing an issue",
            )
            super().add_note("Project URL: https://github.com/acheong08/ChatGPT")
        super().__init__(*args)
 class ActionError(ChatbotError):
    """
    Subclass of ChatbotError
    An object that throws an error because the execution of an operation is blocked
    """
    def __init__(self, *args: object) -> None:
        if SUPPORT_ADD_NOTES:
            super().add_note(
                "The current operation is not allowed, which may be intentional",
            )
        super().__init__(*args)
 class ActionNotAllowedError(ActionError):
    """
    Subclass of ActionError
    An object that throws an error because the execution of an unalloyed operation is blocked
    """
 class ActionRefuseError(ActionError):
    """
    Subclass of ActionError
    An object that throws an error because the execution of a refused operation is blocked.
    """
 class CLIError(ChatbotError):
    """
    Subclass of ChatbotError
    The error caused by a CLI program error
    """
 class ErrorType(Enum):
    """
    Enumeration class for different types of errors.
    """
    USER_ERROR = -1
    UNKNOWN_ERROR = 0
    SERVER_ERROR = 1
    RATE_LIMIT_ERROR = 2
    INVALID_REQUEST_ERROR = 3
    EXPIRED_ACCESS_TOKEN_ERROR = 4
    INVALID_ACCESS_TOKEN_ERROR = 5
    PROHIBITED_CONCURRENT_QUERY_ERROR = 6
    AUTHENTICATION_ERROR = 7
    CLOUDFLARE_ERROR = 8
 class Error(ChatbotError):
    """
    Base class for exceptions in V1 module.
    """
    def __init__(
        self,
        source: str,
        message: str,
        *args: object,
        code: Union[ErrorType, int] = ErrorType.UNKNOWN_ERROR,
    ) -> None:
        self.source: str = source
        self.message: str = message
        self.code: ErrorType | int = code
        super().__init__(*args)
    def __str__(self) -> str:
        return f"{self.source}: {self.message} (code: {self.code})"
    def __repr__(self) -> str:
        return f"{self.source}: {self.message} (code: {self.code})"
 class AuthenticationError(ChatbotError):
    """
    Subclass of ChatbotError
    The object of the error thrown by a validation failure or exception
    """
    def __init__(self, *args: object) -> None:
        if SUPPORT_ADD_NOTES:
            super().add_note(
                "Please check if your key is correct, maybe it may not be valid",
            )
        super().__init__(*args)
 class APIConnectionError(ChatbotError):
    """
    Subclass of ChatbotError
    An exception object thrown when an API connection fails or fails to connect due to network or
    other miscellaneous reasons
    """
    def __init__(self, *args: object) -> None:
        if SUPPORT_ADD_NOTES:
            super().add_note(
                "Please check if there is a problem with your network connection",
            )
        super().__init__(*args)
 class NotAllowRunning(ActionNotAllowedError):
    """
    Subclass of ActionNotAllowedError
    Direct startup is not allowed for some reason
    """
 class ResponseError(APIConnectionError):
    """
    Subclass of APIConnectionError
    Error objects caused by API request errors due to network or other miscellaneous reasons
    """
 class OpenAIError(APIConnectionError):
    """
    Subclass of APIConnectionError
    Error objects caused by OpenAI's own server errors
    """
 class RequestError(APIConnectionError):
    """
    Subclass of APIConnectionError
    There is a problem with the API response due to network or other miscellaneous reasons, or there
    is no reply to the object that caused the error at all
    """
 class Colors:
    """
    Colors for printing
    """
    HEADER = "\033[95m"
    OKBLUE = "\033[94m"
    OKCYAN = "\033[96m"
    OKGREEN = "\033[92m"
    WARNING = "\033[93m"
    FAIL = "\033[91m"
    ENDC = "\033[0m"
    BOLD = "\033[1m"
    UNDERLINE = "\033[4m"
    def __init__(self) -> None:
        if os.getenv("NO_COLOR"):
            Colors.HEADER = ""
            Colors.OKBLUE = ""
            Colors.OKCYAN = ""
            Colors.OKGREEN = ""
            Colors.WARNING = ""
            Colors.FAIL = ""
            Colors.ENDC = ""
            Colors.BOLD = ""
            Colors.UNDERLINE = ""
--- a/swarms/models/revgpt.py
+++ b/swarms/models/revgpt.py
@ -1796,7 +1796,7 @@ class RevChatGPTModel:
    def run(self, task: str) -> str:
        self.start_time = time.time()
        prev_text = ""
-        for data in self.chatbot.ask(task):
+        for data in self.chatbot.ask(task, fileinfo=None):
            message = data["message"][len(prev_text):]
            prev_text = data["message"]
        self.end_time = time.time()