jarvis-ai-assistant 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

jarvis/jarvis_git_utils/git_commiter.py

@@ -1,5 +1,4 @@
 # -*- coding: utf-8 -*-
-import argparse
 import os
 import re
 import subprocess
@@ -7,6 +6,7 @@ import sys
 import tempfile
 from typing import Any, Dict, Optional
 
+import typer
 import yaml  # type: ignore
 
 from jarvis.jarvis_platform.registry import PlatformRegistry
@@ -21,6 +21,8 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
 from jarvis.jarvis_utils.utils import init_env, is_context_overflow
 
+app = typer.Typer(help="Git commit tool")
+
 
 class GitCommitTool:
     name = "git_commit_agent"
@@ -305,34 +307,37 @@ commit信息
             os.chdir(original_dir)
 
 
-def main():
-    init_env("欢迎使用 Jarvis-GitCommitTool，您的Git提交助手已准备就绪！")
-    parser = argparse.ArgumentParser(description="Git commit tool")
-    parser.add_argument(
-        "--root-dir", type=str, default=".", help="Root directory of the Git repository"
-    )
-    parser.add_argument(
+@app.command()
+def cli(
+    root_dir: str = typer.Option(
+        ".", "--root-dir", help="Root directory of the Git repository"
+    ),
+    prefix: str = typer.Option(
+        "",
         "--prefix",
-        type=str,
-        default="",
         help="Prefix to prepend to commit message (separated by space)",
-    )
-    parser.add_argument(
+    ),
+    suffix: str = typer.Option(
+        "",
         "--suffix",
-        type=str,
-        default="",
         help="Suffix to append to commit message (separated by newline)",
-    )
-    args = parser.parse_args()
+    ),
+):
+    init_env("欢迎使用 Jarvis-GitCommitTool，您的Git提交助手已准备就绪！")
     tool = GitCommitTool()
     tool.execute(
         {
-            "root_dir": args.root_dir,
-            "prefix": args.prefix if hasattr(args, "prefix") else "",
-            "suffix": args.suffix if hasattr(args, "suffix") else "",
+            "root_dir": root_dir,
+            "prefix": prefix,
+            "suffix": suffix,
         }
     )
 
 
+def main():
+    """Application entry point"""
+    app()
+
+
 if __name__ == "__main__":
-    sys.exit(main())
+    main()

jarvis/jarvis_methodology/main.py

