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

jarvis/jarvis_multi_agent/__init__.py

@@ -1,15 +1,11 @@
 # -*- coding: utf-8 -*-
+from jarvis.jarvis_utils.jsonnet_compat import loads as json_loads
 import re
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-import yaml
-
 from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_agent.output_handler import OutputHandler
 from jarvis.jarvis_tools.registry import ToolRegistry
-from jarvis.jarvis_agent.edit_file_handler import EditFileHandler
-from jarvis.jarvis_agent.rewrite_file_handler import RewriteFileHandler
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
 
 
@@ -23,6 +19,24 @@ class MultiAgent(OutputHandler):
         self.common_system_prompt: str = common_system_prompt
 
     def prompt(self) -> str:
+        _multiline_example_msg = """  {
+    "content": |||
+      第一行：直接换行，无需 \\n
+      第二行：包含"双引号"，无需转义
+      第三行：包含'单引号'，直接写
+      第四行：支持缩进保留
+    |||
+  }
+  
+  或使用 ``` 代替 |||：
+  {
+    "content": ```
+      第一行：直接换行，无需 \\n
+      第二行：包含"双引号"，无需转义
+      第三行：包含'单引号'，直接写
+      第四行：支持缩进保留
+    ```
+  }"""
         return f"""
 # 多智能体消息发送
 
@@ -33,22 +47,25 @@ class MultiAgent(OutputHandler):
 - **明确性原则**：清晰表达意图、需求和期望结果
 - **上下文保留**：在消息中包含足够的背景信息
 
-### 消息格式标准
+### 消息格式标准（Jsonnet）
 ```
 {ot("SEND_MESSAGE")}
-to: 智能体名称    # 目标智能体名称
-content: |2
-  # 消息主题
-  ## 背景信息
-  [提供必要的上下文和背景]
-  ## 具体需求
-  [明确表达期望完成的任务]
-  ## 相关资源
-  [列出相关文档、数据或工具]
-  ## 期望结果
-  [描述期望的输出格式和内容]
-  ## 下一步计划
-  [描述下一步的计划和行动]
+{{
+  "to": "智能体名称",
+  "content": |||
+# 消息主题
+## 背景信息
+[提供必要的上下文和背景]
+## 具体需求
+[明确表达期望完成的任务]
+## 相关资源
+[列出相关文档、数据或工具]
+## 期望结果
+[描述期望的输出格式和内容]
+## 下一步计划
+[描述下一步的计划和行动]
+  |||
+}}
 {ct("SEND_MESSAGE")}
 ```
 
@@ -56,14 +73,24 @@ content: |2
 
 ```
 {ot("SEND_MESSAGE")}
-to: 智能体名称    # 目标智能体名称
-content: |2
-  # 消息主题
-  ## 任务结果
-  [任务完成结果，用于反馈]
+{{
+  "to": "智能体名称",
+  "content": |||
+# 消息主题
+## 任务结果
+[任务完成结果，用于反馈]
+  |||
+}}
 {ct("SEND_MESSAGE")}
 ```
 
+**Jsonnet 格式说明**：
+- 可以使用双引号 "..." 或单引号 '...' 包裹字符串
+- **多行字符串推荐使用 ||| 或 ``` 分隔符包裹**，直接换行无需转义，支持保留缩进
+  示例：
+{_multiline_example_msg}
+- 支持尾随逗号
+
 ## 可用智能体资源
 {chr(10).join([f"- **{c['name']}**: {c.get('description', '')}" for c in self.agents_config])}
 """
@@ -97,12 +124,10 @@ content: |2
                     + "\n修复建议：\n"
                     "- 必须包含 to 和 content 字段\n"
                     "- to: 目标智能体名称（字符串）\n"
-                    "- content: 发送内容，建议使用多行块 |2 保持格式\n"
+                    "- content: 发送内容（字符串）\n"
                     "示例：\n"
                     f"{ot('SEND_MESSAGE')}\n"
-                    "to: 目标Agent名称\n"
-                    "content: |2\n"
-                    "  这里填写要发送的消息内容\n"
+                    '{{\n  "to": "目标Agent名称",\n  "content": "这里填写要发送的消息内容"\n}}\n'
                     f"{ct('SEND_MESSAGE')}"
                 )
                 return False, guidance
@@ -110,7 +135,7 @@ content: |2
             if not isinstance(to_val, str):
                 return False, "SEND_MESSAGE 字段类型错误：to 必须为字符串。修复建议：将 to 改为字符串，如 to: ChapterPolisher"
             if not isinstance(content_val, str):
