jarvis-ai-assistant 0.1.132__py3-none-any.whl → 0.1.138__py3-none-any.whl

jarvis/jarvis_utils/config.py

@@ -12,66 +12,35 @@ import os
 def get_max_token_count() -> int:
     """
     获取模型允许的最大token数量。
-    
+
     返回:
         int: 模型能处理的最大token数量。
     """
-    return int(os.getenv('JARVIS_MAX_TOKEN_COUNT', '131072'))  # 默认128k
-    
+    return int(os.getenv('JARVIS_MAX_TOKEN_COUNT', '64000'))  # 默认64k
+
 def get_thread_count() -> int:
     """
     获取用于并行处理的线程数。
-    
+
     返回：
         int: 线程数，默认为1
     """
-    return int(os.getenv('JARVIS_THREAD_COUNT', '1'))  
-def dont_use_local_model() -> bool:
-    """
-    检查是否应避免使用本地模型。
-    
-    返回：
-        bool: 如果不使用本地模型则返回True，默认为False
-    """
-    return os.getenv('JARVIS_DONT_USE_LOCAL_MODEL', 'false') == 'true'
-    
+    return int(os.getenv('JARVIS_THREAD_COUNT', '1'))
+
 def is_auto_complete() -> bool:
     """
     检查是否启用了自动补全功能。
-    
+
     返回：
         bool: 如果启用了自动补全则返回True，默认为False
     """
     return os.getenv('JARVIS_AUTO_COMPLETE', 'false') == 'true'
-    
-def is_use_methodology() -> bool:
-    """
-    检查是否应使用方法论。
-    
-    返回：
-        bool: 如果使用方法论则返回True，默认为True
-    """
-    return os.getenv('JARVIS_USE_METHODOLOGY', 'true') == 'true'
-def is_record_methodology() -> bool:
-    """
-    检查是否应记录方法论。
-    
-    返回：
-        bool: 如果记录方法论则返回True，默认为True
-    """
-    return os.getenv('JARVIS_RECORD_METHODOLOGY', 'true') == 'true'
-def is_need_summary() -> bool:
-    """
-    检查是否需要生成摘要。
-    
-    返回：
-        bool: 如果需要摘要则返回True，默认为True
-    """
-    return os.getenv('JARVIS_NEED_SUMMARY', 'true') == 'true'
+
+
 def get_min_paragraph_length() -> int:
     """
     获取文本处理的最小段落长度。
-    
+
     返回：
         int: 最小字符长度，默认为50
     """
@@ -79,7 +48,7 @@ def get_min_paragraph_length() -> int:
 def get_max_paragraph_length() -> int:
     """
     获取文本处理的最大段落长度。
-    
+
     返回：
         int: 最大字符长度，默认为12800
     """
@@ -87,7 +56,7 @@ def get_max_paragraph_length() -> int:
 def get_shell_name() -> str:
     """
     获取系统shell名称。
-    
+
     返回：
         str: Shell名称（例如bash, zsh），默认为bash
     """
@@ -95,71 +64,42 @@ def get_shell_name() -> str:
 def get_normal_platform_name() -> str:
     """
     获取正常操作的平台名称。
-    
+
     返回：
-        str: 平台名称，默认为'kimi'
+        str: 平台名称，默认为'yuanbao'
     """
-    return os.getenv('JARVIS_PLATFORM', 'kimi')
+    return os.getenv('JARVIS_PLATFORM', 'yuanbao')
 def get_normal_model_name() -> str:
     """
     获取正常操作的模型名称。
-    
-    返回：
-        str: 模型名称，默认为'kimi'
-    """
-    return os.getenv('JARVIS_MODEL', 'kimi')
-def get_codegen_platform_name() -> str:
-    """
-    获取代码生成的平台名称。
-    
-    返回：
-        str: 平台名称，默认为'kimi'
-    """
-    return os.getenv('JARVIS_CODEGEN_PLATFORM', os.getenv('JARVIS_PLATFORM', 'kimi'))
-def get_codegen_model_name() -> str:
-    """
-    获取代码生成的模型名称。
-    
+
     返回：
-        str: 模型名称，默认为'kimi'
+        str: 模型名称，默认为'deep_seek'
     """
-    return os.getenv('JARVIS_CODEGEN_MODEL', os.getenv('JARVIS_MODEL', 'kimi'))
+    return os.getenv('JARVIS_MODEL', 'deep_seek_v3')
+
+
 def get_thinking_platform_name() -> str:
     """
     获取思考操作的平台名称。
-    
+
     返回：
-        str: 平台名称，默认为'kimi'
+        str: 平台名称，默认为'yuanbao'
     """
-    return os.getenv('JARVIS_THINKING_PLATFORM', os.getenv('JARVIS_PLATFORM', 'kimi'))
+    return os.getenv('JARVIS_THINKING_PLATFORM', os.getenv('JARVIS_PLATFORM', 'yuanbao'))
 def get_thinking_model_name() -> str:
     """
     获取思考操作的模型名称。
-    
-    返回：
-        str: 模型名称，默认为'kimi'
-    """
-    return os.getenv('JARVIS_THINKING_MODEL', os.getenv('JARVIS_MODEL', 'kimi'))
-def get_cheap_platform_name() -> str:
-    """
-    获取低成本操作的平台名称。
-    
-    返回：
-        str: 平台名称，默认为'kimi'
-    """
-    return os.getenv('JARVIS_CHEAP_PLATFORM', os.getenv('JARVIS_PLATFORM', 'kimi'))
-def get_cheap_model_name() -> str:
-    """
-    获取低成本操作的模型名称。
-    
+
     返回：
-        str: 模型名称，默认为'kimi'
+        str: 模型名称，默认为'deep_seek'
     """
-    return os.getenv('JARVIS_CHEAP_MODEL', os.getenv('JARVIS_MODEL', 'kimi'))
+    return os.getenv('JARVIS_THINKING_MODEL', os.getenv('JARVIS_MODEL', 'deep_seek'))
+
 def is_execute_tool_confirm() -> bool:
     """
     检查工具执行是否需要确认。
-    
+
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
@@ -167,7 +107,7 @@ def is_execute_tool_confirm() -> bool:
 def is_confirm_before_apply_patch() -> bool:
     """
     检查应用补丁前是否需要确认。
-    
+
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
@@ -176,18 +116,18 @@ def is_confirm_before_apply_patch() -> bool:
 def get_rag_ignored_paths() -> list:
     """
     获取RAG索引时需要忽略的路径列表。
-    
+
     首先尝试从.jarvis/rag_ignore.txt文件中读取，
     如果该文件不存在，则返回默认忽略列表。
-    
+
     返回：
         list: 忽略路径的列表，默认包含常见忽略路径
     """
     # 默认忽略路径
     default_ignored = [
-        '.git', 
-        '__pycache__', 
-        'node_modules', 
+        '.git',
+        '__pycache__',
+        'node_modules',
         '.jarvis',
         '.jarvis-*',
         'target',
@@ -224,7 +164,7 @@ def get_rag_ignored_paths() -> list:
         '*.xz',
         '*.rar'
     ]
-    
+
     # 尝试从配置文件中读取
     try:
         config_path = os.path.join('.jarvis', 'rag_ignore.txt')
@@ -234,5 +174,5 @@ def get_rag_ignored_paths() -> list:
                 return custom_ignored
     except Exception:
         pass
-        
+
     return default_ignored