@@ -8,11 +8,13 @@
 - 列出所有方法论
 """
 
-import argparse
 import hashlib
 import json
 import os
+import sys
+from typing import Optional
 
+import typer
 import yaml  # type: ignore
 
 from jarvis.jarvis_platform.registry import PlatformRegistry
@@ -22,8 +24,13 @@ from jarvis.jarvis_utils.methodology import (
 )
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
+app = typer.Typer(help="方法论管理工具")
 
-def import_methodology(input_file):
+
+@app.command("import")
+def import_methodology(
+    input_file: str = typer.Argument(..., help="要导入的方法论文件路径")
+):
     """导入方法论文件（合并策略）"""
     try:
         # 加载现有方法论
@@ -56,9 +63,13 @@ def import_methodology(input_file):
         )
     except (json.JSONDecodeError, OSError) as e:
         PrettyOutput.print(f"导入失败: {str(e)}", OutputType.ERROR)
+        raise typer.Exit(code=1)
 
 
-def export_methodology(output_file):
+@app.command("export")
+def export_methodology(
+    output_file: str = typer.Argument(..., help="导出文件路径")
+):
     """导出当前方法论到单个文件"""
     try:
         methodologies = _load_all_methodologies()
@@ -72,8 +83,10 @@ def export_methodology(output_file):
         )
     except (OSError, TypeError) as e:
         PrettyOutput.print(f"导出失败: {str(e)}", OutputType.ERROR)
+        raise typer.Exit(code=1)
 
 
+@app.command("list")
 def list_methodologies():
     """列出所有方法论"""
     try:
@@ -88,9 +101,13 @@ def list_methodologies():
             PrettyOutput.print(f"{i}. {problem_type}", OutputType.INFO)
     except (OSError, json.JSONDecodeError) as e:
         PrettyOutput.print(f"列出方法论失败: {str(e)}", OutputType.ERROR)
+        raise typer.Exit(code=1)
 
 
-def extract_methodology(input_file):
+@app.command("extract")
+def extract_methodology(
+    input_file: str = typer.Argument(..., help="要提取方法论的文本文件路径")
+):
     """从文本文件中提取方法论"""
     try:
         # 读取文本文件内容
@@ -135,7 +152,7 @@ def extract_methodology(input_file):
         except Exception as e:
             print("❌ 提取失败")
             PrettyOutput.print(f"提取方法论失败: {str(e)}", OutputType.ERROR)
-            return
+            raise typer.Exit(code=1)
 
         # 提取YAML部分
         methodologies_start = response.find("<methodologies>") + len("<methodologies>")
@@ -145,7 +162,7 @@ def extract_methodology(input_file):
             PrettyOutput.print(
                 "大模型未返回有效的<methodologies>格式", OutputType.ERROR
             )
-            return
+            raise typer.Exit(code=1)
 
         yaml_content = response[methodologies_start:methodologies_end].strip()
 
@@ -157,7 +174,7 @@ def extract_methodology(input_file):
         except (yaml.YAMLError, KeyError, TypeError) as e:
             print("❌ YAML解析失败")
             PrettyOutput.print(f"YAML解析错误: {str(e)}", OutputType.ERROR)
-            return
+            raise typer.Exit(code=1)
 
         if not extracted_methodologies:
             print("❌ 未提取到有效方法论")
@@ -190,9 +207,11 @@ def extract_methodology(input_file):
         )
     except Exception as e:
         PrettyOutput.print(f"提取失败: {str(e)}", OutputType.ERROR)
+        raise typer.Exit(code=1)
 
 
-def extract_methodology_from_url(url):
+@app.command("extract-url")
+def extract_methodology_from_url(url: str = typer.Argument(..., help="要提取方法论的URL")):
     """从URL提取方法论"""
     try:
         # 获取平台实例
@@ -234,7 +253,7 @@ def extract_methodology_from_url(url):
         except Exception as e:
             print("❌ 提取失败")
             PrettyOutput.print(f"提取方法论失败: {str(e)}", OutputType.ERROR)
-            return
+            raise typer.Exit(code=1)
 
         # 提取YAML部分
         methodologies_start = response.find("<methodologies>") + len("<methodologies>")
@@ -244,7 +263,7 @@ def extract_methodology_from_url(url):
             PrettyOutput.print(
                 "大模型未返回有效的<methodologies>格式", OutputType.ERROR
             )
-            return
+            raise typer.Exit(code=1)
 
         yaml_content = response[methodologies_start:methodologies_end].strip()
 
@@ -256,7 +275,7 @@ def extract_methodology_from_url(url):
         except (yaml.YAMLError, KeyError, TypeError) as e:
             print("❌ YAML解析失败")
             PrettyOutput.print(f"YAML解析错误: {str(e)}", OutputType.ERROR)
-            return
+            raise typer.Exit(code=1)
 
         if not extracted_methodologies:
             print("❌ 未提取到有效方法论")
@@ -289,46 +308,12 @@ def extract_methodology_from_url(url):
         )
     except Exception as e:
         PrettyOutput.print(f"从URL提取失败: {str(e)}", OutputType.ERROR)
+        raise typer.Exit(code=1)
 
 
-def main():
-    """方法论管理工具主函数"""
-    parser = argparse.ArgumentParser(description="方法论管理工具")
-    subparsers = parser.add_subparsers(dest="command", required=True)
-
-    # import命令
-    import_parser = subparsers.add_parser("import", help="导入方法论文件（合并策略）")
-    import_parser.add_argument("input_file", type=str, help="要导入的方法论文件路径")
-
-    # export命令
-    export_parser = subparsers.add_parser("export", help="导出当前方法论到单个文件")
-    export_parser.add_argument("output_file", type=str, help="导出文件路径")
-
-    # list命令
-    subparsers.add_parser("list", help="列出所有方法论")
-
-    # extract命令
-    extract_parser = subparsers.add_parser("extract", help="从文本文件中提取方法论")
-    extract_parser.add_argument(
-        "input_file", type=str, help="要提取方法论的文本文件路径"
-    )
-
-    # extract-url命令
-    extract_url_parser = subparsers.add_parser("extract-url", help="从URL提取方法论")
-    extract_url_parser.add_argument("url", type=str, help="要提取方法论的URL")
-
-    args = parser.parse_args()
-
-    if args.command == "import":
-        import_methodology(args.input_file)
-    elif args.command == "export":
-        export_methodology(args.output_file)
-    elif args.command == "list":
-        list_methodologies()
-    elif args.command == "extract":
-        extract_methodology(args.input_file)
-    elif args.command == "extract-url":
-        extract_methodology_from_url(args.url)
+def main() -> None:
+    """Application entry point"""
+    app()
 
 
 if __name__ == "__main__":

jarvis/jarvis_multi_agent/main.py

@@ -1,27 +1,26 @@
 # -*- coding: utf-8 -*-
+from typing import Optional
+
+import typer
 import yaml
 
 from jarvis.jarvis_multi_agent import MultiAgent
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.utils import init_env
 
+app = typer.Typer(help="多智能体系统启动器")
 
-def main():
-    """从YAML配置文件初始化并运行多智能体系统
 
