adam-community 1.0.24__py3-none-any.whl → 1.0.25__py3-none-any.whl

adam_community/__version__.py

@@ -1 +1 @@
-__version__ = "1.0.24"
+__version__ = "1.0.25"

adam_community/cli/templates/agent_python.py.j2

@@ -1,33 +1,159 @@
 from adam_community.tool import Tool
 from adam_community.util import runCmd
+import json
+import datetime
+
+
+# 输出处理常量
+MAX_OUTPUT_LENGTH = 10000
+TRUNCATE_PREFIX_LENGTH = 5000
+TRUNCATE_SUFFIX_LENGTH = 5000
+SUMMARY_STDOUT_LENGTH = 300
+SUMMARY_STDERR_LENGTH = 200
+
+
+def truncate_output(text):
+    """截断过长的输出文本。
+    
+    Args:
+        text (str): 需要截断的文本
+        
+    Returns:
+        str: 截断后的文本
+    """
+    if len(text) <= MAX_OUTPUT_LENGTH:
+        return text
+    return (text[:TRUNCATE_PREFIX_LENGTH] + 
+            "\n...输出太长已省略...\n" + 
+            text[-TRUNCATE_SUFFIX_LENGTH:])
+
+
+def get_status_info(exit_code):
+    """获取执行状态信息。
+    
+    Args:
+        exit_code (int): 退出码
+        
+    Returns:
+        tuple: (状态图标, 状态文本) 元组
+    """
+    if exit_code == 0:
+        return "✅", "成功"
+    return "❌", "失败"
+
+
+def format_output_display(stdout, stderr, exit_code, tool_display_name):
+    """格式化输出显示文本。
+    
+    Args:
+        stdout (str): 标准输出
+        stderr (str): 错误输出
+        exit_code (int): 退出码
+        tool_display_name (str): 工具显示名称
+        
+    Returns:
+        str: 格式化的Markdown文本
+    """
+    status_icon, status_text = get_status_info(exit_code)
+    
+    display_parts = [
+        f"## {status_icon} {tool_display_name}处理{status_text}",
+        "",
+        "### 📋 处理结果",
+        f"- 退出码: {exit_code}",
+        "- 标准输出:",
+        "```",
+        stdout,
+        "```"
+    ]
+    
+    if stderr:
+        display_parts.extend([
+            "",
+            "### ⚠️ 错误/警告信息",
+            "```",
+            stderr,
+            "```"
+        ])
+    
+    return "\n".join(display_parts)
+
+
+def generate_summary(stdout, stderr, exit_code, tool_name):
+    """生成执行结果摘要。
+    
+    Args:
+        stdout (str): 标准输出
+        stderr (str): 错误输出
+        exit_code (int): 退出码
+        tool_name (str): 工具名称
+        
+    Returns:
+        str: 执行摘要文本
+    """
+    if exit_code != 0:
+        summary = f"{tool_name}执行失败（退出码: {exit_code}）。"
+        if stderr:
+            summary += f" 错误信息: {stderr[:SUMMARY_STDERR_LENGTH]}"
+        return summary
+    
+    if stderr:
+        summary = f"{tool_name}执行完成但有警告。"
+        summary += f" 输出: {stdout[:SUMMARY_STDERR_LENGTH]}"
+        summary += f" 警告: {stderr[:SUMMARY_STDERR_LENGTH]}"
+        return summary
+    
+    summary = f"{tool_name}执行成功。"
+    summary += f" 输出摘要: {stdout[:SUMMARY_STDOUT_LENGTH]}"
+    return summary
+
 
 class {{ name.replace("-", "_") }}_tool(Tool):
     """
     {{ description }}
     
-    :param str input_text: 输入的文本内容
-    :param str? mode: 处理模式，可选值：basic、advanced
+    Args:
+        input_text (str): 输入的文本内容
+        mode (str, optional): 处理模式，可选值：basic、advanced
     """
-    
     DISPLAY_NAME = "{{ display_name }}工具"
-    NETWORK = False  # 根据需要设置是否需要网络访问
-    CPU = 1  # 需要的CPU数量
-    GPU = 0  # 需要的GPU数量
-    SIF = "adam-base:1.0.0"  # 指定容器镜像（必填）
+    NETWORK = False
+    CPU = 1
+    GPU = 0
+    SIF = "adam-base:1.0.0"
     
-    def call(self, kwargs):
-        """
-        工具的主要实现函数
+    # 输出文件配置
+    OUTPUT_FILE = "output.txt"
+
+    def inputShow(self, **kwargs):
+        """AI决定使用该工具时调用，用于向用户展示当前调用的工具信息。
         
         Args:
-            kwargs: 包含args、message、files、user、task_id等参数的字典
+            kwargs (dict): 包含args、message、files、user、task_id等参数的字典。
         
         Returns:
