jarvis-ai-assistant 0.1.80__py3-none-any.whl → 0.1.82__py3-none-any.whl

jarvis/__init__.py

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

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/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/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/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.dist-info → jarvis_ai_assistant-0.1.82.dist-info}/METADATA

@@ -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.dist-info → jarvis_ai_assistant-0.1.82.dist-info}/RECORD

@@ -1,11 +1,11 @@
-jarvis/__init__.py,sha256=Z6ETkmEWC686kbdrf_q6ASeGGE04U-Rr-b2LXZPRVzE,50
-jarvis/agent.py,sha256=BDBFtvSbKfcUxd-pDvNv3ivfm0X1yCrjZDAkxmE1gwo,19373
+jarvis/__init__.py,sha256=k-UkaWVzxFbaPdj5vipYJrmuGDsEE8ESDzTjJnRduyA,50
+jarvis/agent.py,sha256=QkfW8fYdIFo54IgScP2sgS0NsDH-lWghtAjaYOhYQxs,19382
 jarvis/main.py,sha256=OJc7e5i988eQLByT7SzX7eoa0HKm4LMg814gZv9hBX8,5487
 jarvis/utils.py,sha256=jvo6ylvrTaSmXWcYY0qTTf14TwCkAhPsCUuIl5WHEuw,8640
 jarvis/jarvis_codebase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jarvis/jarvis_codebase/main.py,sha256=DPMPw8qoKHooydELKwf_bXPa0a3459kdFcZOXdrM-N8,24785
+jarvis/jarvis_codebase/main.py,sha256=z6HRTkUjAhRQclfHqj4yt3wq_wxiRlp0OCJsL_99NfI,24884
 jarvis/jarvis_coder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jarvis/jarvis_coder/main.py,sha256=Hr6T5LHC5n08s9tnOt-2_h8TmzothBUhqcd2-fzwuVI,23790
+jarvis/jarvis_coder/main.py,sha256=WZCjtlPDEvg2y6F6e4djc4xYWXM3eB_0pHrPpvlZEC4,23890
 jarvis/jarvis_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jarvis/jarvis_rag/main.py,sha256=drCH9KmwQozxtPICIfTFXmkIhwN8WDreMOCTxtj9SPc,20842
 jarvis/jarvis_smart_shell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,14 +26,14 @@ jarvis/tools/coder.py,sha256=kmotT2Klsug44S51QoSW9DzkxLzcF-XonyYAEoWZV6c,2295
 jarvis/tools/file_ops.py,sha256=h8g0eT9UvlJf4kt0DLXvdSsjcPj7x19lxWdDApeDfpg,3842
 jarvis/tools/generator.py,sha256=S1DhHBfhNvF6SrORnlk8Mz210yDiJPuxbfswbX_UACs,5791
 jarvis/tools/methodology.py,sha256=UG6s5VYRcd9wrKX4cg6f7zJhet5AIcthFGMOAdevBiw,5175
-jarvis/tools/registry.py,sha256=XhFufp57hJZkRj6MUC5m-q4WEHFp8hzGmW2zFIzk12w,9074
+jarvis/tools/registry.py,sha256=sgj5EVbRgfHSzXW5v-kbIZS_8cwAxTjHvSPAicxBRf4,9074
 jarvis/tools/search.py,sha256=c9dXtyICdl8Lm8shNPNyIx9k67uY0rMF8xnIKu2RsnE,8787
 jarvis/tools/shell.py,sha256=UPKshPyOaUwTngresUw-ot1jHjQIb4wCY5nkJqa38lU,2520
 jarvis/tools/sub_agent.py,sha256=rEtAmSVY2ZjFOZEKr5m5wpACOQIiM9Zr_3dT92FhXYU,2621
 jarvis/tools/webpage.py,sha256=d3w3Jcjcu1ESciezTkz3n3Zf-rp_l91PrVoDEZnckOo,2391
-jarvis_ai_assistant-0.1.80.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
-jarvis_ai_assistant-0.1.80.dist-info/METADATA,sha256=mfPbf9OcczggJFffilTxhlHqwDNYDsF9SOKDqstbQ-c,12736
-jarvis_ai_assistant-0.1.80.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-jarvis_ai_assistant-0.1.80.dist-info/entry_points.txt,sha256=sdmIO86MrIUepJTGyHs0i_Ho9VGf1q9YRP4RgQvGWcI,280
-jarvis_ai_assistant-0.1.80.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.1.80.dist-info/RECORD,,
+jarvis_ai_assistant-0.1.82.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.1.82.dist-info/METADATA,sha256=7q-f_Dj2WkdsTrResm6GqTFCIe7McMaZDYO3Rpr_2Ls,12736
+jarvis_ai_assistant-0.1.82.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jarvis_ai_assistant-0.1.82.dist-info/entry_points.txt,sha256=sdmIO86MrIUepJTGyHs0i_Ho9VGf1q9YRP4RgQvGWcI,280
+jarvis_ai_assistant-0.1.82.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.82.dist-info/RECORD,,