jarvis-ai-assistant 0.3.30__py3-none-any.whl → 0.7.6__py3-none-any.whl

jarvis/jarvis_multi_agent/main.py

@@ -3,11 +3,12 @@ from typing import Optional
 
 import typer
 import yaml  # type: ignore[import-untyped]
+import os
 
 from jarvis.jarvis_multi_agent import MultiAgent
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.utils import init_env
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+from jarvis.jarvis_utils.config import set_config
 
 app = typer.Typer(help="多智能体系统启动器")
 
@@ -18,9 +19,36 @@ def cli(
     user_input: Optional[str] = typer.Option(
         None, "--input", "-i", help="用户输入（可选）"
     ),
+    model_group: Optional[str] = typer.Option(
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
+    ),
+    non_interactive: bool = typer.Option(
+        False, "-n", "--non-interactive", help="启用非交互模式：用户无法与命令交互，脚本执行超时限制为5分钟"
+    ),
 ):
     """从YAML配置文件初始化并运行多智能体系统"""
-    init_env("欢迎使用 Jarvis-MultiAgent，您的多智能体系统已准备就绪！")
+    # CLI 标志：非交互模式（不依赖配置文件）
+    if non_interactive:
+        try:
+            os.environ["JARVIS_NON_INTERACTIVE"] = "true"
+        except Exception:
+            pass
+        # 注意：全局配置同步在 init_env 之后执行，避免被覆盖
+    # 非交互模式要求从命令行传入任务
+    if non_interactive and not (user_input and str(user_input).strip()):
+        print("❌ 非交互模式已启用：必须使用 --input 传入任务内容，因多行输入不可用。")
+        raise typer.Exit(code=2)
+    init_env()
+    
+    # 在初始化环境后同步 CLI 选项到全局配置，避免被 init_env 覆盖
+    try:
+        if non_interactive:
+            set_config("JARVIS_NON_INTERACTIVE", True)
+        if model_group:
+            set_config("JARVIS_LLM_GROUP", str(model_group))
+    except Exception:
+        # 静默忽略同步异常，不影响主流程
+        pass
 
     try:
         with open(config, "r", errors="ignore") as f:
@@ -34,7 +62,11 @@ def cli(
             raise ValueError("必须指定main_agent作为主智能体")
 
         # 创建并运行多智能体系统
-        multi_agent = MultiAgent(agents_config, main_agent_name)
+        multi_agent = MultiAgent(
+            agents_config,
+            main_agent_name,
+            common_system_prompt=str(config_data.get("common_system_prompt", "") or "")
+        )
         final_input = (
             user_input
             if user_input is not None
@@ -49,7 +81,7 @@ def cli(
     except typer.Exit:
         return
     except (ValueError, RuntimeError, yaml.YAMLError) as e:
-        PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
+        print(f"❌ 错误: {str(e)}")
         raise typer.Exit(code=1)