-            执行结果字符串（bash命令）或直接执行结果
+            str: 给用户显示的字符串（可使用markdown渲染）。
         """
-        # 获取用户输入的参数
-        input_text = kwargs['args']['input_text']
-        mode = kwargs['args'].get('mode', 'basic')
+        args = kwargs['args']
+        input_text = args['input_text']
+        mode = args.get('mode', 'basic')
+        
+        show_text = f"正在使用 {{ display_name }} 工具处理文本内容"
+        if mode == 'advanced':
+            show_text += "（高级模式）"
+        
+        return Tool.markdown_terminal(
+            show_text,
+            workdir=kwargs.get("task_id", ""),
+            user=kwargs.get("user", "Adam"),
+            conda_env=getattr(self, "CONDA_ENV", "base")
+        )
+
+    def call(self, kwargs):
+        """Tool 逻辑实现函数"""    
+        args = kwargs['args']
+        input_text = args['input_text']
+        mode = args.get('mode', 'basic')
         
         # TODO: 在这里实现你的主要逻辑
         if mode == 'advanced':
@@ -36,8 +162,7 @@ class {{ name.replace("-", "_") }}_tool(Tool):
             result = f"基础模式处理: {input_text}"
         
         # 生成bash命令
-        bash_script = f'''
-# {{ display_name }} 处理脚本
+        bash_script = f'''# {{ display_name }} 处理脚本
 echo "开始处理: {input_text}"
 echo "处理模式: {mode}"
 
@@ -45,79 +170,104 @@ echo "处理模式: {mode}"
 echo "{result}"
 
 # 生成输出文件
-echo "处理完成" > output.txt
-echo "结果: {result}" >> output.txt
+echo "处理完成" > {self.OUTPUT_FILE}
+echo "结果: {result}" >> {self.OUTPUT_FILE}
 
-echo "处理完成，结果已保存到 output.txt"
+echo "处理完成，结果已保存到 {self.OUTPUT_FILE}"
 '''
         
         return runCmd(bash_script)
     
     def outputShow(self, kwargs):
-        """
-        自定义输出显示格式（可选）
-        """
-        stdout = kwargs.get('stdout', '')
-        stderr = kwargs.get('stderr', '')
+        """任务结束时调用，用于向用户展示工具执行结果。
         
-        display_text = f"""## 🎉 {{ display_name }} 处理完成
-
-### 📋 处理结果
-```
-{stdout}
-```
-"""
+        Args:
+            kwargs (dict): 包含stdout、stderr、exit_code等执行结果的字典。
         
-        if stderr:
-            display_text += f"""
-### ⚠️ 警告信息
-```
-{stderr}
-```
-"""
+        Returns:
+            tuple: (用户显示字符串, 文件列表) 元组。
+                - 用户显示字符串 (str): 格式化的显示文本
+                - 文件列表 (list): 生成的文件列表，会在前端提供下载
+        """
+        stdout = truncate_output(kwargs.get('stdout', ''))
+        stderr = truncate_output(kwargs.get('stderr', ''))
+        exit_code = kwargs.get('exit_code', 0)
         
-        # 返回可能生成的文件
-        file_list = ['output.txt'] if 'output.txt' in stdout else []
+        display_text = format_output_display(
+            stdout, stderr, exit_code, "{{ display_name }} "
+        )
+        
+        # 根据执行结果判断生成的文件
+        file_list = []
+        if exit_code == 0 or self.OUTPUT_FILE in kwargs.get('stdout', ''):
+            file_list.append(self.OUTPUT_FILE)
         
         return display_text, file_list
     
     def summary(self, kwargs):
+        """任务结束时调用，用于向AI提供执行结果摘要。
+        
+        Args:
+            kwargs (dict): 包含stdout、stderr、exit_code等执行结果的字典。
+        
+        Returns:
+            str: 给AI的文本摘要，应简洁明了，避免上下文过长。
         """
-        为AI生成简短的执行摘要
-        """
-        stdout = kwargs.get('stdout', '')
-        stderr = kwargs.get('stderr', '')
+        stdout = truncate_output(kwargs.get('stdout', ''))
+        stderr = truncate_output(kwargs.get('stderr', ''))
+        exit_code = kwargs.get('exit_code', 0)
         
-        if stderr:
-            return f"{{ display_name }}工具执行完成但有警告: {stdout[:200]}...。警告: {stderr[:100]}"
-        else:
-            return f"{{ display_name }}工具执行成功: {stdout[:200]}..."
+        return generate_summary(
+            stdout, stderr, exit_code, "{{ display_name }}工具"
+        )
 
 
-class {{ name.replace("-", "_") }}_advanced_tool(Tool, calltype="python"):
-    """
-    {{ description }} - 高级功能
-    
-    :param str input_file: 输入文件路径
-    :param str output_format: 输出格式，可选值：json、csv、txt
-    """
+class {{ name.replace("-", "_") }}_advanced_tool(Tool, calltype="python"):    
+    """{{ description }} - 高级功能
     
+    Args:
+        input_file (str): 输入文件路径
+        output_format (str): 输出格式，可选值：json、csv、txt
+    """    
     DISPLAY_NAME = "{{ display_name }}高级工具"
-    NETWORK = False  # 根据需要设置是否需要网络访问
-    CPU = 1  # 需要的CPU数量
-    GPU = 0  # 需要的GPU数量
-    SIF = "adam-base:1.0.0"  # 指定容器镜像（必填）
-    CONDA_ENV = "base"  # 指定conda环境
+    NETWORK = False
+    CPU = 1
+    GPU = 0
+    SIF = "adam-base:1.0.0"
+    CONDA_ENV = "base"
     
-    def call(self, kwargs):
-        """
-        Python类型工具的实现
+    # 输出文件配置
+    OUTPUT_FILE = "result.json"
+    DEFAULT_OUTPUT_FORMAT = "json"
+
+    def inputShow(self, **kwargs):
+        """AI决定使用该工具时调用，用于向用户展示当前调用的工具信息。
+        
+        Args:
+            kwargs (dict): 包含args、message、files、user、task_id等参数的字典。
+        
+        Returns:
+            str: 给用户显示的字符串（可使用markdown渲染）。
         """
-        import json
-        import datetime
+        args = kwargs['args']
+        input_file = args['input_file']
+        output_format = args.get('output_format', self.DEFAULT_OUTPUT_FORMAT)
         
-        input_file = kwargs['args']['input_file']
-        output_format = kwargs['args'].get('output_format', 'json')
+        show_text = (f"正在使用 {{ display_name }} 高级工具处理文件: {input_file}"
+                    f"（输出格式: {output_format}）")
+        
+        return Tool.markdown_terminal(
+            show_text,
+            workdir=kwargs.get("task_id", ""),
+            user=kwargs.get("user", "Adam"),
+            conda_env=self.CONDA_ENV
+        )
+    
+    def call(self, kwargs):    
+        """Tool 逻辑实现函数"""
+        args = kwargs['args']
+        input_file = args['input_file']
+        output_format = args.get('output_format', self.DEFAULT_OUTPUT_FORMAT)
         
         # TODO: 在这里实现你的Python逻辑
         result = {
@@ -134,6 +284,51 @@ class {{ name.replace("-", "_") }}_advanced_tool(Tool, calltype="python"):
         
         # 生成输出文件
         if output_format == 'json':
-            with open('result.json', 'w', encoding='utf-8') as f:
+            with open(self.OUTPUT_FILE, 'w', encoding='utf-8') as f:
                 json.dump(result, f, indent=2, ensure_ascii=False)
-            print("结果已保存到 result.json")
+            print(f"结果已保存到 {self.OUTPUT_FILE}")
+    
+    def outputShow(self, kwargs):
+        """任务结束时调用，用于向用户展示工具执行结果。
+        
+        Args:
+            kwargs (dict): 包含stdout、stderr、exit_code等执行结果的字典。
+        
+        Returns:
+            tuple: (用户显示字符串, 文件列表) 元组。
+                - 用户显示字符串 (str): 格式化的显示文本
+                - 文件列表 (list): 生成的文件列表，会在前端提供下载
+        """
+        stdout = truncate_output(kwargs.get('stdout', ''))
+        stderr = truncate_output(kwargs.get('stderr', ''))
+        exit_code = kwargs.get('exit_code', 0)
+        
+        display_text = format_output_display(
+            stdout, stderr, exit_code, "{{ display_name }} 高级工具"
+        )
+        
+        # 根据输出格式判断生成的文件
+        file_list = []
+        args = kwargs.get('args', {})
+        output_format = args.get('output_format', self.DEFAULT_OUTPUT_FORMAT)
+        if output_format == 'json' and exit_code == 0:
+            file_list.append(self.OUTPUT_FILE)
+        
+        return display_text, file_list
+    
+    def summary(self, kwargs):
+        """任务结束时调用，用于向AI提供执行结果摘要。
+        
+        Args:
+            kwargs (dict): 包含stdout、stderr、exit_code等执行结果的字典。
+        
+        Returns:
+            str: 给AI的文本摘要，应简洁明了，避免上下文过长。
+        """
+        stdout = truncate_output(kwargs.get('stdout', ''))
+        stderr = truncate_output(kwargs.get('stderr', ''))
+        exit_code = kwargs.get('exit_code', 0)
+        
+        return generate_summary(
+            stdout, stderr, exit_code, "{{ display_name }}高级工具"
+        )

{adam_community-1.0.24.dist-info → adam_community-1.0.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: adam_community
-Version: 1.0.24
+Version: 1.0.25
 Summary: Adam Community Tools and Utilities
 Home-page: https://github.com/yourusername/adam-community
 Author: Adam Community

{adam_community-1.0.24.dist-info → adam_community-1.0.25.dist-info}/RECORD

@@ -1,5 +1,5 @@
 adam_community/__init__.py,sha256=vAmF9VQR6D4peppH0hnrHDmZK5cFeFPh11GIsTKUXhE,429
-adam_community/__version__.py,sha256=rnKYfyIc2hXBxsPKz6c7OchEPYsBq3xHKgj-1s8ShT8,23
+adam_community/__version__.py,sha256=_TgC2mrlgs6XxMzx9bzihItIy81yZaza2GnTmMNF5EA,23
 adam_community/tool.py,sha256=F6jxRU3urqTfgjLIZSW-hVWyj0FpNwvY65jOODXI19w,4954
 adam_community/util.py,sha256=SDXnWLHLfJxEJ5WBnwNq_FeHNwXxGK_F36dpI_hVkNc,17766
 adam_community/cli/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
@@ -13,7 +13,7 @@ adam_community/cli/templates/README_agent.md.j2,sha256=lbhaEPZrgbCfUG-T7GItkgfIV
 adam_community/cli/templates/README_kit.md.j2,sha256=5_1ISGUxJKdDild2vNSWTC_hNhZHnEHvyN9Ez4Xbqwk,5715
 adam_community/cli/templates/README_toolbox.md.j2,sha256=lbhaEPZrgbCfUG-T7GItkgfIV9h1rqxyfRQuGhcdOqU,8901
 adam_community/cli/templates/__init__.py,sha256=43rU9rFkpsVrWjxR-brDnT2eakgRtb4XpnunbE-ais4,34
-adam_community/cli/templates/agent_python.py.j2,sha256=EOnmsJUvQRrcO7K7c88kI42gMmcM0Z4ab46qwOJXbH8,4192
+adam_community/cli/templates/agent_python.py.j2,sha256=7lim4KHtkZ78y6g0dE5YyDjhA6QKfIq-XaNKs2ryjho,10676
 adam_community/cli/templates/configure.json.j2,sha256=HWhhFGEZsXxDJCkciehY8SqaV7kSPQ4588TR9WN__cI,354
 adam_community/cli/templates/initial_assistant_message.md.j2,sha256=R17qhco2Ds5es2HMlAPcinVi5Z-vLfyyu1lY9HMBFk4,595
 adam_community/cli/templates/initial_assistant_message_en.md.j2,sha256=o1nPDF_DBOA0pjnrxYCIjFWPLaNmE8G07OA-DMA5uNo,733
@@ -25,8 +25,8 @@ adam_community/cli/templates/long_description.md.j2,sha256=Rj6hcuNzEL0Sp17GQVCRJ
 adam_community/cli/templates/long_description_en.md.j2,sha256=xSbahwGarXlWopZqHw7lrcv1dQuvwj2ChhZv5pJmUy4,1725
 adam_community/cli/templates/rag_python.py.j2,sha256=YJL7-WIx-Dumt7lHuUGxl3Rbaw0kpkh8hpcCJ5lz9lA,2494
 adam_community/cli/templates/toolbox_python.py.j2,sha256=EOnmsJUvQRrcO7K7c88kI42gMmcM0Z4ab46qwOJXbH8,4192
-adam_community-1.0.24.dist-info/METADATA,sha256=sxFaC3OQBjjOuHER7x6kcuYZxgV6pwqRlevJ-rVa800,3151
-adam_community-1.0.24.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-adam_community-1.0.24.dist-info/entry_points.txt,sha256=4I7yRkn7cHwPY8-fWQLeAvKjc24zUy8Z65VsZNs0Wos,56
-adam_community-1.0.24.dist-info/top_level.txt,sha256=MS8jbePXKZChih9kGizNVX0I1MFZFGWBMCIW_r86qhU,15
-adam_community-1.0.24.dist-info/RECORD,,
+adam_community-1.0.25.dist-info/METADATA,sha256=3i_FpqlCX-IHuep5VdeEkel5JFB6MD13no1QsuY_3MY,3151
+adam_community-1.0.25.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+adam_community-1.0.25.dist-info/entry_points.txt,sha256=4I7yRkn7cHwPY8-fWQLeAvKjc24zUy8Z65VsZNs0Wos,56
+adam_community-1.0.25.dist-info/top_level.txt,sha256=MS8jbePXKZChih9kGizNVX0I1MFZFGWBMCIW_r86qhU,15
+adam_community-1.0.25.dist-info/RECORD,,