config.py.example

# src/config.py.example

class AIConfig:
    def __init__(self):
        # AI服务配置
        self.api_key = "YOUR_API_KEY_HERE"  # 替换为你的AI服务的API密钥
        self.api_endpoint = "https://api.siliconflow.cn/v1/chat/completions"  # AI服务的API端点 (如果需要，请修改)
        self.model_name = "Qwen/Qwen3-8B"  # 默认使用的AI模型 (如果需要，请修改)
        
        # 其他配置项
        self.max_tokens = 1000  # 每次请求的最大token数 
        self.temperature = 0.7  # AI回复的创造性程度（0-1）
        # 文件夹查询时的限制
        self.max_folder_files = 4000           # 单次查询文件的最大数量
        self.max_folder_json_size_mb = 1.0     # 单次查询序列化JSON最大大小 (MB)
        self.avg_bytes_per_token = 1.5         # 平均每个token对应的字节数，用于更加准确的token预估
        
    def update_config(self, **kwargs):
        """更新配置项
        
        :param kwargs: 配置项键值对，例如：
                      api_key="your-api-key"
                      api_endpoint="your-endpoint"
                      model_name="gpt-4"
                      max_tokens=2000
                      temperature=0.5
        """
        for key, value in kwargs.items():
            if hasattr(self, key):
                setattr(self, key, value)
            else:
                print(f"警告：未知的配置项 '{key}'")
    
    def is_configured(self):
        """检查是否已配置必要的设置"""
        return bool(self.api_key and self.api_key != "YOUR_API_KEY_HERE")

# 创建全局配置实例
ai_config = AIConfig()