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

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

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

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

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.9}/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.9}/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.9}/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.9}/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,

{fiuai_sdk_python-0.6.7 → fiuai_sdk_python-0.6.9}/src/fiuai_sdk_python/utils/logger.py

@@ -8,49 +8,78 @@
 import logging
 import os
 import sys
+from typing import Any, Callable, Dict, Optional
 
 
+def _make_context_filter(
+    context_injector: Optional[Callable[[], Optional[Dict[str, Any]]]],
+) -> logging.Filter:
+    """根据 context_injector 构建过滤器：非 None 时注入返回值到 record，否则仅设置默认 trace_id。"""
+
+    class _InjectorFilter(logging.Filter):
+        def __init__(self, injector: Optional[Callable[[], Optional[Dict[str, Any]]]]) -> None:
+            super().__init__()
+            self._injector = injector
+
+        def filter(self, record: logging.LogRecord) -> bool:
+            if self._injector is not None:
+                try:
+                    ctx = self._injector()
+                    if ctx:
+                        for k, v in ctx.items():
+                            setattr(record, k, v)
+                except Exception:
+                    pass
+            if not getattr(record, "trace_id", None):
+                record.trace_id = "-"
+            return True
+
+    return _InjectorFilter(context_injector)
 
 
 def init_logger(
     log_path: str = "logs/",
     log_level: str = "INFO",
     json_log: bool = False,
-    caller: bool = True
+    caller: bool = True,
+    context_injector: Optional[Callable[[], Optional[Dict[str, Any]]]] = None,
 ):
     """初始化日志配置
-    
+
     Args:
         log_path: 日志文件路径
         log_level: 日志级别
         json_log: 是否使用JSON格式日志
         caller: 是否显示调用者信息
+        context_injector: 可选，无参可调用对象，返回 dict（如 {"trace_id": "xxx"}），
+            非 None 时会将返回的键值注入到每条日志记录中。
     """
     # 创建日志目录
     if log_path and not os.path.exists(log_path):
         os.makedirs(log_path)
-    
+
     # 设置日志级别
     level = getattr(logging, log_level.upper(), logging.INFO)
-    
+
     # 获取根日志记录器
     root_logger = logging.getLogger()
-    
+
     # 设置根日志记录器的级别
     root_logger.setLevel(level)
-    
+
     # 清除现有的处理器（避免重复添加）
     root_logger.handlers.clear()
-    
+
+    context_filter = _make_context_filter(context_injector)
+    log_format = '%(asctime)s - %(name)s - %(levelname)s - [trace_id:%(trace_id)s] - %(message)s'
+
     # 配置控制台处理器
     console_handler = logging.StreamHandler(sys.stdout)
     console_handler.setLevel(level)
-    # 添加 event_id 和 task_id 到日志格式中
-    log_format = '%(asctime)s - %(name)s - %(levelname)s - [trace_id:%(trace_id)s] - %(message)s'
     console_handler.setFormatter(logging.Formatter(log_format))
-    # 添加上下文过滤器
+    console_handler.addFilter(context_filter)
     root_logger.addHandler(console_handler)
-    
+
     # 如果指定了日志文件路径，添加文件处理器
     if log_path:
         file_handler = logging.FileHandler(
@@ -58,13 +87,11 @@ def init_logger(
             encoding='utf-8'
         )
         file_handler.setLevel(level)
-        # 添加 event_id 和 task_id 到日志格式中
         file_handler.setFormatter(logging.Formatter(log_format))
-        # 添加上下文过滤器
+        file_handler.addFilter(context_filter)
         root_logger.addHandler(file_handler)
-    
+
     # 过滤第三方库的日志，只显示 ERROR 及以上级别
-    # 这样可以减少不必要的 INFO、DEBUG 和 WARNING 日志输出
     third_party_loggers = [
         'nebula3.logger',
         'nebula3',
@@ -77,21 +104,13 @@ def init_logger(
         third_party_logger.setLevel(logging.ERROR)
 
 
-# 全局过滤器实例
-
-
 def get_logger(name: str) -> logging.Logger:
     """获取日志记录器
-    
-    自动添加 ContextFilter，从上下文中获取 event_id 和 task_id 并记录到日志中
-    
+
     Args:
         name: 日志记录器名称
-        
+
     Returns:
         logging.Logger: 日志记录器实例
     """
-    logger = logging.getLogger(name)
-    
-    
-    return logger
+    return logging.getLogger(name)