fiuai-sdk-python 0.6.6__py3-none-any.whl → 0.6.7__py3-none-any.whl

fiuai_sdk_python/auth/__init__.py

@@ -9,6 +9,15 @@ from .helper import (
     get_impersonation,
     is_impersonating
 )
+from .context_mgr import (
+    ContextManager,
+    init_context,
+    WorldData,
+    get_auth_data_from_context,
+    set_auth_data,
+    update_auth_data,
+    get_world_data,
+)
 
 __all__ = [
     "parse_auth_headers",
@@ -22,4 +31,11 @@ __all__ = [
     "get_company_unique_no",
     "get_impersonation",
     "is_impersonating",
+    "ContextManager",
+    "init_context",
+    "WorldData",
+    "get_auth_data_from_context",
+    "set_auth_data",
+    "update_auth_data",
+    "get_world_data",
 ]

fiuai_sdk_python/auth/context_mgr.py

@@ -0,0 +1,228 @@
+# -- coding: utf-8 --
+# Project: fiuai_sdk_python
+# Created Date: 2025-01-31
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2025 FiuAI
+
+"""
+上下文管理：ContextManager、init_context、WorldData，
+以及从当前上下文获取/设置认证数据的 get_auth_data_from_context、set_auth_data、update_auth_data、get_world_data。
+"""
+
+import uuid
+import logging
+from typing import Dict, Any, Optional
+
+from pydantic import BaseModel, Field
+
+from ..context import RequestContext, get_current_headers, is_current_context_valid
+from .type import AuthData
+from .header import parse_auth_headers
+
+logger = logging.getLogger(__name__)
+
+_current_context_manager: Optional["ContextManager"] = None
+
+
+class WorldData(BaseModel):
+    """World 上下文数据模型"""
+    event_id: Optional[str] = Field(default=None, description="事件ID")
+    task_id: Optional[str] = Field(default=None, description="任务ID")
+
+
+class ContextManager:
+    """
+    上下文管理器，用于在后台任务中初始化和管理请求上下文。
+    使用 context manager 模式，确保上下文在任务结束时自动清理。
+
+    注意：
+    - 上下文只在 `with` 语句块内生效
+    - 退出 `with` 块时，上下文会被自动清理
+    - 在异步环境中，上下文会在同一个异步任务中保持
+    """
+
+    def __init__(
+        self,
+        auth_data: Optional[AuthData] = None,
+        headers: Optional[Dict[str, Any]] = None,
+        world_data: Optional[WorldData] = None,
+        event_id: Optional[str] = None,
+        task_id: Optional[str] = None,
+    ):
+        if auth_data and headers:
+            raise ValueError("不能同时提供 auth_data 和 headers，请只提供其中一个")
+
+        if world_data:
+            self.world_data = world_data
+        elif event_id is not None or task_id is not None:
+            self.world_data = WorldData(event_id=event_id, task_id=task_id)
+        else:
+            self.world_data = WorldData()
+
+        if auth_data:
+            self._context = self._create_context_from_auth_data(auth_data)
+        elif headers:
+            self._context = RequestContext.from_dict(headers)
+        else:
+            raise ValueError("必须提供 auth_data 或 headers 之一")
+
+    def _create_context_from_auth_data(self, auth_data: AuthData) -> RequestContext:
+        """从认证数据创建请求上下文"""
+        headers = {
+            "x-fiuai-user": auth_data.user_id,
+            "x-fiuai-auth-tenant-id": auth_data.auth_tenant_id,
+            "x-fiuai-current-company": auth_data.current_company,
+            "x-fiuai-impersonation": auth_data.impersonation or "",
+            "x-fiuai-unique-no": auth_data.company_unique_no or auth_data.current_company,
+            "x-fiuai-trace-id": auth_data.trace_id or str(uuid.uuid4()),
+            "x-fiuai-client": auth_data.client or "unknown",
+            "x-fiuai-channel": (auth_data.channel or "default").lower(),
+            "x-fiuai-lang": "zh",
+            "accept-language": "zh",
+        }
+        return RequestContext.from_dict(headers)
+
+    def __enter__(self):
+        global _current_context_manager
+        _current_context_manager = self
+        self._context.__enter__()
+        logger.debug("Context initialized")
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self._context.__exit__(exc_type, exc_val, exc_tb)
+        global _current_context_manager
+        _current_context_manager = None
+        logger.debug("Context cleaned up")
+
+    def update_auth_data(self, auth_data: AuthData) -> None:
+        """更新上下文中的认证数据"""
+        new_context = self._create_context_from_auth_data(auth_data)
+        if hasattr(self._context, "_token") and self._context._token:
+            self._context.__exit__(None, None, None)
+        self._context = new_context
+        self._context.__enter__()
+        logger.debug("Context auth data updated")
+
+    def update_headers(self, headers: Dict[str, Any]) -> None:
+        """更新上下文中的请求头"""
+        new_context = RequestContext.from_dict(headers)
+        if hasattr(self._context, "_token") and self._context._token:
+            self._context.__exit__(None, None, None)
+        self._context = new_context
+        self._context.__enter__()
+        logger.debug("Context headers updated")
+
+
+def init_context(
+    auth_data: Optional[AuthData] = None,
+    headers: Optional[Dict[str, Any]] = None,
+    user_id: Optional[str] = None,
+    auth_tenant_id: Optional[str] = None,
+    current_company: Optional[str] = None,
+    world_data: Optional[WorldData] = None,
+    event_id: Optional[str] = None,
+    task_id: Optional[str] = None,
+    **kwargs: Any,
+) -> ContextManager:
+    """
+    初始化上下文，用于后台任务。返回的 ContextManager 必须在 `with` 语句中使用。
+
+    支持三种方式：1) auth_data 对象  2) headers 字典  3) user_id, auth_tenant_id, current_company。
+    """
+    if auth_data:
+        return ContextManager(auth_data=auth_data, world_data=world_data, event_id=event_id, task_id=task_id)
+    if headers:
+        return ContextManager(headers=headers, world_data=world_data, event_id=event_id, task_id=task_id)
+    if user_id and auth_tenant_id and current_company:
+        auth_data = AuthData(
+            user_id=user_id,
+            auth_tenant_id=auth_tenant_id,
+            current_company=current_company,
+            impersonation=kwargs.get("impersonation", ""),
+            company_unique_no=kwargs.get("company_unique_no", current_company),
+            trace_id=kwargs.get("trace_id", str(uuid.uuid4())),
+            client=kwargs.get("client", "unknown"),
+            channel=kwargs.get("channel", "default"),
+        )
+        return ContextManager(auth_data=auth_data, world_data=world_data, event_id=event_id, task_id=task_id)
+    raise ValueError(
+        "必须提供以下之一："
+        "1. auth_data 对象 "
+        "2. headers 字典 "
+        "3. user_id, auth_tenant_id, current_company 基本参数"
+    )
+
+
+def get_auth_data_from_context() -> Optional[AuthData]:
+    """从当前上下文中获取认证数据。上下文无效或不存在时返回 None。"""
+    try:
+        if not is_current_context_valid():
+            logger.warning("Current context is invalid, cannot get auth data")
+            return None
+        headers = get_current_headers()
+        if not headers:
+            logger.warning("Cannot get headers from current context")
+            return None
+        return parse_auth_headers(headers)
+    except Exception as e:
+        logger.warning("Failed to get auth data from context: %s", e)
+        return None
+
+
+def set_auth_data(auth_data: AuthData) -> bool:
+    """设置当前上下文中的认证数据。需在 init_context 的 with 块内使用。"""
+    global _current_context_manager
+    try:
+        if _current_context_manager is None:
+            logger.warning(
+                "ContextManager does not exist, cannot set auth data. "
+                "Please initialize context with init_context first"
+            )
+            return False
+        _current_context_manager.update_auth_data(auth_data)
+        return True
+    except Exception as e:
+        logger.warning("Failed to set auth data: %s", e)
+        return False
+
+
+def update_auth_data(
+    user_id: Optional[str] = None,
+    auth_tenant_id: Optional[str] = None,
+    current_company: Optional[str] = None,
+    **kwargs: Any,
+) -> bool:
+    """更新当前上下文中部分认证数据，未提供的字段保持不变。"""
+    try:
+        current = get_auth_data_from_context()
+        if not current:
+            logger.warning("Cannot get current auth data, please initialize context first")
+            return False
+        new_auth_data = AuthData(
+            user_id=user_id or current.user_id,
+            auth_tenant_id=auth_tenant_id or current.auth_tenant_id,
+            current_company=current_company or current.current_company,
+            impersonation=kwargs.get("impersonation", current.impersonation),
+            company_unique_no=kwargs.get("company_unique_no", current.company_unique_no),
+            trace_id=kwargs.get("trace_id", current.trace_id),
+            client=kwargs.get("client", current.client),
+            channel=kwargs.get("channel", current.channel),
+        )
+        return set_auth_data(new_auth_data)
+    except Exception as e:
+        logger.warning("Failed to update auth data: %s", e)
+        return False
+
+
+def get_world_data() -> Optional[WorldData]:
+    """从当前上下文中获取 World 上下文数据。"""
+    global _current_context_manager
+    try:
+        if _current_context_manager is None:
+            return None
+        return _current_context_manager.world_data
+    except Exception as e:
+        logger.warning("Failed to get world data: %s", e)
+        return None

fiuai_sdk_python/auth/header.py

@@ -6,7 +6,7 @@
 # Copyright (c) 2025 FiuAI
 
 from typing import Dict, Optional, Union, Literal
-from .type import AuthData
+from .type import AuthData, FIUAI_CHANNEL_VALUES
 from fastapi import Request
 
 
@@ -32,7 +32,9 @@ def parse_auth_headers(headers: Dict[str, str]) -> Optional[AuthData]:
         unique_no = headers.get("x-fiuai-unique-no", "")
         trace_id = headers.get("x-fiuai-trace-id", "")
         client = headers.get("x-fiuai-client", "unknown")
-        
+        channel_raw = (headers.get("x-fiuai-channel") or "").strip().lower()
+        channel = channel_raw if channel_raw in FIUAI_CHANNEL_VALUES else "default"
+
         # 验证必需字段
         if not user_id:
             raise ValueError("Missing required header: x-fiuai-user")
@@ -59,7 +61,8 @@ def parse_auth_headers(headers: Dict[str, str]) -> Optional[AuthData]:
             impersonation=impersonation,
             company_unique_no=unique_no,
             trace_id=trace_id,
-            client=client
+            client=client,
+            channel=channel,
         )
         
     except Exception as e:

fiuai_sdk_python/auth/type.py

@@ -7,7 +7,11 @@
 
 
 from pydantic import BaseModel, Field
-from typing import List
+from typing import List, Literal
+
+# 渠道枚举：wechat, web, default, app
+FiuaiChannel = Literal["wechat", "web", "default", "app"]
+FIUAI_CHANNEL_VALUES: tuple = ("wechat", "web", "default", "app")
 
 class AuthData(BaseModel):
     user_id: str = Field(description="用户ID")
@@ -17,6 +21,7 @@ class AuthData(BaseModel):
     company_unique_no: str = Field(description="当前公司唯一编号,正常情况等于current_company")
     trace_id: str = Field(description="追踪ID", default="")
     client: str = Field(description="客户端标识", default="unknown")
+    channel: str = Field(description="渠道: wechat, web, default, app", default="default")
 
 
 
@@ -32,7 +37,8 @@ class AuthHeader(BaseModel):
     x_fiuai_lang: str = Field(alias="x-fiuai-lang", default="zh", description="用户当前语言")
     x_fiuai_trace_id: str = Field(alias="x-fiuai-trace-id", default="", description="追踪ID")
     x_fiuai_client: str = Field(alias="x-fiuai-client", default="unknown", description="客户端标识")
+    x_fiuai_channel: str = Field(alias="x-fiuai-channel", default="default", description="渠道: wechat, web, default, app")
     accept_language: str = Field(alias="accept-language", default="zh", description="语言")
-    
+
     class Config:
         populate_by_name = True

fiuai_sdk_python/client.py

@@ -95,6 +95,7 @@ class FiuaiSDK(object):
                 "x-fiuai-unique-no": company_unique_no_value,
                 "x-fiuai-trace-id": context_headers.get("x-fiuai-trace-id", self.headers.x_fiuai_trace_id),
                 "x-fiuai-client": context_headers.get("x-fiuai-client", self.headers.x_fiuai_client),
+                "x-fiuai-channel": context_headers.get("x-fiuai-channel", self.headers.x_fiuai_channel),
                 "x-fiuai-lang": context_headers.get("x-fiuai-lang", self.headers.x_fiuai_lang),
                 "accept-language": context_headers.get("accept-language", self.headers.accept_language),
             })
@@ -114,6 +115,7 @@ class FiuaiSDK(object):
                 "x-fiuai-unique-no": company_unique_no_value,
                 "x-fiuai-trace-id": self.headers.x_fiuai_trace_id,
                 "x-fiuai-client": self.headers.x_fiuai_client,
+                "x-fiuai-channel": self.headers.x_fiuai_channel,
                 "x-fiuai-lang": self.headers.x_fiuai_lang,
                 "accept-language": self.headers.accept_language,
             }

fiuai_sdk_python/context.py

@@ -230,6 +230,7 @@ def extract_auth_headers_from_context(
     unique_no: str = "",
     trace_id: str = "",
     client: str = "",
+    channel: str = "default",
     lang: str = "zh",
     accept_language: str = "zh"
 ) -> AuthHeader:
@@ -244,6 +245,7 @@ def extract_auth_headers_from_context(
         unique_no: 公司唯一编号
         trace_id: 追踪ID
         client: 客户端类型（小程序、web、app等）
+        channel: 渠道 wechat/web/default/app，可为空默认 default
         lang: 语言
         accept_language: 接受的语言
         
@@ -263,17 +265,19 @@ def extract_auth_headers_from_context(
             x_fiuai_unique_no=unique_no or current_company,
             x_fiuai_trace_id=trace_id or str(uuid.uuid4()),
             x_fiuai_client=client,
+            x_fiuai_channel=channel or "default",
             x_fiuai_lang=lang,
             accept_language=accept_language
         )
-    
+
     # 从上下文中提取信息，优先使用上下文中的值
     context_trace_id = context_headers.get("x-fiuai-trace-id", "")
     context_unique_no = context_headers.get("x-fiuai-unique-no", "")
     context_client = context_headers.get("x-fiuai-client", "")
+    context_channel = context_headers.get("x-fiuai-channel", "") or "default"
     context_lang = context_headers.get("x-fiuai-lang", "")
     context_accept_language = context_headers.get("accept-language", "")
-    
+
     return AuthHeader(
         x_fiuai_user=username,
         x_fiuai_auth_tenant_id=auth_tenant_id,
@@ -282,6 +286,7 @@ def extract_auth_headers_from_context(
         x_fiuai_unique_no=context_unique_no or unique_no or current_company,
         x_fiuai_trace_id=context_trace_id or trace_id or str(uuid.uuid4()),
         x_fiuai_client=context_client or client,
+        x_fiuai_channel=context_channel or channel or "default",
         x_fiuai_lang=context_lang or lang,
         accept_language=context_accept_language or accept_language
     )

fiuai_sdk_python/http/__init__.py

@@ -0,0 +1,30 @@
+# -- coding: utf-8 --
+# Project: fiuai_sdk_python
+# Created Date: 2025-01-31
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2025 FiuAI
+
+from .client import (
+    get_async_http_client,
+    create_http_client,
+    get_sync_http_client,
+    create_sync_http_client,
+    close_global_clients,
+    extract_auth_headers,
+    auth_header_interceptor,
+    sync_auth_header_interceptor,
+    AUTH_HEADER_KEYS,
+)
+
+__all__ = [
+    "get_async_http_client",
+    "create_http_client",
+    "get_sync_http_client",
+    "create_sync_http_client",
+    "close_global_clients",
+    "extract_auth_headers",
+    "auth_header_interceptor",
+    "sync_auth_header_interceptor",
+    "AUTH_HEADER_KEYS",
+]

fiuai_sdk_python/http/client.py

@@ -0,0 +1,396 @@
+# -- coding: utf-8 --
+# Project: fiuai_sdk_python
+# Created Date: 2025-01-31
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2025 FiuAI
+
+"""
+HTTP 客户端与认证头拦截器：从当前上下文注入认证头，支持同步/异步、可选重试。
+"""
+
+import asyncio
+import logging
+import time
+from typing import Any, Dict, Optional, Union
+
+import httpx
+
+from ..context import get_current_headers, is_current_context_valid
+
+logger = logging.getLogger(__name__)
+
+# 需要传递的认证头列表（与 auth 目录下 AuthHeader 对齐）
+AUTH_HEADER_KEYS = [
+    "x-fiuai-user",
+    "x-fiuai-auth-tenant-id",
+    "x-fiuai-current-company",
+    "x-fiuai-impersonation",
+    "x-fiuai-unique-no",
+    "x-fiuai-trace-id",
+    "x-fiuai-client",
+    "x-fiuai-channel",
+    "x-fiuai-lang",
+    "accept-language",
+]
+
+
+def extract_auth_headers() -> Dict[str, str]:
+    """
+    从当前上下文中提取认证头信息。
+
+    Returns:
+        Dict[str, str]: 认证头字典，若上下文无效则返回空字典。
+    """
+    try:
+        if not is_current_context_valid():
+            logger.debug("current context is invalid, cannot extract auth headers")
+            return {}
+        headers = get_current_headers()
+        if not headers:
+            logger.debug("cannot get request headers from current context")
+            return {}
+        auth_headers = {}
+        for key in AUTH_HEADER_KEYS:
+            value = headers.get(key)
+            if value:
+                auth_headers[key] = value
+        return auth_headers
+    except Exception as e:
+        logger.warning("failed to extract auth headers from context: %s", e)
+        return {}
+
+
+async def auth_header_interceptor(request: httpx.Request) -> None:
+    """
+    异步认证头拦截器：为请求注入当前上下文的认证头，并在有 body 时设置默认 Content-Type。
+    """
+    auth_headers = extract_auth_headers()
+    if auth_headers:
+        for key, value in auth_headers.items():
+            request.headers[key] = value
+        logger.debug("added auth headers to request: %s", list(auth_headers.keys()))
+    else:
+        logger.debug("no auth headers found in context, skipping header injection")
+    if request.content and "content-type" not in request.headers:
+        request.headers["content-type"] = "application/json"
+        logger.debug("added default Content-Type: application/json")
+
+
+def sync_auth_header_interceptor(request: httpx.Request) -> None:
+    """
+    同步认证头拦截器：为请求注入当前上下文的认证头，并在有 body 时设置默认 Content-Type。
+    """
+    auth_headers = extract_auth_headers()
+    if auth_headers:
+        for key, value in auth_headers.items():
+            request.headers[key] = value
+        logger.debug("added auth headers to request: %s", list(auth_headers.keys()))
+    else:
+        logger.debug("no auth headers found in context, skipping header injection")
+    if request.content and "content-type" not in request.headers:
+        request.headers["content-type"] = "application/json"
+        logger.debug("added default Content-Type: application/json")
+
+
+class RetryableAsyncClient(httpx.AsyncClient):
+    """支持重试的异步 httpx 客户端。"""
+
+    def __init__(
+        self,
+        retry_count: int = 0,
+        retry_interval: float = 1.0,
+        *args: Any,
+        **kwargs: Any,
+    ):
+        super().__init__(*args, **kwargs)
+        self.retry_count = retry_count
+        self.retry_interval = retry_interval
+
+    async def _request_with_retry(
+        self,
+        method: str,
+        url: Union[httpx.URL, str],
+        *args: Any,
+        **kwargs: Any,
+    ) -> httpx.Response:
+        last_exception: Optional[Exception] = None
+        for attempt in range(self.retry_count + 1):
+            try:
+                response = await super().request(method, url, *args, **kwargs)
+                if response.status_code >= 500 and attempt < self.retry_count:
+                    logger.warning(
+                        "request failed with status %s, retrying (%s/%s)",
+                        response.status_code,
+                        attempt + 1,
+                        self.retry_count,
+                    )
+                    await response.aclose()
+                    await asyncio.sleep(self.retry_interval)
+                    continue
+                return response
+            except httpx.HTTPStatusError as e:
+                if e.response.status_code >= 500 and attempt < self.retry_count:
+                    logger.warning(
+                        "request failed with status %s, retrying (%s/%s)",
+                        e.response.status_code,
+                        attempt + 1,
+                        self.retry_count,
+                    )
+                    await e.response.aclose()
+                    await asyncio.sleep(self.retry_interval)
+                    continue
+                last_exception = e
+                logger.error("request failed with client error: %s", e)
+                raise
+            except httpx.RequestError as e:
+                last_exception = e
+                if attempt < self.retry_count:
+                    logger.warning(
+                        "request failed: %s, retrying (%s/%s)",
+                        e,
+                        attempt + 1,
+                        self.retry_count,
+                    )
+                    await asyncio.sleep(self.retry_interval)
+                else:
+                    logger.error("request failed after %s attempts: %s", self.retry_count + 1, e)
+                    raise
+        if last_exception:
+            raise last_exception
+        raise httpx.HTTPError("request failed")
+
+    async def get(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("GET", url, *args, **kwargs)
+
+    async def post(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("POST", url, *args, **kwargs)
+
+    async def put(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("PUT", url, *args, **kwargs)
+
+    async def delete(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("DELETE", url, *args, **kwargs)
+
+    async def patch(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("PATCH", url, *args, **kwargs)
+
+    async def head(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("HEAD", url, *args, **kwargs)
+
+    async def options(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return await self._request_with_retry("OPTIONS", url, *args, **kwargs)
+
+
+class RetryableSyncClient(httpx.Client):
+    """支持重试的同步 httpx 客户端。"""
+
+    def __init__(
+        self,
+        retry_count: int = 0,
+        retry_interval: float = 1.0,
+        *args: Any,
+        **kwargs: Any,
+    ):
+        super().__init__(*args, **kwargs)
+        self.retry_count = retry_count
+        self.retry_interval = retry_interval
+
+    def _request_with_retry(
+        self,
+        method: str,
+        url: Union[httpx.URL, str],
+        *args: Any,
+        **kwargs: Any,
+    ) -> httpx.Response:
+        last_exception: Optional[Exception] = None
+        for attempt in range(self.retry_count + 1):
+            try:
+                response = super().request(method, url, *args, **kwargs)
+                if response.status_code >= 500 and attempt < self.retry_count:
+                    logger.warning(
+                        "request failed with status %s, retrying (%s/%s)",
+                        response.status_code,
+                        attempt + 1,
+                        self.retry_count,
+                    )
+                    response.close()
+                    time.sleep(self.retry_interval)
+                    continue
+                return response
+            except httpx.HTTPStatusError as e:
+                if e.response.status_code >= 500 and attempt < self.retry_count:
+                    logger.warning(
+                        "request failed with status %s, retrying (%s/%s)",
+                        e.response.status_code,
+                        attempt + 1,
+                        self.retry_count,
+                    )
+                    e.response.close()
+                    time.sleep(self.retry_interval)
+                    continue
+                last_exception = e
+                logger.error("request failed with client error: %s", e)
+                raise
+            except httpx.RequestError as e:
+                last_exception = e
+                if attempt < self.retry_count:
+                    logger.warning(
+                        "request failed: %s, retrying (%s/%s)",
+                        e,
+                        attempt + 1,
+                        self.retry_count,
+                    )
+                    time.sleep(self.retry_interval)
+                else:
+                    logger.error("request failed after %s attempts: %s", self.retry_count + 1, e)
+                    raise
+        if last_exception:
+            raise last_exception
+        raise httpx.HTTPError("request failed")
+
+    def get(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("GET", url, *args, **kwargs)
+
+    def post(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("POST", url, *args, **kwargs)
+
+    def put(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("PUT", url, *args, **kwargs)
+
+    def delete(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("DELETE", url, *args, **kwargs)
+
+    def patch(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("PATCH", url, *args, **kwargs)
+
+    def head(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("HEAD", url, *args, **kwargs)
+
+    def options(self, url: Union[httpx.URL, str], *args: Any, **kwargs: Any) -> httpx.Response:
+        return self._request_with_retry("OPTIONS", url, *args, **kwargs)
+
+
+def create_http_client(
+    base_url: Optional[str] = None,
+    timeout: float = 30.0,
+    follow_redirects: bool = True,
+    verify: bool = True,
+    retry_count: int = 0,
+    retry_interval: float = 1.0,
+    **kwargs: Any,
+) -> httpx.AsyncClient:
+    """创建带认证头拦截器的 httpx 异步客户端。"""
+    client_kwargs: Dict[str, Any] = {
+        "timeout": timeout,
+        "follow_redirects": follow_redirects,
+        "verify": verify,
+        **kwargs,
+    }
+    if base_url is not None:
+        client_kwargs["base_url"] = base_url
+    if retry_count > 0:
+        return RetryableAsyncClient(
+            retry_count=retry_count,
+            retry_interval=retry_interval,
+            **client_kwargs,
+            event_hooks={"request": [auth_header_interceptor]},
+        )
+    return httpx.AsyncClient(
+        **client_kwargs,
+        event_hooks={"request": [auth_header_interceptor]},
+    )
+
+
+def create_sync_http_client(
+    base_url: Optional[str] = None,
+    timeout: float = 30.0,
+    follow_redirects: bool = True,
+    verify: bool = True,
+    retry_count: int = 0,
+    retry_interval: float = 1.0,
+    **kwargs: Any,
+) -> httpx.Client:
+    """创建带认证头拦截器的 httpx 同步客户端。"""
+    client_kwargs: Dict[str, Any] = {
+        "timeout": timeout,
+        "follow_redirects": follow_redirects,
+        "verify": verify,
+        **kwargs,
+    }
+    if base_url is not None:
+        client_kwargs["base_url"] = base_url
+    if retry_count > 0:
+        return RetryableSyncClient(
+            retry_count=retry_count,
+            retry_interval=retry_interval,
+            **client_kwargs,
+            event_hooks={"request": [sync_auth_header_interceptor]},
+        )
+    return httpx.Client(
+        **client_kwargs,
+        event_hooks={"request": [sync_auth_header_interceptor]},
+    )
+
+
+_global_async_client: Optional[httpx.AsyncClient] = None
+_global_sync_client: Optional[httpx.Client] = None
+
+
+def get_async_http_client(
+    base_url: Optional[str] = None,
+    timeout: float = 30.0,
+    follow_redirects: bool = True,
+    verify: bool = True,
+    retry_count: int = 0,
+    retry_interval: float = 1.0,
+    **kwargs: Any,
+) -> httpx.AsyncClient:
+    """获取全局 httpx 异步客户端（单例）。"""
+    global _global_async_client
+    if _global_async_client is None:
+        _global_async_client = create_http_client(
+            base_url=base_url,
+            timeout=timeout,
+            follow_redirects=follow_redirects,
+            verify=verify,
+            retry_count=retry_count,
+            retry_interval=retry_interval,
+            **kwargs,
+        )
+    return _global_async_client
+
+
+def get_sync_http_client(
+    base_url: Optional[str] = None,
+    timeout: float = 30.0,
+    follow_redirects: bool = True,
+    verify: bool = True,
+    retry_count: int = 0,
+    retry_interval: float = 1.0,
+    **kwargs: Any,
+) -> httpx.Client:
+    """获取全局 httpx 同步客户端（单例）。"""
+    global _global_sync_client
+    if _global_sync_client is None:
+        _global_sync_client = create_sync_http_client(
+            base_url=base_url,
+            timeout=timeout,
+            follow_redirects=follow_redirects,
+            verify=verify,
+            retry_count=retry_count,
+            retry_interval=retry_interval,
+            **kwargs,
+        )
+    return _global_sync_client
+
+
+def close_global_clients() -> None:
+    """关闭全局客户端。"""
+    global _global_async_client, _global_sync_client
+    if _global_async_client:
+        _global_async_client.close()
+        _global_async_client = None
+    if _global_sync_client:
+        _global_sync_client.close()
+        _global_sync_client = None

{fiuai_sdk_python-0.6.6.dist-info → fiuai_sdk_python-0.6.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai_sdk_python
-Version: 0.6.6
+Version: 0.6.7
 Summary: FiuAI Python SDK - 企业级AI服务集成开发工具包
 Project-URL: Homepage, https://github.com/fiuai/fiuai-sdk-python
 Project-URL: Documentation, https://github.com/fiuai/fiuai-sdk-python#readme

{fiuai_sdk_python-0.6.6.dist-info → fiuai_sdk_python-0.6.7.dist-info}/RECORD

@@ -1,10 +1,10 @@
 fiuai_sdk_python/__init__.py,sha256=WWAL1xTmdB8EKGatITf6wBKtCmbvBCDGhppOYt_8ZuU,581
 fiuai_sdk_python/bank.py,sha256=GRAmkI33rRZw9UE67SpkYsqPN-kdLm-iuJuuBLxSvrE,5438
-fiuai_sdk_python/client.py,sha256=rDc0l4nhcv6sT_T_5lbcfkKbk4GT0WibStUH7C0YVtA,18917
+fiuai_sdk_python/client.py,sha256=AGc8iqXaHOYUsAPDekV8dYZL6OFgERufR7FvAZscW8g,19087
 fiuai_sdk_python/company.py,sha256=ZELOAxCTIA-F9cUWakvq7hv6SflM19Rq3XMDkV0mtbg,4771
 fiuai_sdk_python/config.py,sha256=vCD-aW2Y9OpJitfTDB3Acjw_PCurYT69vQjZ_wY950I,825
 fiuai_sdk_python/const.py,sha256=fPoUeOJ4WPymEQqarA7cJE3bmdz5fCPq2PtYvzZJ4Tk,346
-fiuai_sdk_python/context.py,sha256=Gu77o_tQN9eEh42RYC9KoJQcJxlmi4i8LH50IFfFH4I,9319
+fiuai_sdk_python/context.py,sha256=uVUFvk9cjyZDzOaGjPK5j3f_426HDN_xBiQQ8GJCQ-s,9607
 fiuai_sdk_python/datatype.py,sha256=WQjsoJgpkkw0GtAQLzTrsem4joT8e3cnhcfy3dKN0bA,8919
 fiuai_sdk_python/doctype.py,sha256=5o6WsZDeKuCWfTiQa0H_BNh4jb96EpZt9WaDh8TgKZo,5013
 fiuai_sdk_python/error.py,sha256=YYsqP39vY8N7wWD4ervsx7ngcdXIMR59Wc4A4h4Rb-k,235
@@ -15,11 +15,14 @@ fiuai_sdk_python/resp.py,sha256=4twCxmwqe2e1vlhfFnu-5_FR8PwQiveB9njY3exaQ88,1018
 fiuai_sdk_python/setup.py,sha256=ER0IPAouHhrVSzG0Iu87Ky0R5c4kCgOF77kRAOO-1MI,8025
 fiuai_sdk_python/type.py,sha256=vinZKflNvmQNhqO5mDARAE6O133k0LiR1s1ZvexN_q4,28940
 fiuai_sdk_python/util.py,sha256=x3TkNsC8_nzA-8x6ndIGrIpE9sRKpn3vlxnj2Hqpxwo,2326
-fiuai_sdk_python/auth/__init__.py,sha256=MvI8FZ96-etsNGaHk1XKsQHXGEw2j9Qd2k8lnEAx97M,585
-fiuai_sdk_python/auth/header.py,sha256=vMK9a2AP09zntDc4EqoZ4YMBM7Uas-T61bv62gJzPNo,3401
+fiuai_sdk_python/auth/__init__.py,sha256=fcb8EwuKP341XJaYLlM8heMHLIctA6H_hbkakbUfrkg,920
+fiuai_sdk_python/auth/context_mgr.py,sha256=s5h80Xky1YIWiMKasH3IJ62niolWBLLaJ8CIsExvkY0,8871
+fiuai_sdk_python/auth/header.py,sha256=JoMT6OJQUU7laGP6235kolfG-Qy_AP6ZQQQ9Ca8U88c,3606
 fiuai_sdk_python/auth/helper.py,sha256=GeuRwOZivtIsAo4_XBt0PDDbOLntF2uxSrO9mPvVMBs,5014
-fiuai_sdk_python/auth/type.py,sha256=v3O6jlyNNek_praTCdIHhWBtQYgir13MLfEr0wJKCww,1841
+fiuai_sdk_python/auth/type.py,sha256=lPUgbSOW-mQfOZpxPGhwDmpbKr5UfjznEuSKGwDosxY,2233
 fiuai_sdk_python/examples/fastapi_integration.py,sha256=6VonD8xkpwUFh3qwQh6mdHGbITscX-MMJF2k3Gj2bJc,1701
+fiuai_sdk_python/http/__init__.py,sha256=RMtrf0O-iuAGMIqfpQg6sQHj_O9Lo4Lhn7KejcapZzo,692
+fiuai_sdk_python/http/client.py,sha256=omcM8R8NGSszZlKBPikZr-NJXHgwpNYu7qDa_9H1Jug,14321
 fiuai_sdk_python/pkg/cache/__init__.py,sha256=7mVRUKkAxCAHIWVZrIslel_kr0S5Je_l0I9Fy_iZjzI,112
 fiuai_sdk_python/pkg/cache/redis_manager.py,sha256=APSRRmsJKRWfDvzIJiwNMtIqjzbheOdULskj5mI55Fk,6114
 fiuai_sdk_python/pkg/db/__init__.py,sha256=IK-zw5tTiSpVMtT3zdVGMaqup08TACIWcEYWpe1htkc,709
@@ -33,7 +36,7 @@ fiuai_sdk_python/utils/__init__.py,sha256=UwwsvqBsaRCHbWdx-wvM48szT3j50h95k9MZdb
 fiuai_sdk_python/utils/ids.py,sha256=ZDtEqt_Woth8ytPB2tdnnTIv7noWr8XYhSsUvkZ7Hc0,6448
 fiuai_sdk_python/utils/logger.py,sha256=OcH8l7nQdLOtsZW5s4lBkd6kxPK7oIWF6KdGazQIzzc,2675
 fiuai_sdk_python/utils/text.py,sha256=bnob_W0nj_Vj8Hp93B0cYmFOY8IhUWF0C8UedOYCNvs,1667
-fiuai_sdk_python-0.6.6.dist-info/METADATA,sha256=6d_UtJCCx_dJ8BsIOrAmjoqf99tpmVfJCesCkHiTSw0,1523
-fiuai_sdk_python-0.6.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-fiuai_sdk_python-0.6.6.dist-info/licenses/LICENSE,sha256=PFMF0dFErrBFqU-rryEby0yW8GBagYqrdbyZQHMUCJg,1062
-fiuai_sdk_python-0.6.6.dist-info/RECORD,,
+fiuai_sdk_python-0.6.7.dist-info/METADATA,sha256=F-yN6GFq3Aa4BibEspDm0iDd5r5zZ8QupcP4JurTekA,1523
+fiuai_sdk_python-0.6.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+fiuai_sdk_python-0.6.7.dist-info/licenses/LICENSE,sha256=PFMF0dFErrBFqU-rryEby0yW8GBagYqrdbyZQHMUCJg,1062
+fiuai_sdk_python-0.6.7.dist-info/RECORD,,