fiuai-sdk-python 0.6.1__py3-none-any.whl → 0.6.4__py3-none-any.whl

fiuai_sdk_python/config.py

@@ -0,0 +1,38 @@
+# -- coding: utf-8 --
+# Project: fiuai_sdk_python
+# Created Date: 2026 01 Su
+# Author: liming
+# Email: lmlala@aliyun.com
+# Copyright (c) 2026 FiuAI
+
+
+"""各类枚举值"""
+
+from pydantic import BaseModel, Field
+from enum import Enum, StrEnum
+
+
+class CompanyNetworkConnectionType(StrEnum):
+    """公司网络连接类型"""
+    Customer = "Customer"
+    Supplier = "Supplier"
+
+
+class CompanySize(StrEnum):
+    """公司规模"""
+    One = "1"
+    TwoToFifteen = "2-15"
+    SixteenToFortyNine = "16-49"
+    FiftyToNinetyNine = "50-99"
+    MoreThanOneHundred = "100-249"
+    MoreThanTwoHundredFifty = "250-499"
+    MoreThanFiveHundred = "500-999"
+    MoreThanOneThousand = "1000+"
+
+
+
+class CompanyEntityType(StrEnum):
+    """公司实体类型"""
+    Enterprise = "Enterprise"
+    Individual = "Individual"
+    Other = "Other"

fiuai_sdk_python/const.py

@@ -12,4 +12,6 @@ V2_GET_LIST_URI = "/document"
 ASSISTANT_USER = "assistant@fiuai.com"
 
 DEFAULT_TENANT_ID = "8888888888888888888"
-PASSIVE_TENANT_ID = "7777777777777777777"
+PASSIVE_TENANT_ID = "7777777777777777777"
+
+

fiuai_sdk_python/utils/ids.py

