chore: initial public snapshot for github upload
This commit is contained in:
Your Name
@@ -0,0 +1,268 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import types
|
||||
from abc import ABC, abstractmethod
|
||||
from typing import TYPE_CHECKING, Any
|
||||
|
||||
import httpx
|
||||
|
||||
from litellm.types.containers.main import ContainerCreateOptionalRequestParams
|
||||
from litellm.types.router import GenericLiteLLMParams
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from litellm.litellm_core_utils.litellm_logging import Logging as _LiteLLMLoggingObj
|
||||
from litellm.types.containers.main import (
|
||||
ContainerFileListResponse as _ContainerFileListResponse,
|
||||
)
|
||||
from litellm.types.containers.main import (
|
||||
ContainerListResponse as _ContainerListResponse,
|
||||
)
|
||||
from litellm.types.containers.main import ContainerObject as _ContainerObject
|
||||
from litellm.types.containers.main import (
|
||||
DeleteContainerResult as _DeleteContainerResult,
|
||||
)
|
||||
|
||||
from ..chat.transformation import BaseLLMException as _BaseLLMException
|
||||
|
||||
LiteLLMLoggingObj = _LiteLLMLoggingObj
|
||||
BaseLLMException = _BaseLLMException
|
||||
ContainerObject = _ContainerObject
|
||||
DeleteContainerResult = _DeleteContainerResult
|
||||
ContainerListResponse = _ContainerListResponse
|
||||
ContainerFileListResponse = _ContainerFileListResponse
|
||||
else:
|
||||
LiteLLMLoggingObj = Any
|
||||
BaseLLMException = Any
|
||||
ContainerObject = Any
|
||||
DeleteContainerResult = Any
|
||||
ContainerListResponse = Any
|
||||
ContainerFileListResponse = Any
|
||||
|
||||
|
||||
class BaseContainerConfig(ABC):
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@classmethod
|
||||
def get_config(cls):
|
||||
return {
|
||||
k: v
|
||||
for k, v in cls.__dict__.items()
|
||||
if not k.startswith("__")
|
||||
and not k.startswith("_abc")
|
||||
and not isinstance(
|
||||
v,
|
||||
(
|
||||
types.FunctionType,
|
||||
types.BuiltinFunctionType,
|
||||
classmethod,
|
||||
staticmethod,
|
||||
),
|
||||
)
|
||||
and v is not None
|
||||
}
|
||||
|
||||
@abstractmethod
|
||||
def get_supported_openai_params(self) -> list:
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def map_openai_params(
|
||||
self,
|
||||
container_create_optional_params: ContainerCreateOptionalRequestParams,
|
||||
drop_params: bool,
|
||||
) -> dict:
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def validate_environment(
|
||||
self,
|
||||
headers: dict,
|
||||
api_key: str | None = None,
|
||||
) -> dict:
|
||||
return {}
|
||||
|
||||
@abstractmethod
|
||||
def get_complete_url(
|
||||
self,
|
||||
api_base: str | None,
|
||||
litellm_params: dict,
|
||||
) -> str:
|
||||
"""Get the complete url for the request.
|
||||
|
||||
OPTIONAL - Some providers need `model` in `api_base`.
|
||||
"""
|
||||
if api_base is None:
|
||||
msg = "api_base is required"
|
||||
raise ValueError(msg)
|
||||
return api_base
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_create_request(
|
||||
self,
|
||||
name: str,
|
||||
container_create_optional_request_params: dict,
|
||||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
) -> dict:
|
||||
"""Transform the container creation request.
|
||||
|
||||
Returns:
|
||||
dict: Request data for container creation.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_create_response(
|
||||
self,
|
||||
raw_response: httpx.Response,
|
||||
logging_obj: LiteLLMLoggingObj,
|
||||
) -> ContainerObject:
|
||||
"""Transform the container creation response."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_list_request(
|
||||
self,
|
||||
api_base: str,
|
||||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
after: str | None = None,
|
||||
limit: int | None = None,
|
||||
order: str | None = None,
|
||||
extra_query: dict[str, Any] | None = None,
|
||||
) -> tuple[str, dict]:
|
||||
"""Transform the container list request into a URL and params.
|
||||
|
||||
Returns:
|
||||
tuple[str, dict]: (url, params) for the container list request.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_list_response(
|
||||
self,
|
||||
raw_response: httpx.Response,
|
||||
logging_obj: LiteLLMLoggingObj,
|
||||
) -> ContainerListResponse:
|
||||
"""Transform the container list response."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_retrieve_request(
|
||||
self,
|
||||
container_id: str,
|
||||
api_base: str,
|
||||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
) -> tuple[str, dict]:
|
||||
"""Transform the container retrieve request into a URL and data/params.
|
||||
|
||||
Returns:
|
||||
tuple[str, dict]: (url, params) for the container retrieve request.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_retrieve_response(
|
||||
self,
|
||||
raw_response: httpx.Response,
|
||||
logging_obj: LiteLLMLoggingObj,
|
||||
) -> ContainerObject:
|
||||
"""Transform the container retrieve response."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_delete_request(
|
||||
self,
|
||||
container_id: str,
|
||||
api_base: str,
|
||||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
) -> tuple[str, dict]:
|
||||
"""Transform the container delete request into a URL and data.
|
||||
|
||||
Returns:
|
||||
tuple[str, dict]: (url, data) for the container delete request.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_delete_response(
|
||||
self,
|
||||
raw_response: httpx.Response,
|
||||
logging_obj: LiteLLMLoggingObj,
|
||||
) -> DeleteContainerResult:
|
||||
"""Transform the container delete response."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_file_list_request(
|
||||
self,
|
||||
container_id: str,
|
||||
api_base: str,
|
||||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
after: str | None = None,
|
||||
limit: int | None = None,
|
||||
order: str | None = None,
|
||||
extra_query: dict[str, Any] | None = None,
|
||||
) -> tuple[str, dict]:
|
||||
"""Transform the container file list request into a URL and params.
|
||||
|
||||
Returns:
|
||||
tuple[str, dict]: (url, params) for the container file list request.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_file_list_response(
|
||||
self,
|
||||
raw_response: httpx.Response,
|
||||
logging_obj: LiteLLMLoggingObj,
|
||||
) -> ContainerFileListResponse:
|
||||
"""Transform the container file list response."""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_file_content_request(
|
||||
self,
|
||||
container_id: str,
|
||||
file_id: str,
|
||||
api_base: str,
|
||||
litellm_params: GenericLiteLLMParams,
|
||||
headers: dict,
|
||||
) -> tuple[str, dict]:
|
||||
"""Transform the container file content request into a URL and params.
|
||||
|
||||
Returns:
|
||||
tuple[str, dict]: (url, params) for the container file content request.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def transform_container_file_content_response(
|
||||
self,
|
||||
raw_response: httpx.Response,
|
||||
logging_obj: LiteLLMLoggingObj,
|
||||
) -> bytes:
|
||||
"""Transform the container file content response.
|
||||
|
||||
Returns:
|
||||
bytes: The raw file content.
|
||||
"""
|
||||
...
|
||||
|
||||
def get_error_class(
|
||||
self,
|
||||
error_message: str,
|
||||
status_code: int,
|
||||
headers: dict | httpx.Headers,
|
||||
) -> BaseLLMException:
|
||||
from ..chat.transformation import BaseLLMException
|
||||
|
||||
raise BaseLLMException(
|
||||
status_code=status_code,
|
||||
message=error_message,
|
||||
headers=headers,
|
||||
)
|
||||
Reference in New Issue
Block a user