fiuai-sdk-python 0.8.5__tar.gz → 0.8.7__tar.gz

{fiuai_sdk_python-0.8.5 → fiuai_sdk_python-0.8.7}/PKG-INFO

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

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

{fiuai_sdk_python-0.8.5 → fiuai_sdk_python-0.8.7}/src/fiuai_sdk_python/client.py

@@ -2,7 +2,7 @@ import httpx
 from tenacity import retry, stop_after_attempt, wait_exponential
 import json
 from urllib.parse import quote
-from typing import Any, Literal, List
+from typing import Any, Dict, List, Literal, Optional
 
 from .util import get_client_config, is_initialized
 from .error import FiuaiGeneralError, FiuaiAuthError
@@ -12,7 +12,6 @@ from .profile import UserProfileInfo
 from .auth import AuthHeader
 from .resp import parse_response, ApiResponse
 from .context import get_current_headers
-from typing import Dict, Any, Optional
 
 
 logger = getLogger(__name__)
@@ -298,6 +297,27 @@ class FiuaiSDK(object):
         res = self.client.get(self.url + '/api/v2/internal/doctype/get', params=d, headers=headers)
         return self.post_process(res)
 
+    def get_decrypted_password(
+        self,
+        doctype: str,
+        name: str,
+        fieldname: str = "password",
+        extra_headers: Optional[Dict[str, Any]] = None,
+    ) -> Optional[str]:
+        """从 Frappe __Auth 读取 Password 类型字段明文 (internal get_password)."""
+        params = {"doctype": doctype, "name": name, "fieldname": fieldname}
+        headers = self._get_merged_headers(extra_headers)
+        res = self.client.get(
+            self.url + "/api/v2/internal/doctype/get_password",
+            params=params,
+            headers=headers,
+        )
+        resp = self.post_process(res)
+        if resp.is_success() and resp.data and isinstance(resp.data, dict):
+            pw = resp.data.get("password")
+            return pw if isinstance(pw, str) else None
+        return None
+
     def internal_get_list(self, doctype, filters=None, fields=None, limit_start=0, limit_page_length=20, order_by=None,
                          extra_headers: Optional[Dict[str, Any]] = None) -> ApiResponse:
         d = {

{fiuai_sdk_python-0.8.5 → fiuai_sdk_python-0.8.7}/src/fiuai_sdk_python/profile.py

@@ -14,9 +14,70 @@
 # Copyright (c) 2025 FiuAI
 
 
-from pydantic import BaseModel, Field
-from typing import List, Literal
+from pydantic import BaseModel, Field, field_validator, model_validator
+from typing import List, Literal, Optional, Self
 from .utils.text import safe_string_name
+from enum import StrEnum
+
+class TaxPayerType(StrEnum):
+    GeneralTaxPayer = "general_tax_payer"
+    SmallTaxPayer = "small_tax_payer"
+    OtherTaxPayer = "other_tax_payer"
+
+
+PROVINCE_CODE = {
+    "China": [
+        ["1100", "北京市"],
+        ["1200", "天津市"],
+        ["1300", "河北省"],
+        ["1400", "山西省"],
+        ["1500", "内蒙古自治区"],
+        ["2100", "辽宁省"],
+        ["2200", "吉林省"],
+        ["2300", "黑龙江省"],
+        ["3100", "上海市"],
+        ["3200", "江苏省"],
+        ["3300", "浙江省"],
+        ["3400", "安徽省"],
+        ["3500", "福建省"],
+        ["3600", "江西省"],
+        ["3700", "山东省"],
+        ["4100", "河南省"],
+        ["4200", "湖北省"],
+        ["4300", "湖南省"],
+        ["4400", "广东省"],
+        ["4500", "广西壮族自治区"],
+        ["4600", "海南省"],
+        ["5000", "重庆市"],
+        ["5100", "四川省"],
+        ["5200", "贵州省"],
+        ["5300", "云南省"],
+        ["5400", "西藏自治区"],
+        ["6100", "陕西省"],
+        ["6200", "甘肃省"],
+        ["6300", "青海省"],
+        ["6400", "宁夏回族自治区"],
+        ["6500", "新疆维吾尔自治区"],
+        ["7100", "台湾省"],
+        ["8100", "香港特别行政区"],
+        ["8200", "澳门特别行政区"],
+    ],
+}
+
+_CHINA_PROVINCE_CODES: frozenset[str] = frozenset(
+    row[0] for row in PROVINCE_CODE["China"]
+)
+
+
+def _is_china_country_region(region: str) -> bool:
+    r = region.strip()
+    if not r:
+        return False
+    else:
+        if r.casefold() == "china":
+            return True
+        else:
+            return False
 
 
 class UserCompanyInfo(BaseModel):
@@ -85,8 +146,59 @@ class UserCurrentCompanyInfo(BaseModel):
     default_currency: str = Field(description="货币")
     company_size: str = Field(description="公司规模")
     entity_type: Literal["Enterprise", "Individual", "Other"] = Field(description="公司类型")
+    tax_payer_type: Optional[TaxPayerType] = Field(default=None, description="纳税人类型,一般纳税人,小规模纳税人,其他")
+    tax_source_location: Optional[str] = Field(default=None, description="税务登记地,中国特有配置，按省级行政区划分,比如江苏是3200")
     company_profile: str = Field(description="公司业务特点提示词", default="")
-    
+
+    @field_validator("tax_payer_type", mode="before")
+    @classmethod
+    def validate_tax_payer_type_when_present(cls, value: object) -> TaxPayerType | None:
+        if value is None or value == "":
+            return None
+        if isinstance(value, TaxPayerType):
+            return value
+        if isinstance(value, str):
+            try:
+                return TaxPayerType(value)
+            except ValueError:
+                allowed = ", ".join(repr(m.value) for m in TaxPayerType)
+                raise ValueError(
+                    f"tax_payer_type must be one of [{allowed}], got {value!r}"
+                ) from None
+        raise TypeError(
+            f"tax_payer_type must be str, TaxPayerType or None, got {type(value).__name__}"
+        )
+
+    @field_validator("tax_source_location", mode="before")
+    @classmethod
+    def normalize_tax_source_location(cls, value: object) -> str | None:
+        if value is None:
+            return None
+        if not isinstance(value, str):
+            raise TypeError(
+                "tax_source_location must be str or None, "
+                f"got {type(value).__name__}"
+            )
+        stripped = value.strip()
+        return None if stripped == "" else stripped
+
+    @model_validator(mode="after")
+    def validate_tax_source_location_against_china_provinces(self) -> Self:
+        loc = self.tax_source_location
+        china = _is_china_country_region(self.country_region)
+
+        if loc is not None and not china:
+            raise ValueError(
+                "tax_source_location is only applicable when country_region is China, CN or 中国"
+            )
+        if loc is not None and china and loc not in _CHINA_PROVINCE_CODES:
+            raise ValueError(
+                "tax_source_location must be a valid China province admin code "
+                f"(e.g. 3200 for Jiangsu), got {loc!r}"
+            )
+        return self
+
+
 class UserProfileInfo(BaseModel):
     user_base_info: UserBaseInfo = Field(description="用户基础信息")
     user_permissions: UserPermissionInfo = Field(description="用户权限")
@@ -123,6 +235,8 @@ class UserProfileInfo(BaseModel):
                 default_currency=self.user_current_company_info.default_currency,
                 company_size=self.user_current_company_info.company_size,
                 entity_type=self.user_current_company_info.entity_type,
+                tax_payer_type=self.user_current_company_info.tax_payer_type,
+                tax_source_location=self.user_current_company_info.tax_source_location,
                 company_profile=self.user_current_company_info.company_profile,
             ),
         )