@@ -0,0 +1,226 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Project: FiuAI
+Created Date: 2024-03-21
+Author: liming
+Email: lmlala@aliyun.com
+Copyright (c) 2025 FiuAI
+"""
+
+import threading
+import time
+import os
+import hashlib
+from typing import Optional
+from snowflake import SnowflakeGenerator
+
+
+class SnowflakeIdGenerator:
+    """
+    雪花算法ID生成器
+    使用 snowflake 包实现，支持多进程环境
+    """
+    _instance: Optional['SnowflakeIdGenerator'] = None
+    _lock = threading.Lock()
+
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            with cls._lock:
+                if cls._instance is None:
+                    cls._instance = super().__new__(cls)
+        return cls._instance
+
+    def __init__(self, instance: int = None):
+        """
+        初始化雪花算法ID生成器
+        
+        Args:
+            instance: 实例ID (0-1023)，如果为None则自动生成
+        """
+        if hasattr(self, '_initialized'):
+            return
+            
+        # 在初始化时获取一次进程信息并缓存
+        self._process_id = os.getpid()
+        self._start_time = self._get_process_start_time()
+        self._object_id = id(self)
+            
+        # 自动生成实例ID，确保多进程环境下唯一性
+        if instance is None:
+            instance = self._generate_process_instance_id()
+        elif not 0 <= instance <= 1023:
+            raise ValueError("实例ID必须在0-1023之间")
+            
+        self.generator = SnowflakeGenerator(
+            instance=instance
+        )
+        self._last_timestamp = 0
+        self._instance_id = instance
+        self._initialized = True
+
+    def _get_process_start_time(self) -> float:
+        """
+        获取进程启动时间
+        
+        Returns:
+            float: 进程启动时间
+        """
+        try:
+            # 在Linux/macOS上使用更精确的时间
+            if hasattr(os, 'times'):
+                return os.times().elapsed
+            else:
+                return time.time()
+        except:
+            return time.time()
+
+    def _generate_process_instance_id(self) -> int:
+        """
+        基于进程信息生成唯一的实例ID
+        
+        Returns:
+            int: 0-1023之间的唯一实例ID
+        """
+        # 使用缓存的进程信息，避免重复调用
+        process_info = f"{self._process_id}_{self._start_time}_{self._object_id}"
+        hash_obj = hashlib.md5(process_info.encode())
+        hash_int = int(hash_obj.hexdigest()[:8], 16)
+        
+        # 确保在0-1023范围内
+        instance_id = hash_int % 1024
+        
+        return instance_id
+
+    def next_id(self) -> int:
+        """
+        生成下一个ID
+        
+        Returns:
+            int: 64位唯一ID
+            
+        Raises:
+            OverflowError: 当时间戳达到最大值时抛出
+            RuntimeError: 当检测到时间回退时抛出
+        """
+        try:
+            result = next(self.generator)
+            if result is None:
+                raise ValueError("Generator returned None")
+            
+            # 检查时间回退（简单的时间戳检查）
+            current_timestamp = int(time.time() * 1000)
+            if current_timestamp < self._last_timestamp:
+                raise RuntimeError(f"检测到时间回退: 当前时间 {current_timestamp} < 上次时间 {self._last_timestamp}")
+            
+            self._last_timestamp = current_timestamp
+            return result
+            
+        except StopIteration:
+            # 如果序列号达到最大值，重新初始化生成器
+            with self._lock:
+                self.generator = SnowflakeGenerator(instance=self._instance_id)
+                result = next(self.generator)
+                if result is None:
+                    raise ValueError("Generator returned None")
+                return result
+
+    @property
+    def instance_id(self) -> int:
+        """获取当前实例ID"""
+        return self._instance_id
+
+    @property
+    def process_id(self) -> int:
+        """获取当前进程ID"""
+        return self._process_id
+
+    @property
+    def start_time(self) -> float:
+        """获取进程启动时间"""
+        return self._start_time
+
+
+# 全局单例实例和锁
+_id_generator = None
+_id_generator_lock = threading.Lock()
+
+
+def gen_id() -> str:
+    """
+    获取下一个唯一ID
+    
+    Returns:
+        str: 64位唯一ID字符串
+        
+    Raises:
+        RuntimeError: 当ID生成失败时抛出
+    """
+    global _id_generator
+    
+    # 双重检查锁定模式，确保线程安全
+    if _id_generator is None:
+        with _id_generator_lock:
+            if _id_generator is None:
+                _id_generator = SnowflakeIdGenerator()
+    
+    try:
+        return str(_id_generator.next_id())
+    except Exception as e:
+        # 记录错误并重新抛出
+        raise RuntimeError(f"ID生成失败: {str(e)}") from e
+
+
+def get_instance_info() -> dict:
+    """
+    获取当前ID生成器的实例信息
+    
+    Returns:
+        dict: 包含进程ID和实例ID的信息
+    """
+    global _id_generator
+    
+    if _id_generator is None:
+        return {"process_id": os.getpid(), "instance_id": None, "status": "not_initialized"}
+    
+    return {
+        "process_id": _id_generator.process_id,
+        "instance_id": _id_generator.instance_id,
+        "start_time": _id_generator.start_time,
+        "status": "initialized"
+    }
+
+
+def reset_id_generator():
+    """
+    重置ID生成器（主要用于测试或异常恢复）
+    """
+    global _id_generator
+    with _id_generator_lock:
+        _id_generator = None
+
+
+def set_custom_instance_id(instance_id: int):
+    """
+    设置自定义实例ID（谨慎使用，确保多进程环境下唯一性）
+    
+    Args:
+        instance_id: 自定义实例ID (0-1023)
+        
+    Raises:
+        ValueError: 实例ID超出范围
+        RuntimeError: 生成器已初始化
+    """
+    global _id_generator
+    
+    if not 0 <= instance_id <= 1023:
+        raise ValueError("实例ID必须在0-1023之间")
+    
+    if _id_generator is not None:
+        raise RuntimeError("生成器已初始化，无法修改实例ID")
+    
+    with _id_generator_lock:
+        if _id_generator is None:
+            _id_generator = SnowflakeIdGenerator(instance=instance_id)
+        else:
+            raise RuntimeError("生成器已初始化，无法修改实例ID")

{fiuai_sdk_python-0.6.1.dist-info → fiuai_sdk_python-0.6.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai_sdk_python
-Version: 0.6.1
+Version: 0.6.4
 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
@@ -28,6 +28,7 @@ Requires-Dist: langgraph-checkpoint-redis>=0.3.1
 Requires-Dist: pydantic>=2.11.7
 Requires-Dist: pytest>=8.4.1
 Requires-Dist: redis>=7.1.0
+Requires-Dist: snowflake-id>=1.0.2
 Requires-Dist: tenacity>=9.1.2
 Description-Content-Type: text/markdown
 

{fiuai_sdk_python-0.6.1.dist-info → fiuai_sdk_python-0.6.4.dist-info}/RECORD

@@ -2,7 +2,8 @@ fiuai_sdk_python/__init__.py,sha256=TBQNujrI8Y7Weuvw3zih22CipKUd0nyfeVGOGySA5zA,
 fiuai_sdk_python/bank.py,sha256=GRAmkI33rRZw9UE67SpkYsqPN-kdLm-iuJuuBLxSvrE,5438
 fiuai_sdk_python/client.py,sha256=rDc0l4nhcv6sT_T_5lbcfkKbk4GT0WibStUH7C0YVtA,18917
 fiuai_sdk_python/company.py,sha256=ZELOAxCTIA-F9cUWakvq7hv6SflM19Rq3XMDkV0mtbg,4771
-fiuai_sdk_python/const.py,sha256=S1sXurb9yE1fS3mWYw0qYRJDwJ4UlmqqWwpatt5lh-0,300
+fiuai_sdk_python/config.py,sha256=vCD-aW2Y9OpJitfTDB3Acjw_PCurYT69vQjZ_wY950I,825
+fiuai_sdk_python/const.py,sha256=zxYOMpdkTOd-Xwr-ImnhlFItpQNydFQUdJQICJNA9HU,303
 fiuai_sdk_python/context.py,sha256=Gu77o_tQN9eEh42RYC9KoJQcJxlmi4i8LH50IFfFH4I,9319
 fiuai_sdk_python/datatype.py,sha256=WQjsoJgpkkw0GtAQLzTrsem4joT8e3cnhcfy3dKN0bA,8919
 fiuai_sdk_python/doctype.py,sha256=5o6WsZDeKuCWfTiQa0H_BNh4jb96EpZt9WaDh8TgKZo,5013
@@ -29,8 +30,9 @@ fiuai_sdk_python/pkg/db/utils.py,sha256=qqZ14eV7fy9x_mwcEDKJDz1W54xPndpJ1QfM_D0E
 fiuai_sdk_python/pkg/vector/__init__.py,sha256=fvbetFxbHiFeojnTfKT3roA1hOcoNHF_eCRuG6HwMpI,404
 fiuai_sdk_python/pkg/vector/vector.py,sha256=gC8CQTZm56zXbhM3g_hFa7SGtAo9D1atNs3HwuEbJHQ,28994
 fiuai_sdk_python/utils/__init__.py,sha256=UwwsvqBsaRCHbWdx-wvM48szT3j50h95k9MZdbfawRQ,72
+fiuai_sdk_python/utils/ids.py,sha256=ZDtEqt_Woth8ytPB2tdnnTIv7noWr8XYhSsUvkZ7Hc0,6448
 fiuai_sdk_python/utils/text.py,sha256=bnob_W0nj_Vj8Hp93B0cYmFOY8IhUWF0C8UedOYCNvs,1667
-fiuai_sdk_python-0.6.1.dist-info/METADATA,sha256=0FAbgudXGsl0ZG4ANaFLpGCEUcOZvPFev9O_o2xPbcE,1488
-fiuai_sdk_python-0.6.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-fiuai_sdk_python-0.6.1.dist-info/licenses/LICENSE,sha256=PFMF0dFErrBFqU-rryEby0yW8GBagYqrdbyZQHMUCJg,1062
-fiuai_sdk_python-0.6.1.dist-info/RECORD,,
+fiuai_sdk_python-0.6.4.dist-info/METADATA,sha256=ltSIEs8tgMsUwqCLYJAzSvQw5zgNwNMfLfqMHeDW1RI,1523
+fiuai_sdk_python-0.6.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+fiuai_sdk_python-0.6.4.dist-info/licenses/LICENSE,sha256=PFMF0dFErrBFqU-rryEby0yW8GBagYqrdbyZQHMUCJg,1062
+fiuai_sdk_python-0.6.4.dist-info/RECORD,,