mirror of
https://github.com/Ladebeze66/llm_ticket3.git
synced 2025-12-16 20:47:49 +01:00
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
|
|
|
from ._hooks import SDKHooks
|
|
from ._version import (
|
|
__gen_version__,
|
|
__openapi_doc_version__,
|
|
__user_agent__,
|
|
__version__,
|
|
)
|
|
from .httpclient import AsyncHttpClient, HttpClient
|
|
from .utils import Logger, RetryConfig, remove_suffix
|
|
from dataclasses import dataclass
|
|
from mistralai_azure import models
|
|
from mistralai_azure.types import OptionalNullable, UNSET
|
|
from pydantic import Field
|
|
from typing import Callable, Dict, Optional, Tuple, Union
|
|
|
|
|
|
SERVER_EU = "eu"
|
|
r"""EU Production server"""
|
|
SERVERS = {
|
|
SERVER_EU: "https://api.mistral.ai",
|
|
}
|
|
"""Contains the list of servers available to the SDK"""
|
|
|
|
|
|
@dataclass
|
|
class SDKConfiguration:
|
|
client: Union[HttpClient, None]
|
|
client_supplied: bool
|
|
async_client: Union[AsyncHttpClient, None]
|
|
async_client_supplied: bool
|
|
debug_logger: Logger
|
|
security: Optional[Union[models.Security, Callable[[], models.Security]]] = None
|
|
server_url: Optional[str] = ""
|
|
server: Optional[str] = ""
|
|
language: str = "python"
|
|
openapi_doc_version: str = __openapi_doc_version__
|
|
sdk_version: str = __version__
|
|
gen_version: str = __gen_version__
|
|
user_agent: str = __user_agent__
|
|
retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
|
|
timeout_ms: Optional[int] = None
|
|
|
|
def __post_init__(self):
|
|
self._hooks = SDKHooks()
|
|
|
|
def get_server_details(self) -> Tuple[str, Dict[str, str]]:
|
|
if self.server_url is not None and self.server_url:
|
|
return remove_suffix(self.server_url, "/"), {}
|
|
if not self.server:
|
|
self.server = SERVER_EU
|
|
|
|
if self.server not in SERVERS:
|
|
raise ValueError(f'Invalid server "{self.server}"')
|
|
|
|
return SERVERS[self.server], {}
|
|
|
|
def get_hooks(self) -> SDKHooks:
|
|
return self._hooks
|