-                return False, "SEND_MESSAGE 字段类型错误：content 必须为字符串。修复建议：将 content 改为字符串或使用多行块 content: |2"
+                return False, "SEND_MESSAGE 字段类型错误：content 必须为字符串。修复建议：将 content 改为字符串"
             # 目标校验
             if to_val not in self.agents_config_map:
                 available = ", ".join(self.agents_config_map.keys())
@@ -150,7 +175,7 @@ content: |2
                 "  这里填写要发送的消息内容\n"
                 f"{ct_tag}"
             )
-        # 尝试提取原始块并指出 YAML 问题
+        # 尝试提取原始块并指出 JSON 问题
         import re as _re
         pattern = _re.compile(
             rf"{_re.escape(ot_tag)}[ \t]*\n(.*?)(?:\n)?[ \t]*{_re.escape(ct_tag)}",
@@ -167,21 +192,19 @@ content: |2
             return (
                 False,
                 "SEND_MESSAGE 格式错误：未能识别完整的消息块。\n"
-                "修复建议：确保起止标签在单独行上，且中间内容为合法的 YAML，包含 to 与 content 字段。"
+                "修复建议：确保起止标签在单独行上，且中间内容为合法的 JSON，包含 to 与 content 字段。"
             )
         raw = blocks[0]
         try:
-            msg_obj = yaml.safe_load(raw)
+            msg_obj = json_loads(raw)
             if not isinstance(msg_obj, dict):
                 return (
                     False,
-                    "SEND_MESSAGE 内容必须为 YAML 对象（键值对）。\n"
+                    "SEND_MESSAGE 内容必须为 JSON 对象（键值对）。\n"
                     "修复建议：使用 to 与 content 字段构成的对象。\n"
                     "示例：\n"
                     f"{ot('SEND_MESSAGE')}\n"
-                    "to: 目标Agent名称\n"
-                    "content: |2\n"
-                    "  这里填写要发送的消息内容\n"
+                    '{{\n  "to": "目标Agent名称",\n  "content": "这里填写要发送的消息内容"\n}}\n'
                     f"{ct('SEND_MESSAGE')}"
                 )
             missing_keys = [k for k in ("to", "content") if k not in msg_obj]
@@ -192,16 +215,14 @@ content: |2
                     "修复建议：补充缺失字段。\n"
                     "示例：\n"
                     f"{ot('SEND_MESSAGE')}\n"
-                    "to: 目标Agent名称\n"
-                    "content: |2\n"
-                    "  这里填写要发送的消息内容\n"
+                    '{{\n  "to": "目标Agent名称",\n  "content": "这里填写要发送的消息内容"\n}}\n'
                     f"{ct('SEND_MESSAGE')}"
                 )
             # 针对值类型的提示（更细）
             if not isinstance(msg_obj.get("to"), str):
                 return False, "SEND_MESSAGE 字段类型错误：to 必须为字符串。"
             if not isinstance(msg_obj.get("content"), str):
-                return False, "SEND_MESSAGE 字段类型错误：content 必须为字符串，建议使用多行块 |2。"
+                return False, "SEND_MESSAGE 字段类型错误：content 必须为字符串。"
             # 若到此仍未返回，说明结构基本正确，但 _extract_send_msg 未命中，给出泛化建议
             return (
                 False,
@@ -209,26 +230,27 @@ content: |2
                 "修复建议：\n"
                 "- 确保起止标签各占一行\n"
                 "- 标签与内容之间保留换行\n"
-                "- 使用 content: |2 并保证 YAML 缩进一致\n"
+                "- 使用正确的 Jsonnet 格式\n"
                 "示例：\n"
                 f"{ot('SEND_MESSAGE')}\n"
-                "to: 目标Agent名称\n"
-                "content: |2\n"
-                "  这里填写要发送的消息内容\n"
+                '{{\n  "to": "目标Agent名称",\n  "content": "这里填写要发送的消息内容"\n}}\n'
                 f"{ct('SEND_MESSAGE')}"
             )
         except Exception as e:
             return (
                 False,
-                f"SEND_MESSAGE YAML 解析失败：{str(e)}\n"
+                f"SEND_MESSAGE Jsonnet 解析失败：{str(e)}\n"
                 "修复建议：\n"
-                "- 检查冒号、缩进与引号是否正确\n"
-                "- 使用 content: |2 多行块以避免缩进歧义\n"
+                "- 检查 Jsonnet 格式是否正确（引号、逗号、大括号）\n"
+                "- 可以使用双引号 \"...\" 或单引号 '...' 包裹字符串\n"
+                "- 对于多行内容，推荐使用 ||| 或 ``` 分隔符包裹，直接换行无需转义，支持保留缩进\n"
                 "示例：\n"
                 f"{ot('SEND_MESSAGE')}\n"
-                "to: 目标Agent名称\n"
-                "content: |2\n"
-                "  这里填写要发送的消息内容\n"
+                '{{\n  "to": "目标Agent名称",\n  "content": "这里填写要发送的消息内容"\n}}\n'
+                f"{ct('SEND_MESSAGE')}\n"
+                "或使用多行字符串（推荐使用 ||| 或 ``` 分隔符）：\n"
+                f"{ot('SEND_MESSAGE')}\n"
+                '{{\n  "to": "目标Agent名称",\n  "content": |||\n多行消息内容\n可以包含换行\n包含"双引号"和\'单引号\'都无需转义\n更清晰易读\n  |||\n}}\n'
                 f"{ct('SEND_MESSAGE')}"
             )
 
