fiuai-sdk-python 0.2.9__tar.gz → 0.3.0__tar.gz

{fiuai_sdk_python-0.2.9 → fiuai_sdk_python-0.3.0}/PKG-INFO

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

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

{fiuai_sdk_python-0.2.9 → fiuai_sdk_python-0.3.0}/src/fiuai_sdk_python/__init__.py

@@ -1,10 +1,14 @@
 from .client import FiuaiSDK, get_client
 from .token import TokenConfig
 from .util import init_fiuai
+from .profile import UserProfileInfo
+from .type import UserProfile
 
 __all__ = [
     'FiuaiSDK',
     'TokenConfig',
     'init_fiuai',
-    'get_client'
+    'get_client',
+    'UserProfileInfo',
+    'UserProfile'
     ]

{fiuai_sdk_python-0.2.9 → fiuai_sdk_python-0.3.0}/src/fiuai_sdk_python/client.py

@@ -8,6 +8,7 @@ from .util import get_tokens, get_client_config, is_initialized
 from .error import FiuaiGeneralError, FiuaiAuthError
 from logging import getLogger
 from .type import UserProfile
+from .profile import UserProfileInfo
 
 
 logger = getLogger(__name__)
@@ -140,6 +141,7 @@ class FiuaiSDK(object):
         return self.headers["Fiuai-Internal-Company"]
 
     def get_user_profile(self) -> UserProfile:
+        """获取简单的用户信息"""
         d = self.get_doc("User", self.username, fields=["name", "full_name", "email", "phone", "roles", "auth_tenant_id", "default_company_id"])
         return UserProfile(
             name=self.username,
@@ -151,6 +153,17 @@ class FiuaiSDK(object):
             default_company_id=d.get("default_company_id") or "",
         )
             
+
+    def get_user_profile_info(self, user_id: str) -> UserProfileInfo:
+        """获取详细的用户信息"""
+        d = self.get_v2_api(f"internal/user/profile/{user_id}")
+
+        if not d:
+            return None
+
+        return UserProfileInfo.model_validate(d)
+
+
     def get_list(self, doctype, fields='"*"', filters=None, limit_start=0, limit_page_length=0, order_by=None):
         '''Returns list of records of a particular type'''
         if not isinstance(fields, str):

fiuai_sdk_python-0.3.0/src/fiuai_sdk_python/profile.py

@@ -0,0 +1,95 @@
+# -- coding: utf-8 --
+# Project: fiuai_sdk_python
+# Created Date: 2025 09 Th
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2025 FiuAI
+
+
+# -- coding: utf-8 --
+# Project: response
+# Created Date: 2025 05 Tu
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2025 FiuAI
+
+
+from pydantic import BaseModel, Field
+from typing import List, Literal
+
+
+class UserCompanyInfo(BaseModel):
+    name: str = Field(description="公司id")
+    full_name: str = Field(description="公司名称")
+
+class UserBaseInfo(BaseModel):
+    name: str = Field(description="用户名id")
+    email: str = Field(description="邮箱")
+    first_name: str = Field(description="名")
+    full_name: str = Field(description="全名")
+    language: str = Field(description="语言")
+    time_zone: str = Field(description="时区")
+    auth_tenant_id: str = Field(description="租户id")
+    current_company: str = Field(description="当前公司")
+    available_companies: List[UserCompanyInfo] = Field(description="用户在租户下可以访问的公司列表", default_factory=list)
+
+
+class UserPermissionInfo(BaseModel):
+    can_select: List[str] = Field(description="可以查询的文档", default_factory=list)
+    can_read: List[str] = Field(description="可以读取的文档", default_factory=list)
+    can_write: List[str] = Field(description="可以写入的文档", default_factory=list)
+    can_create: List[str] = Field(description="可以创建的文档", default_factory=list)
+    can_delete: List[str] = Field(description="可以删除的文档", default_factory=list)
+    can_submit: List[str] = Field(description="可以提交的文档", default_factory=list)
+    can_cancel: List[str] = Field(description="可以取消的文档", default_factory=list)
+    can_search: List[str] = Field(description="可以搜索的文档", default_factory=list)
+
+    # def __init__(self, user: str):
+    #     super().__init__()
+    
+    # def can_select(self, doctype: str) -> bool:
+    #     return doctype in self.can_select
+    
+    # def can_read(self, doctype: str) -> bool:
+    #     return doctype in self.can_read
+    
+    # def can_write(self, doctype: str) -> bool:
+    #     return doctype in self.can_write
+    
+    # def can_create(self, doctype: str) -> bool:
+    #     return doctype in self.can_create
+    
+    # def can_delete(self, doctype: str) -> bool:
+    #     return doctype in self.can_delete
+    
+    # def can_submit(self, doctype: str) -> bool:
+    #     return doctype in self.can_submit
+    
+    # def can_cancel(self, doctype: str) -> bool:
+    #     return doctype in self.can_cancel
+    
+    # def can_search(self, doctype: str) -> bool:
+    #     return doctype in self.can_search
+    
+class UserTenantInfo(BaseModel):
+    name: str = Field(description="租户id")
+    tenant_name: str = Field(description="租户名称")
+
+
+class UserCurrentCompanyInfo(BaseModel):
+    name: str = Field(description="公司id")
+    full_name: str = Field(description="公司名称")
+    unique_no: str = Field(description="公司唯一编号")
+    country_region: str = Field(description="国家")
+    default_currency: str = Field(description="货币")
+    company_size: str = Field(description="公司规模")
+    entity_type: Literal["Enterprise", "Individual", "Other"] = Field(description="公司类型")
+    company_profile: str = Field(description="公司业务特点提示词", default="")
+    
+class UserProfileInfo(BaseModel):
+    user_base_info: UserBaseInfo = Field(description="用户基础信息")
+    user_permissions: UserPermissionInfo = Field(description="用户权限")
+    user_tenant_info: UserTenantInfo = Field(description="用户租户信息")
+    user_current_company_info: UserCurrentCompanyInfo = Field(description="用户当前公司信息")
+    
+