bella-openapi 1.0.2__py3-none-any.whl → 1.0.2.1__py3-none-any.whl

bella_openapi/bella_trace/record_log.py

@@ -1,58 +1,58 @@
-# -*- coding: utf-8 -*-
-# ======================
-# Date    : 2024/12/30
-# Author  : Liu Yuchen
-# Content : 
-# 
-# ======================
-import functools
-import json
-import logging
-import time
-import traceback
-
-from ._context import TraceContext
-
-__all__ = ["trace", "BellaTraceHandler"]
-
-
-def trace(logger=logging):
-    def decorator(func):
-        @functools.wraps(func)
-        def wrapper(*args, **kwargs):
-            data = {
-                "trace_info": {
-                    "bellaTraceId": TraceContext.trace_id,
-                    "serviceId": TraceContext.service_id,
-                    "start": time.time()
-                },
-                "function": func.__name__,
-                "args": args,
-                "kwargs": kwargs
-            }
-            try:
-                result = func(*args, **kwargs)
-                data["result"] = result
-                data["trace_info"]["end"] = time.time()
-                logger.info(json.dumps(data, ensure_ascii=False))
-                return result
-            except Exception as e:
-                try:
-                    data["error_msg"] = traceback.format_exception(e)
-                    logger.error(json.dumps(data, ensure_ascii=False))
-                except Exception as i_e:
-                    logger.error(traceback.format_exception(i_e))
-                raise e
-        return wrapper
-    return decorator
-
-
-class BellaTraceHandler(logging.Handler):
-    
-    def __init__(self, fmt: str = "{name}={value}"):
-        super().__init__()
-        self.format = fmt
-    
-    def emit(self, record):
-        if trace_id := TraceContext.trace_id:
-            record.msg = f"{self.format.format(name='trace_id', value=trace_id)} {record.msg}"
+# -*- coding: utf-8 -*-
+# ======================
+# Date    : 2024/12/30
+# Author  : Liu Yuchen
+# Content : 
+# 
+# ======================
+import functools
+import json
+import logging
+import time
+import traceback
+
+from ._context import TraceContext
+
+__all__ = ["trace", "BellaTraceHandler"]
+
+
+def trace(logger=logging):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            data = {
+                "trace_info": {
+                    "bellaTraceId": TraceContext.trace_id,
+                    "serviceId": TraceContext.service_id,
+                    "start": time.time()
+                },
+                "function": func.__name__,
+                "args": args,
+                "kwargs": kwargs
+            }
+            try:
+                result = func(*args, **kwargs)
+                data["result"] = result
+                data["trace_info"]["end"] = time.time()
+                logger.info(json.dumps(data, ensure_ascii=False))
+                return result
+            except Exception as e:
+                try:
+                    data["error_msg"] = traceback.format_exception(e)
+                    logger.error(json.dumps(data, ensure_ascii=False))
+                except Exception as i_e:
+                    logger.error(traceback.format_exception(i_e))
+                raise e
+        return wrapper
+    return decorator
+
+
+class BellaTraceHandler(logging.Handler):
+    
+    def __init__(self, fmt: str = "{name}={value}"):
+        super().__init__()
+        self.format = fmt
+    
+    def emit(self, record):
+        if trace_id := TraceContext.trace_id:
+            record.msg = f"{self.format.format(name='trace_id', value=trace_id)} {record.msg}"

bella_openapi/bella_trace/trace_requests.py

@@ -1,58 +1,58 @@
-# -*- coding: utf-8 -*-
-# ======================
-# Date    : 2024/12/30
-# Author  : Liu Yuchen
-# Content : 
-# 
-# ======================
-from requests import sessions
-from requests.adapters import HTTPAdapter
-
-from ._context import TraceContext
-
-
-__all__ = ["BellaTraceAdapter"]
-
-
-class BellaTraceAdapter(HTTPAdapter):
-    
-    def send(self, request, **kwargs):
-        request.headers.update(TraceContext.headers)
-        return super().send(request, **kwargs)
-
-
-def request(method, url, **kwargs):
-    with sessions.Session() as session:
-        session.mount("http://", BellaTraceAdapter())
-        session.mount("https://", BellaTraceAdapter())
-        return session.request(method=method, url=url, **kwargs)
-
-
-def get(url, params=None, **kwargs):
-    return request("get", url, params=params, **kwargs)
-
-
-def options(url, **kwargs):
-    return request("options", url, **kwargs)
-
-
-def head(url, **kwargs):
-    kwargs.setdefault("allow_redirects", False)
-    return request("head", url, **kwargs)
-
-
-def post(url, data=None, json=None, **kwargs):
-    return request("post", url, data=data, json=json, **kwargs)
-
-
-def put(url, data=None, **kwargs):
-    return request("put", url, data=data, **kwargs)
-
-
-def patch(url, data=None, **kwargs):
-    return request("patch", url, data=data, **kwargs)
-
-
-def delete(url, **kwargs):
-    return request("delete", url, **kwargs)
-
+# -*- coding: utf-8 -*-
+# ======================
+# Date    : 2024/12/30
+# Author  : Liu Yuchen
+# Content : 
+# 
+# ======================
+from requests import sessions
+from requests.adapters import HTTPAdapter
+
+from ._context import TraceContext
+
+
+__all__ = ["BellaTraceAdapter"]
+
+
+class BellaTraceAdapter(HTTPAdapter):
+    
+    def send(self, request, **kwargs):
+        request.headers.update(TraceContext.headers)
+        return super().send(request, **kwargs)
+
+
+def request(method, url, **kwargs):
+    with sessions.Session() as session:
+        session.mount("http://", BellaTraceAdapter())
+        session.mount("https://", BellaTraceAdapter())
+        return session.request(method=method, url=url, **kwargs)
+
+
+def get(url, params=None, **kwargs):
+    return request("get", url, params=params, **kwargs)
+
+
+def options(url, **kwargs):
+    return request("options", url, **kwargs)
+
+
+def head(url, **kwargs):
+    kwargs.setdefault("allow_redirects", False)
+    return request("head", url, **kwargs)
+
+
+def post(url, data=None, json=None, **kwargs):
+    return request("post", url, data=data, json=json, **kwargs)
+
+
+def put(url, data=None, **kwargs):
+    return request("put", url, data=data, **kwargs)
+
+
+def patch(url, data=None, **kwargs):
+    return request("patch", url, data=data, **kwargs)
+
+
+def delete(url, **kwargs):
+    return request("delete", url, **kwargs)
+

bella_openapi/config.py

@@ -1,15 +1,15 @@
-from typing import Optional
-import pydantic
-
-if pydantic.version.VERSION.startswith('1.'):
-    from pydantic import BaseSettings
-else:
-    from pydantic.v1 import BaseSettings
-
-
-class OpenAPIConfig(BaseSettings):
-    OPENAPI_HOST: Optional[str]
-    OPENAPI_CONSOLE_KEY: Optional[str]
-
-
-openapi_config = OpenAPIConfig()
+from typing import Optional
+import pydantic
+
+if pydantic.version.VERSION.startswith('1.'):
+    from pydantic import BaseSettings
+else:
+    from pydantic.v1 import BaseSettings
+
+
+class OpenAPIConfig(BaseSettings):
+    OPENAPI_HOST: Optional[str]
+    OPENAPI_CONSOLE_KEY: Optional[str]
+
+
+openapi_config = OpenAPIConfig()

bella_openapi/console/__init__.py

@@ -1,3 +1,3 @@
-from .models import get_model_list
-
+from .models import get_model_list
+
 __all__ = ["get_model_list"]

bella_openapi/console/models.py

@@ -1,45 +1,45 @@
-from typing import Optional, Dict, Any, Mapping
-
-import httpx
-
-from bella_openapi.config import openapi_config
-
-
-def get_model_list(
-        token: str = openapi_config.OPENAPI_CONSOLE_KEY,
-        extra_headers: Optional[Mapping[str, str]] = None,
-        extra_query: Optional[Mapping[str, object]] = None,
-) -> Dict[str, Any]:
-    """
-    获取模型列表
-
-    Args:
-        token (str): 访问令牌
-        extra_headers: 额外的请求头
-        extra_query: 额外的查询参数，可以包含status、features等
-
-    Returns:
-        Dict[str, Any]: API响应的JSON数据
-
-    Examples:
-        >>> get_model_list(token='****',extra_query={"status": "active", "features": "vision"})
-    """
-    url = openapi_config.OPENAPI_HOST + '/console/model/list'
-    headers = {
-        'Authorization': token,
-        'Content-Type': 'application/json',
-    }
-    # 添加额外的请求头
-    if extra_headers:
-        headers.update(extra_headers)
-    # 构建查询参数
-    params = {}
-    # 添加额外的查询参数
-    if extra_query:
-        params.update(extra_query)
-    # 发送请求
-    response = httpx.get(url, headers=headers, params=params)
-    # 检查响应状态
-    response.raise_for_status()
-    # 返回JSON数据
+from typing import Optional, Dict, Any, Mapping
+
+import httpx
+
+from bella_openapi.config import openapi_config
+
+
+def get_model_list(
+        token: str = openapi_config.OPENAPI_CONSOLE_KEY,
+        extra_headers: Optional[Mapping[str, str]] = None,
+        extra_query: Optional[Mapping[str, object]] = None,
+) -> Dict[str, Any]:
+    """
+    获取模型列表
+
+    Args:
+        token (str): 访问令牌
+        extra_headers: 额外的请求头
+        extra_query: 额外的查询参数，可以包含status、features等
+
+    Returns:
+        Dict[str, Any]: API响应的JSON数据
+
+    Examples:
+        >>> get_model_list(token='****',extra_query={"status": "active", "features": "vision"})
+    """
+    url = openapi_config.OPENAPI_HOST + '/console/model/list'
+    headers = {
+        'Authorization': token,
+        'Content-Type': 'application/json',
+    }
+    # 添加额外的请求头
+    if extra_headers:
+        headers.update(extra_headers)
+    # 构建查询参数
+    params = {}
+    # 添加额外的查询参数
+    if extra_query:
+        params.update(extra_query)
+    # 发送请求
+    response = httpx.get(url, headers=headers, params=params)
+    # 检查响应状态
+    response.raise_for_status()
+    # 返回JSON数据
     return response.json()

bella_openapi/exception.py

@@ -1,7 +1,7 @@
-class AuthorizationException(Exception):
-    def __init__(self, message, http_status_code=None):
-        self.message = message
-        self.http_stauts_code = http_status_code
-
-    def __repr__(self):
-        return f"AuthorizationError(message={self.message}, http_status_code={self.http_stauts_code})"
+class AuthorizationException(Exception):
+    def __init__(self, message, http_status_code=None):
+        self.message = message
+        self.http_stauts_code = http_status_code
+
+    def __repr__(self):
+        return f"AuthorizationError(message={self.message}, http_status_code={self.http_stauts_code})"