jarvis-ai-assistant 0.1.80__tar.gz → 0.1.82__tar.gz

{jarvis_ai_assistant-0.1.80/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.80
+Version: 0.1.82
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.80"
+version = "0.1.82"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "Your Name", email = "your.email@example.com" }]

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.80",
+    version="0.1.82",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.80" 
+__version__ = "0.1.82" 

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/src/jarvis/agent.py

@@ -33,7 +33,7 @@ class Agent:
         
         # 从环境变量加载配置
         self.embedding_dimension = 1536  # Default for many embedding models
-        self.max_context_length = int(os.getenv('JARVIS_MAX_CONTEXT_LENGTH', '30720'))  # 默认30k
+        self.max_context_length = int(os.getenv('JARVIS_MAX_CONTEXT_LENGTH', '65536'))  # 默认30k
         
         # 初始化嵌入模型
         try:
@@ -320,12 +320,12 @@ class Agent:
             tools_prompt = ""
 
             # 选择工具
-            PrettyOutput.section("选择工具", OutputType.PLANNING)
+            PrettyOutput.section("可用工具", OutputType.PLANNING)
             tools = self.tool_registry.get_all_tools()
             if tools:
                 tools_prompt += "可用工具:\n"
                 for tool in tools:
-                    PrettyOutput.print(f"选择工具: {tool['name']}", OutputType.INFO)
+                    PrettyOutput.print(f"{tool['name']}: {tool['description']}", OutputType.INFO)
                     tools_prompt += f"- 名称: {tool['name']}\n"
                     tools_prompt += f"  描述: {tool['description']}\n"
                     tools_prompt += f"  参数: {tool['parameters']}\n"

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/src/jarvis/jarvis_codebase/main.py

@@ -19,6 +19,7 @@ class CodeBase:
         self.root_dir = root_dir
         os.chdir(self.root_dir)
         self.thread_count = int(os.environ.get("JARVIS_THREAD_COUNT") or 10)
+        self.max_context_length = int(os.getenv("JARVIS_MAX_CONTEXT_LENGTH", 65536))
             
         # 初始化数据目录
         self.data_dir = os.path.join(self.root_dir, ".jarvis-codebase")
@@ -450,7 +451,7 @@ class CodeBase:
         
         return initial_results
 
-    def search_similar(self, query: str, top_k: int = 20) -> List[Tuple[str, float, str]]:
+    def search_similar(self, query: str, top_k: int = 30) -> List[Tuple[str, float, str]]:
         """搜索相似文件"""
         model = PlatformRegistry.get_global_platform_registry().get_normal_platform()
         model.set_suppress_output(True)
@@ -515,7 +516,7 @@ class CodeBase:
 """
         for path, _, _ in results:
             try:
-                if len(prompt) > 30 * 1024:
+                if len(prompt) > self.max_context_length:
                     PrettyOutput.print(f"避免上下文超限，丢弃低相关度文件：{path}", OutputType.WARNING)
                     continue
                 content = open(path, "r", encoding="utf-8").read()

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/src/jarvis/jarvis_coder/main.py

@@ -22,6 +22,8 @@ class JarvisCoder:
     def __init__(self, root_dir: str, language: str):
         """初始化代码修改工具"""
 
+        self.max_context_length = int(os.getenv("JARVIS_MAX_CONTEXT_LENGTH", 65536))
+
 
         self.root_dir = find_git_root(root_dir)
         if not self.root_dir:
@@ -154,7 +156,7 @@ class JarvisCoder:
 文件列表如下：
 """
         for i, file in enumerate(related_files):
-            if len(prompt) > 30 * 1024:
+            if len(prompt) > self.max_context_length:
                 PrettyOutput.print(f'避免上下文超限，丢弃低相关度文件：{file["file_path"]}', OutputType.WARNING)
                 continue
             prompt += f"""{i}. {file["file_path"]}\n"""

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82}/src/jarvis/tools/registry.py

@@ -19,7 +19,7 @@ class ToolRegistry:
         # 加载内置工具和外部工具
         self._load_builtin_tools()
         self._load_external_tools()
-        self.max_context_length = int(os.getenv('JARVIS_MAX_CONTEXT_LENGTH', '30720'))  # 默认30k
+        self.max_context_length = int(os.getenv('JARVIS_MAX_CONTEXT_LENGTH', '65536'))  # 默认30k
 
     @staticmethod
     def get_global_tool_registry():
@@ -184,7 +184,7 @@ class ToolRegistry:
                         # 如果输出超过30k，只取最后30k字符
                         if len(output) > self.max_context_length:
                             output_to_summarize = output[-self.max_context_length:]
-                            truncation_notice = "\n(注意: 由于输出过长，仅总结最后30720字符)"
+                            truncation_notice = "\n(注意: 由于输出过长，仅总结最后65536字符)"
                         else:
                             output_to_summarize = output
                             truncation_notice = ""

{jarvis_ai_assistant-0.1.80 → jarvis_ai_assistant-0.1.82/src/jarvis_ai_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.80
+Version: 0.1.82
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire