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

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

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

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

{fiuai_sdk_python-0.2.7 → 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.7 → fiuai_sdk_python-0.3.0}/src/fiuai_sdk_python/client.py

@@ -4,10 +4,11 @@ import json
 from urllib.parse import quote
 from typing import Any, Literal
 
-from .util import create_headers, get_tokens, get_client_config, is_initialized
+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):
@@ -234,10 +247,14 @@ class FiuaiSDK(object):
 
         
         docs = self.post_process(res)
-        if isinstance(docs, list):
-            return docs[0]
+
+        if not docs:
+            return None
         else:
-            return docs
+            if isinstance(docs, list):
+                return docs[0]
+            else:
+                return docs
 
 
     def get_api(self, method, params={}):

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="用户当前公司信息")
+    
+    

{fiuai_sdk_python-0.2.7 → fiuai_sdk_python-0.3.0}/src/fiuai_sdk_python/setup.py

@@ -73,7 +73,7 @@ def load_language_data(client: FiuaiSDK)-> List[Language]:
     )
     return [Language(name=language["name"], language_name=language["language_name"]) for language in language_list]
 
-def load_doctype_meta(client: FiuaiSDK, doctype: str, max_api_retry: int = 3)-> DocTypeMeta:
+def load_doctype_meta(client: FiuaiSDK, doctype: str, max_api_retry: int = 3, only_has_prompt: bool = True, show_hidden: bool = False)-> DocTypeMeta:
     """
     从frappe获取doctype数据
     """
@@ -82,7 +82,7 @@ def load_doctype_meta(client: FiuaiSDK, doctype: str, max_api_retry: int = 3)->
     doc_meta = None
     while retry_count < max_api_retry:
             
-        doc_meta = _get_meta(client, doctype)
+        doc_meta = _get_meta(client, doctype, only_has_prompt, show_hidden)
         
         if doc_meta:
             break
@@ -91,7 +91,7 @@ def load_doctype_meta(client: FiuaiSDK, doctype: str, max_api_retry: int = 3)->
            
     return doc_meta        
 
-def _get_meta(client: FiuaiSDK, doctype: str) -> DocTypeMeta:
+def _get_meta(client: FiuaiSDK, doctype: str, only_has_prompt: bool = True, show_hidden: bool = False) -> DocTypeMeta:
     try:
         m = client.get_meta(doctype)
         
@@ -109,6 +109,7 @@ def _get_meta(client: FiuaiSDK, doctype: str) -> DocTypeMeta:
             hidden = _f.get("hidden", 0)
             reqd = _f.get("reqd", 0)
             read_only = _f.get("read_only", 0)
+            description = _f.get("description", "")
 
             options = []
             match field_type:
@@ -124,22 +125,26 @@ def _get_meta(client: FiuaiSDK, doctype: str) -> DocTypeMeta:
                     pass
 
 
-            if field_prompt == "":
+            if field_prompt == "" and only_has_prompt:
                 # 仅有prompt的field才是应该关心的字段，降低复杂度
                 continue
-            if hidden == 1:
+            if hidden == 1 and not show_hidden:
                 continue
             
             
             _fields.append(DocField(
                 fieldname=field_name, 
+                description=description,
                 fieldtype=field_type, 
                 hidden=True if _f.get("hidden", 0) == 1 else False,
                 read_only=True if _f.get("read_only", 0) == 1 else False,
                 reqd=True if _f.get("reqd", 0) == 1 else False,
                 options=options,
                 field_prompt=field_prompt,
-                ai_recognition_value=AiRecognitionValue(value=None, confidence=0, ai_comment="")
+                mandatory=True if _f.get("mandatory", 0) == 1 else False,
+                in_list_view=True if _f.get("in_list_view", 0) == 1 else False,
+                in_mobile_view=True if _f.get("in_mobile_view", 0) == 1 else False,
+                # ai_recognition_value=AiRecognitionValue(value=None, confidence=0, ai_comment="")
             ))
         
         # links, child 去重
@@ -148,23 +153,23 @@ def _get_meta(client: FiuaiSDK, doctype: str) -> DocTypeMeta:
 
         doc_meta = DocTypeMeta(
                 name=doctype, 
-                doctype_prompts=m["doctype_prompts"], 
+                doctype_prompts=m.get("doctype_prompts", ""), 
                 fields=_fields,
                 link_docs=links,
                 child_docs=child_docs
             )
         return doc_meta
     except Exception as e:
-        logger.error(f"load doctype meta failed: {e.args}")
+        logger.error(f"load doctype {doctype} meta failed: {e.args}")
         return None
         
         
-def load_all_allowed_doctype_meta(client: FiuaiSDK)-> List[DocTypeMeta]:
+def load_all_allowed_doctype_meta(client: FiuaiSDK, only_has_prompt: bool = True, show_hidden: bool = False)-> List[DocTypeMeta]:
     """
     从frappe获取doctype数据
     """
     r = []
     for _d in AI_ALLOWED_DOCTYPE_META:
-        m = load_doctype_meta(client, _d)
+        m = load_doctype_meta(client, _d, only_has_prompt, show_hidden)
         r.append(m)
     return r

{fiuai_sdk_python-0.2.7 → fiuai_sdk_python-0.3.0}/src/fiuai_sdk_python/type.py

@@ -124,12 +124,16 @@ class DocField(BaseModel):
     frappe 文档字段文档数据
     """
     fieldname: str = Field(description="字段名称,比如Invoice,Purchase Order,etc.")
+    description: str = Field(description="字段描述", default="")
     hidden: bool = Field(description="字段是否隐藏")
     reqd: bool = Field(description="字段是否必填")
     read_only: bool = Field(description="字段是否只读")
     fieldtype: str = Field(description="字段类型,比如Data,Float,Link,etc.")
     options: List[str] = Field(description="字段选项,Select类型的字段的枚举值, Link类型字段的Link目标表名")
     field_prompt: str = Field(description="字段对应的prompt")
+    mandatory: bool = Field(description="字段是否必填", default=False)
+    in_list_view: bool = Field(description="字段是否在列表视图中显示", default=False)
+    in_mobile_view: bool = Field(description="字段是否在移动端视图中显示", default=False)
     # ai_recognition_value: AiRecognitionValue = Field(description="字段对应的AI识别结果", default=None)
 
 
@@ -211,6 +215,18 @@ class DocTypeMeta(BaseModel):
                 r.append(i.model_dump_json())
         return r
     
+
+    ##### 获取单个field
+    def get_field(self, fieldname: str) -> DocField | None:
+        for field in self.fields:
+            if field.fieldname == fieldname:
+                return field
+        return None
+
+
+
+
+    #### create datas
     def create_empty_data_json(self, mock_type: Literal["full", "json"]="json", with_child_tables: bool = False) -> str:
         return json.dumps(self.create_empty_data(mock_type, with_child_tables), indent=4)
 
@@ -248,7 +264,7 @@ class DocTypeMeta(BaseModel):
                     if with_child_tables:
                         try:
                             # 动态导入避免循环依赖
-                            from pkg.fiuaiclient .setup import load_doctype_meta
+                            from .setup import load_doctype_meta
                             # 自动加载子表元数据
                             child_meta = load_doctype_meta(child_doctype)
                             if child_meta: