fiuai-sdk-python 0.4.2__tar.gz → 0.4.3__tar.gz

{fiuai_sdk_python-0.4.2 → fiuai_sdk_python-0.4.3}/PKG-INFO

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

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

{fiuai_sdk_python-0.4.2 → fiuai_sdk_python-0.4.3}/src/fiuai_sdk_python/auth/header.py

@@ -31,6 +31,7 @@ def parse_auth_headers(headers: Dict[str, str]) -> Optional[AuthData]:
         impersonation = headers.get("x-fiuai-impersonation", "")
         unique_no = headers.get("x-fiuai-unique-no", "")
         trace_id = headers.get("x-fiuai-trace-id", "")
+        client = headers.get("x-fiuai-client", "unknown")
         
         # 验证必需字段
         if not user_id:
@@ -57,7 +58,8 @@ def parse_auth_headers(headers: Dict[str, str]) -> Optional[AuthData]:
             current_company=current_company[0],
             impersonation=impersonation,
             company_unique_no=unique_no,
-            trace_id=trace_id
+            trace_id=trace_id,
+            client=client
         )
         
     except Exception as e:

{fiuai_sdk_python-0.4.2 → fiuai_sdk_python-0.4.3}/src/fiuai_sdk_python/auth/helper.py

@@ -143,6 +143,24 @@ def get_impersonation(
     return auth_data.impersonation
 
 
+def get_current_client(
+    request: Union[Request, Dict[str, str]], 
+    engine: Literal["fastapi", "dict"] = "fastapi"
+) -> str:
+    """
+    获取当前客户端标识
+    
+    Args:
+        request: FastAPI Request 对象或包含认证数据的字典
+        engine: 请求引擎类型，默认为 "fastapi"
+        
+    Returns:
+        str: 客户端标识
+    """
+    auth_data = get_auth_data(request, engine)
+    return auth_data.client
+
+
 def is_impersonating(
     request: Union[Request, Dict[str, str]], 
     engine: Literal["fastapi", "dict"] = "fastapi"

{fiuai_sdk_python-0.4.2 → fiuai_sdk_python-0.4.3}/src/fiuai_sdk_python/auth/type.py

@@ -16,6 +16,7 @@ class AuthData(BaseModel):
     impersonation: str = Field(description="当前代表的租户ID", default="")
     company_unique_no: str = Field(description="当前公司唯一编号,正常情况等于current_company")
     trace_id: str = Field(description="追踪ID", default="")
+    client: str = Field(description="客户端标识", default="unknown")
 
 
 
@@ -30,6 +31,7 @@ class AuthHeader(BaseModel):
     x_fiuai_impersonation: str = Field(alias="x-fiuai-impersonation", default="", description="当前代表的租户ID")
     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="客户端标识")
     accept_language: str = Field(alias="accept-language", default="zh", description="语言")
     
     class Config:

{fiuai_sdk_python-0.4.2 → fiuai_sdk_python-0.4.3}/src/fiuai_sdk_python/client.py

@@ -94,6 +94,7 @@ class FiuaiSDK(object):
                 "x-fiuai-impersonation": context_headers.get("x-fiuai-impersonation", self.headers.x_fiuai_impersonation),
                 "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-lang": context_headers.get("x-fiuai-lang", self.headers.x_fiuai_lang),
                 "accept-language": context_headers.get("accept-language", self.headers.accept_language),
             })
@@ -112,6 +113,7 @@ class FiuaiSDK(object):
                 "x-fiuai-impersonation": self.headers.x_fiuai_impersonation,
                 "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-lang": self.headers.x_fiuai_lang,
                 "accept-language": self.headers.accept_language,
             }

{fiuai_sdk_python-0.4.2 → fiuai_sdk_python-0.4.3}/src/fiuai_sdk_python/context.py

@@ -100,9 +100,9 @@ class RequestContext:
                 # 生成新的 trace id
                 processed_headers["x-fiuai-trace-id"] = str(uuid.uuid4())
         
-        # 确保 X-Fiuai-Client 字段存在（如果不存在则设为空字符串）
+        # 确保 X-Fiuai-Client 字段存在（如果不存在则设为 "unknown"）
         if "x-fiuai-client" not in processed_headers:
-            processed_headers["x-fiuai-client"] = ""
+            processed_headers["x-fiuai-client"] = "unknown"
         
         return processed_headers