@@ -272,7 +294,7 @@ content: |2
         ret = []
         for item in data:
             try:
-                msg = yaml.safe_load(item)
+                msg = json_loads(item)
                 if isinstance(msg, dict) and "to" in msg and "content" in msg:
                     ret.append(msg)
             except Exception:
@@ -308,7 +330,7 @@ content: |2
                 f"{system_prompt}\n\n# 原始问题\n{self.original_question}"
             )
 
-        agent = Agent(output_handler=[ToolRegistry(),  EditFileHandler(), RewriteFileHandler(), self],**config)
+        agent = Agent(output_handler=[ToolRegistry(), self], **config)
         self.agents[name] = agent
         return agent
 
@@ -329,12 +351,12 @@ content: |2
 
             if not isinstance(msg, Dict):
                 # Should not happen if agent.run() returns str or Dict
-                PrettyOutput.print(f"未知消息类型: {type(msg)}", OutputType.WARNING)
+                print(f"⚠️ 未知消息类型: {type(msg)}")
                 break
 
             # Generate a brief summary via direct model call to avoid run-loop recursion
             # 如果在配置中显式设置了 summary_on_send=False，则不生成摘要
-            sender_config = self.agents_config_map.get(last_agent_name, {}) if hasattr(self, "agents_config_map") else {}
+            sender_config = self.agents_config_map.get(last_agent_name, {})
             summary_on_send = sender_config.get("summary_on_send", True)
             summary_text = ""
             if summary_on_send:
@@ -368,9 +390,7 @@ content: {msg['content']}
                 return "消息中未指定 `to` 字段"
 
             if to_agent_name not in self.agents_config_map:
-                PrettyOutput.print(
-                    f"未找到智能体 {to_agent_name}，正在重试...", OutputType.WARNING
-                )
+                print(f"⚠️ 未找到智能体 {to_agent_name}，正在重试...")
                 agent = self._get_agent(last_agent_name)
                 if not agent:
                     return f"智能体 {last_agent_name} 未找到"
@@ -379,9 +399,7 @@ content: {msg['content']}
                 )
                 continue
 
-            PrettyOutput.print(
-                f"{last_agent_name} 正在向 {to_agent_name} 发送消息...", OutputType.INFO
-            )
+            print(f"ℹ️ {last_agent_name} 正在向 {to_agent_name} 发送消息...")
 
             # Keep a reference to the sender before switching to the receiver
             sender_agent = agent
@@ -394,7 +412,7 @@ content: {msg['content']}
             sender_config = self.agents_config_map.get(last_agent_name, {})
             if sender_config.get("clear_after_send_message"):
                 if sender_agent:
-                    PrettyOutput.print(f"清除智能体 {last_agent_name} 在发送消息后的历史记录...", OutputType.INFO)
+                    print(f"ℹ️ 清除智能体 {last_agent_name} 在发送消息后的历史记录...")
                     sender_agent.clear_history()
 
             last_agent_name = agent.name

jarvis/jarvis_multi_agent/main.py

@@ -9,7 +9,6 @@ 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.config import set_config
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 app = typer.Typer(help="多智能体系统启动器")
 
@@ -37,12 +36,9 @@ def cli(
         # 注意：全局配置同步在 init_env 之后执行，避免被覆盖
     # 非交互模式要求从命令行传入任务
     if non_interactive and not (user_input and str(user_input).strip()):
-        PrettyOutput.print(
-            "非交互模式已启用：必须使用 --input 传入任务内容，因多行输入不可用。",
-            OutputType.ERROR,
-        )
+        print("❌ 非交互模式已启用：必须使用 --input 传入任务内容，因多行输入不可用。")
         raise typer.Exit(code=2)
-    init_env("欢迎使用 Jarvis-MultiAgent，您的多智能体系统已准备就绪！")
+    init_env()
     
     # 在初始化环境后同步 CLI 选项到全局配置，避免被 init_env 覆盖
     try:
@@ -85,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)