-    Returns:
-        最终处理结果
-    """
+@app.command()
+def cli(
+    config: str = typer.Option(..., "--config", "-c", help="YAML配置文件路径"),
+    user_input: Optional[str] = typer.Option(None, "--input", "-i", help="用户输入（可选）"),
+):
+    """从YAML配置文件初始化并运行多智能体系统"""
     init_env("欢迎使用 Jarvis-MultiAgent，您的多智能体系统已准备就绪！")
-    import argparse
-
-    parser = argparse.ArgumentParser(description="多智能体系统启动器")
-    parser.add_argument("--config", "-c", required=True, help="YAML配置文件路径")
-    parser.add_argument("--input", "-i", help="用户输入（可选）")
-    args = parser.parse_args()
 
     try:
-        with open(args.config, "r", errors="ignore") as f:
+        with open(config, "r", errors="ignore") as f:
             config_data = yaml.safe_load(f)
 
         # 获取agents配置
@@ -33,20 +32,26 @@ def main():
 
         # 创建并运行多智能体系统
         multi_agent = MultiAgent(agents_config, main_agent_name)
-        user_input = (
-            args.input
-            if args.input is not None
+        final_input = (
+            user_input
+            if user_input is not None
             else get_multiline_input("请输入内容（输入空行结束）：")
         )
-        if user_input == "":
-            return
-        return multi_agent.run(user_input)
+        if not final_input:
+            raise typer.Exit(code=0)
+        multi_agent.run(final_input)
+
+    except typer.Exit:
+        raise
+    except (ValueError, RuntimeError, yaml.YAMLError) as e:
+        PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
+        raise typer.Exit(code=1)
+
 
-    except yaml.YAMLError as e:
-        raise ValueError(f"YAML配置文件解析错误: {str(e)}")
-    except Exception as e:
-        raise RuntimeError(f"多智能体系统初始化失败: {str(e)}")
+def main() -> None:
+    """Application entry point."""
+    app()
 
 
 if __name__ == "__main__":
-    result = main()
+    main()

jarvis/jarvis_platform/ai8.py

@@ -32,15 +32,21 @@ class AI8Model(BasePlatform):
 
         self.headers = {
             "Authorization": self.token,
+            "sec-ch-ua-platform": '"Windows"',
+            "sec-ch-ua": '"Not)A;Brand";v="8", "Chromium";v="138", "Google Chrome";v="138"',
+            "sec-ch-ua-mobile": "?0",
             "Content-Type": "application/json",
             "Accept": "application/json, text/plain, */*",
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
-            "X-APP-VERSION": "2.3.0",
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36",
+            "X-APP-VERSION": "2.4.2",
             "Origin": self.BASE_URL,
             "Referer": f"{self.BASE_URL}/chat?_userMenuKey=chat",
             "Sec-Fetch-Site": "same-origin",
             "Sec-Fetch-Mode": "cors",
             "Sec-Fetch-Dest": "empty",
+            "Accept-Language": "zh-CN,zh;q=0.9",
+            "Accept-Encoding": "gzip, deflate, br, zstd",
+            "Connection": "keep-alive",
         }
 
         self.model_name = os.getenv("JARVIS_MODEL") or "deepseek-chat"
@@ -60,7 +66,14 @@ class AI8Model(BasePlatform):
             # 1. 创建会话
             response = while_success(
                 lambda: http.post(
-                    f"{self.BASE_URL}/api/chat/session", headers=self.headers, json={}
+                    f"{self.BASE_URL}/api/chat/session",
+                    headers=self.headers,
+                    json={
+                        "mcp": [],
+                        "model": self.model_name,
+                        "plugins": [],
+                        "rags": [],
+                    },
                 ),
                 sleep_time=5,
             )
@@ -83,6 +96,7 @@ class AI8Model(BasePlatform):
                 "plugins": [],
                 "localPlugins": None,
                 "useAppId": 0,
+                "temperature": 0,
             }
 
             response = while_success(
@@ -127,33 +141,30 @@ class AI8Model(BasePlatform):
                 "files": [],
             }
 
+            # 为流式请求构造专用的请求头，避免 'Accept' 和 'accept' 键冲突
+            stream_headers = self.headers.copy()
+            stream_headers["Accept"] = "text/event-stream"  # 添加流式专用的accept头
+
             # 使用stream_post进行流式请求
             response_stream = while_success(
                 lambda: http.stream_post(
                     f"{self.BASE_URL}/api/chat/completions",
-                    headers=self.headers,
+                    headers=stream_headers,
                     json=payload,
                 ),
                 sleep_time=5,
             )
 
             # 处理流式响应
-            for chunk in response_stream:
-                if chunk:
+            for line in response_stream:
+                if line and line.startswith("data: "):
                     try:
-                        line = chunk.decode("utf-8")
-                        if line.startswith("data: "):
-                            try:
-                                data = json.loads(line[6:])
-                                if data.get("type") == "string":
-                                    chunk_data = data.get("data", "")
-                                    if chunk_data:
-                                        yield chunk_data
-
-                            except json.JSONDecodeError:
-                                continue
-
-                    except UnicodeDecodeError:
+                        data = json.loads(line[6:])
+                        if data.get("type") == "string":
+                            chunk_data = data.get("data", "")
+                            if chunk_data:
+                                yield chunk_data
+                    except json.JSONDecodeError:
                         continue
 
             return None
@@ -239,9 +250,7 @@ class AI8Model(BasePlatform):
             PrettyOutput.print(f"会话文件未找到: {file_path}", OutputType.ERROR)
             return False
         except KeyError as e:
-            PrettyOutput.print(
-                f"恢复失败: 会话文件格式不正确，缺少键 {e}", OutputType.ERROR
-            )
+            PrettyOutput.print(f"恢复失败: 会话文件格式不正确，缺少键 {e}", OutputType.ERROR)
             return False
         except Exception as e:
             PrettyOutput.print(f"恢复会话失败: {str(e)}", OutputType.ERROR)

jarvis/jarvis_platform/kimi.py

@@ -156,39 +156,24 @@ class KimiModel(BasePlatform):
                 sleep_time=5,
             )
 
-            response_data = b""
-
             # 处理流式响应
-            for chunk in response_stream:
-                response_data += chunk
-
-                # 尝试解析SSE格式的数据
-                try:
-                    # 查找完整的数据行
-                    lines = response_data.decode("utf-8").split("\n")
-                    response_data = b""  # 重置缓冲区
-
-                    for line in lines:
-                        if not line.strip():
-                            continue
-
-                        # SSE格式的行通常以"data: "开头
-                        if line.startswith("data: "):
-                            try:
-                                data = json.loads(line[6:])
-                                if data.get("event") == "resp":
-                                    status = data.get("file_info", {}).get("status")
-                                    if status == "parsed":
-                                        return True
-                                    elif status == "failed":
-                                        return False
-                            except json.JSONDecodeError:
-                                continue
-
-                except UnicodeDecodeError:
-                    # 如果解码失败，继续累积数据
+            for line in response_stream:
+                if not line.strip():
                     continue
 
+                # SSE格式的行通常以"data: "开头
+                if line.startswith("data: "):
+                    try:
+                        data = json.loads(line[6:])
+                        if data.get("event") == "resp":
+                            status = data.get("file_info", {}).get("status")
+                            if status == "parsed":
+                                return True
+                            elif status == "failed":
+                                return False
+                    except json.JSONDecodeError:
+                        continue
+
             retry_count += 1
             time.sleep(1)
 
@@ -299,40 +284,25 @@ class KimiModel(BasePlatform):
                 sleep_time=5,
             )
 
-            response_data = b""
-
             # 处理流式响应
-            for chunk in response_stream:
-                response_data += chunk
-
-                # 尝试解析SSE格式的数据
-                try:
-                    # 查找完整的数据行
-                    lines = response_data.decode("utf-8").split("\n")
-                    response_data = b""  # 重置缓冲区
-
-                    for line in lines:
-                        if not line.strip():
-                            continue
-
-                        # SSE格式的行通常以"data: "开头
-                        if line.startswith("data: "):
-                            try:
-                                data = json.loads(line[6:])
-                                event = data.get("event")
-
-                                if event == "cmpl":
-                                    # 处理补全文本
-                                    text = data.get("text", "")
-                                    if text:
-                                        yield text
-                            except json.JSONDecodeError:
-                                continue
-
-                except UnicodeDecodeError:
-                    # 如果解码失败，继续累积数据
+            for line in response_stream:
+                if not line.strip():
                     continue
 
+                # SSE格式的行通常以"data: "开头
+                if line.startswith("data: "):
+                    try:
+                        data = json.loads(line[6:])
+                        event = data.get("event")
+
+                        if event == "cmpl":
+                            # 处理补全文本
+                            text = data.get("text", "")
+                            if text:
+                                yield text
+                    except json.JSONDecodeError:
+                        continue
+
             return None
 
         except Exception as e: