fiuai-sdk-python 0.6.7__tar.gz → 0.6.8__tar.gz

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai_sdk_python
-Version: 0.6.7
+Version: 0.6.8
 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.7 → fiuai_sdk_python-0.6.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fiuai_sdk_python"
-version = "0.6.7"
+version = "0.6.8"
 description = "FiuAI Python SDK - 企业级AI服务集成开发工具包"
 readme = "README.md"
 requires-python = ">=3.12"

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.8}/src/fiuai_sdk_python/__init__.py

@@ -3,11 +3,12 @@ from .util import init_fiuai
 from .profile import UserProfileInfo
 from .type import UserProfile
 from .context import (
-    RequestContext, 
+    RequestContext,
     get_current_headers,
     validate_current_context,
     get_current_missing_fields,
-    is_current_context_valid
+    is_current_context_valid,
+    get_trace_id,
 )
 
 __version__ = "0.6.5"
@@ -22,5 +23,6 @@ __all__ = [
     'get_current_headers',
     'validate_current_context',
     'get_current_missing_fields',
-    'is_current_context_valid'
-    ]
+    'is_current_context_valid',
+    'get_trace_id',
+]

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.8}/src/fiuai_sdk_python/auth/__init__.py

@@ -1,19 +1,20 @@
 from .header import parse_auth_headers, extract_auth_from_request
 from .type import AuthData, AuthHeader
 from .helper import (
-    get_auth_data,
+    get_auth_data as get_auth_data_from_request,
     get_current_user_id,
     get_current_tenant_id,
     get_current_company,
     get_company_unique_no,
     get_impersonation,
-    is_impersonating
+    is_impersonating,
 )
 from .context_mgr import (
     ContextManager,
     init_context,
     WorldData,
     get_auth_data_from_context,
+    get_auth_data,
     set_auth_data,
     update_auth_data,
     get_world_data,
@@ -24,7 +25,9 @@ __all__ = [
     "extract_auth_from_request",
     "AuthData",
     "AuthHeader",
+    "get_auth_data_from_request",
     "get_auth_data",
+    "get_auth_data_from_context",
     "get_current_user_id",
     "get_current_tenant_id",
     "get_current_company",
@@ -34,7 +37,6 @@ __all__ = [
     "ContextManager",
     "init_context",
     "WorldData",
-    "get_auth_data_from_context",
     "set_auth_data",
     "update_auth_data",
     "get_world_data",

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.8}/src/fiuai_sdk_python/auth/context_mgr.py

@@ -171,6 +171,15 @@ def get_auth_data_from_context() -> Optional[AuthData]:
         return None
 
 
+def get_auth_data() -> Optional[AuthData]:
+    """
+    从当前上下文获取认证数据（无参）。
+    与 get_auth_data_from_context() 行为一致，供业务方统一使用无参 get_auth_data()。
+    上下文无效或不存在时返回 None。
+    """
+    return get_auth_data_from_context()
+
+
 def set_auth_data(auth_data: AuthData) -> bool:
     """设置当前上下文中的认证数据。需在 init_context 的 with 块内使用。"""
     global _current_context_manager

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.8}/src/fiuai_sdk_python/context.py

@@ -222,6 +222,22 @@ def is_current_context_valid(required_fields: list = None) -> bool:
     return is_valid
 
 
+def get_trace_id(default: str = "") -> str:
+    """
+    从当前上下文获取 trace_id。
+
+    Args:
+        default: 无上下文或无 x-fiuai-trace-id 时的返回值，默认为 ""。
+
+    Returns:
+        str: x-fiuai-trace-id 的值，否则返回 default。
+    """
+    headers = get_current_headers()
+    if not headers:
+        return default
+    return headers.get("x-fiuai-trace-id") or default
+
+
 def extract_auth_headers_from_context(
     username: str,
     auth_tenant_id: str,