beswarm 0.2.50__tar.gz → 0.2.51__tar.gz

{beswarm-0.2.50 → beswarm-0.2.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.2.50
+Version: 0.2.51
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/agents/chatgroup.py

@@ -15,8 +15,8 @@ from typing import List, Dict, Union
 from ..broker import MessageBroker
 from ..bemcp.bemcp import MCPManager
 from ..utils import register_mcp_tools
-from ..aient.src.aient.models import chatgpt
-from ..aient.src.aient.plugins import get_function_call_list, registry
+from ..aient.aient.models import chatgpt
+from ..aient.aient.plugins import get_function_call_list, registry
 
 worker_system_prompt = """
 你是{name}。帮助用户头脑风暴。请分析不同用户的观点，并给出你的观点。

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/agents/planact.py

@@ -10,8 +10,8 @@ from datetime import datetime
 from typing import List, Dict, Union
 
 from ..broker import MessageBroker
-from ..aient.src.aient.models import chatgpt
-from ..aient.src.aient.plugins import get_function_call_list, registry
+from ..aient.aient.models import chatgpt
+from ..aient.aient.plugins import get_function_call_list, registry
 from ..prompt import worker_system_prompt, instruction_system_prompt
 from ..utils import extract_xml_content, get_current_screen_image_message, replace_xml_content, register_mcp_tools, setup_logger
 

{beswarm-0.2.50/beswarm/aient/src → beswarm-0.2.51/beswarm/aient}/aient/core/response.py

@@ -33,6 +33,8 @@ def gemini_json_poccess(response_str):
         promptTokenCount = safe_get(response_json, "usageMetadata", "promptTokenCount", default=0)
         candidatesTokenCount = safe_get(response_json, "usageMetadata", "candidatesTokenCount", default=0)
         totalTokenCount = safe_get(response_json, "usageMetadata", "totalTokenCount", default=0)
+        if finishReason != "STOP":
+            logger.error(f"finishReason: {finishReason}")
 
     content = reasoning_content = safe_get(json_data, "parts", 0, "text", default="")
     b64_json = safe_get(json_data, "parts", 0, "inlineData", "data", default="")

beswarm-0.2.51/beswarm/aient/aient/models/__init__.py

@@ -0,0 +1,2 @@
+from .chatgpt import *
+from .audio import *

{beswarm-0.2.50/beswarm/aient/src → beswarm-0.2.51/beswarm/aient}/aient/plugins/excute_command.py

@@ -276,4 +276,4 @@ for i in content.split("\\n"):
 #     print(excute_command(python_long_task_command))
 
     # print(get_python_executable("python -c 'print(123)'"))
-# python -m beswarm.aient.src.aient.plugins.excute_command
+# python -m beswarm.aient.aient.plugins.excute_command

{beswarm-0.2.50/beswarm/aient/src → beswarm-0.2.51/beswarm/aient}/aient/plugins/read_file.py

@@ -181,7 +181,7 @@ Examples:
         return f"<tool_error>读取文件时发生错误: {e}</tool_error>"
 
 if __name__ == "__main__":
-    # python -m beswarm.aient.src.aient.plugins.read_file
+    # python -m beswarm.aient.aient.plugins.read_file
     result = read_file("./work/cax/Lenia Notebook.ipynb")
     print(result)
     print(len(result))

{beswarm-0.2.50/beswarm/aient/src → beswarm-0.2.51/beswarm/aient}/aient/plugins/websearch.py

@@ -344,7 +344,7 @@ async def get_search_results(query):
 
 if __name__ == "__main__":
     os.system("clear")
-    # python -m beswarm.aient.src.aient.plugins.websearch
+    # python -m beswarm.aient.aient.plugins.websearch
     print(get_url_content(""))
     # from aient.models import chatgpt
     # print(get_search_results("今天的微博热搜有哪些？", chatgpt.chatgpt_api_url.v1_url))

{beswarm-0.2.50/beswarm/aient/src → beswarm-0.2.51/beswarm/aient}/aient/plugins/write_file.py

@@ -87,4 +87,4 @@ if __name__ == "__main__":
     with open("test.txt", "r", encoding="utf-8") as file:
         content = file.read()
     print(write_to_file("test.txt", content))
-    # python -m beswarm.aient.src.aient.plugins.write_file
+    # python -m beswarm.aient.aient.plugins.write_file

{beswarm-0.2.50/beswarm/aient/src → beswarm-0.2.51/beswarm/aient}/aient/utils/scripts.py

@@ -1050,4 +1050,4 @@ if __name__ == "__main__":
     # print(parse_function_xml(test_xml))
     print(remove_xml_tags_and_content(test_xml))
 
-# 运行本文件：python -m beswarm.aient.src.aient.utils.scripts
+# 运行本文件：python -m beswarm.aient.aient.utils.scripts

beswarm-0.2.51/beswarm/aient/main.py

@@ -0,0 +1,15 @@
+import os
+
+from aient.utils import prompt
+from aient.models import chatgpt
+GPT_ENGINE = os.environ.get('MODEL')
+
+API = os.environ.get('API_KEY')
+API_URL = os.environ.get('BASE_URL', None)
+
+message = "hi"
+systemprompt = os.environ.get('SYSTEMPROMPT', prompt.chatgpt_system_prompt)
+
+bot = chatgpt(api_key=API, api_url=API_URL , engine=GPT_ENGINE, system_prompt=systemprompt)
+for text in bot.ask_stream(message):
+    print(text, end="")

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/knowledge_graph.py

@@ -1,7 +1,7 @@
 import uuid
 import networkx as nx
 from pathlib import Path
-from .aient.src.aient.utils.scripts import unescape_html
+from .aient.aient.utils.scripts import unescape_html
 
 class KnowledgeGraphManager:
     """
@@ -130,6 +130,32 @@ class KnowledgeGraphManager:
         self._save_graph()
         return f"✅ 成功在 '{parent_path}' 下添加节点 '{node_name}'。"
 
+    def _get_path_by_node_id(self, node_id: str) -> str:
+        """通过节点ID获取其在图中的完整路径。"""
+        if node_id == "root":
+            return "."
+
+        path_segments = []
+        current_id = node_id
+        while current_id != "root":
+            if current_id not in self.graph:
+                return None
+
+            # 假设每个节点只有一个父节点（树状结构）
+            predecessors = list(self.graph.predecessors(current_id))
+            if not predecessors:
+                return None
+
+            parent_id = predecessors[0]
+            node_name = self.graph.nodes[current_id].get('name')
+            if node_name is None:
+                return None
+
+            path_segments.append(node_name)
+            current_id = parent_id
+
+        return "/".join(reversed(path_segments))
+
     def _get_tags(self, node_id: str) -> list[str]:
         """获取节点的标签列表。"""
         tags_str = self.graph.nodes[node_id].get('tags', '')
@@ -248,7 +274,26 @@ class KnowledgeGraphManager:
         node_path = unescape_html(node_path)
         node_id = self._get_node_id_by_path(node_path)
         if node_id is None:
-            return f"❌ 错误：路径 '{node_path}' 不存在。"
+            path_segments = [s for s in node_path.strip('/').split('/') if s and s != '.']
+            if not path_segments:
+                return f"❌ 错误：路径 '{node_path}' 不存在。"
+
+            target_name = path_segments[-1]
+            suggestions = []
+            for n_id, data in self.graph.nodes(data=True):
+                if n_id != 'root' and target_name in data.get('name', ''):
+                    full_path = self._get_path_by_node_id(n_id)
+                    if full_path:
+                        suggestions.append(full_path)
+
+            if suggestions:
+                # 去重并排序
+                suggestions = sorted(list(set(suggestions)))
+                message = f"❌ 错误：路径 '{node_path}' 不存在。\n\n您是不是要找：\n"
+                message += "\n".join([f"- {s}" for s in suggestions])
+                return message
+            else:
+                return f"❌ 错误：路径 '{node_path}' 不存在。"
 
         node_data = self.graph.nodes[node_id]
         description = node_data.get('description', '无描述。')

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/taskmanager.py

@@ -5,7 +5,7 @@ import asyncio
 from enum import Enum
 from pathlib import Path
 
-from .aient.src.aient.plugins import registry
+from .aient.aient.plugins import registry
 
 class TaskStatus(Enum):
     """任务状态枚举"""

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/__init__.py

@@ -21,7 +21,7 @@ from .click import find_and_click_element, scroll_screen
 from .subtasks import create_task, resume_task, get_all_tasks_status, get_task_result, create_tasks_from_csv
 
 #显式导入 aient.plugins 中的所需内容
-from ..aient.src.aient.plugins import (
+from ..aient.aient.plugins import (
     get_time,
     read_file,
     read_image,

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/click.py

@@ -7,10 +7,10 @@ import base64
 import pyperclip # 新增：用于操作剪贴板
 import platform  # 新增：用于检测操作系统
 from PIL import Image, ImageDraw
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
-from ..aient.src.aient.models import chatgpt
-from ..aient.src.aient.core.utils import get_image_message, get_text_message
+from ..aient.aient.models import chatgpt
+from ..aient.aient.core.utils import get_image_message, get_text_message
 
 def display_image_with_bounding_boxes_and_masks_py(
     original_image,

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/completion.py

@@ -1,4 +1,4 @@
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 @register_tool()
 def task_complete(message: str) -> str:

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/edit_file.py

@@ -1,8 +1,8 @@
 import os
 import re
 import difflib
-from ..aient.src.aient.plugins import register_tool
-from ..aient.src.aient.utils.scripts import unescape_html
+from ..aient.aient.plugins import register_tool
+from ..aient.aient.utils.scripts import unescape_html
 
 @register_tool()
 def edit_file(file_path, diff_content, match_precision=0.9):

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/graph.py

@@ -2,7 +2,7 @@ import ast
 from typing import List
 
 from ..core import kgm
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 @register_tool()
 def add_knowledge_node(parent_path: str, node_name: str, description: str = "", tags: List[str] = None) -> str:

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/planner.py

@@ -1,8 +1,8 @@
 import os
-from ..aient.src.aient.plugins import register_tool, get_function_call_list
+from ..aient.aient.plugins import register_tool, get_function_call_list
 
-from ..aient.src.aient.models import chatgpt
-from ..aient.src.aient.prompt import planner_system_prompt
+from ..aient.aient.models import chatgpt
+from ..aient.aient.prompt import planner_system_prompt
 
 @register_tool()
 async def planner(goal, tools, work_dir):

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/repomap.py

@@ -10,7 +10,7 @@ import warnings
 from pathlib import Path
 from collections import Counter, defaultdict, namedtuple
 
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 from tqdm import tqdm
 from diskcache import Cache

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/request_input.py

@@ -1,4 +1,4 @@
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 @register_tool()
 def request_admin_input(prompt: str) -> str:

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/screenshot.py

@@ -1,4 +1,4 @@
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 @register_tool()
 def save_screenshot_to_file(save_path):

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/search_arxiv.py

@@ -1,7 +1,7 @@
 import requests
 import csv
 from datetime import datetime
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 @register_tool()
 def search_arxiv(query, max_results=5, sort_by='relevance', sort_order='descending',

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/search_web.py

@@ -4,7 +4,7 @@ import json
 import httpx
 import threading
 
-from ..aient.src.aient.plugins import register_tool, get_url_content # Assuming a similar plugin structure
+from ..aient.aient.plugins import register_tool, get_url_content # Assuming a similar plugin structure
 
 class ThreadWithReturnValue(threading.Thread):
     def run(self):

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/subtasks.py

@@ -1,7 +1,7 @@
 import ast
 from pathlib import Path
 from ..core import task_manager
-from ..aient.src.aient.plugins import register_tool, registry
+from ..aient.aient.plugins import register_tool, registry
 
 worker_fun = registry.tools["worker"]
 

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/worker.py

@@ -4,7 +4,7 @@ from typing import List, Dict, Union
 from ..core import mcp_manager, broker, task_manager, kgm
 from ..agents.planact import BrokerWorker
 from ..agents.chatgroup import ChatGroupWorker
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 
 @register_tool()

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/tools/write_csv.py

@@ -1,7 +1,7 @@
 import csv
 import ast
 import os
-from ..aient.src.aient.plugins import register_tool
+from ..aient.aient.plugins import register_tool
 
 @register_tool()
 def append_row_to_csv(file_path: str, data: list):

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm/utils.py

@@ -37,7 +37,7 @@ def replace_xml_content(original_string: str, tag_name: str, replacement_content
 
 import io
 import base64
-from .aient.src.aient.core.utils import get_image_message, get_text_message
+from .aient.aient.core.utils import get_image_message, get_text_message
 
 async def get_current_screen_image_message(prompt):
     print("instruction agent 正在截取当前屏幕...")

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.2.50
+Version: 0.2.51
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.2.50 → beswarm-0.2.51}/beswarm.egg-info/SOURCES.txt

@@ -16,74 +16,46 @@ beswarm.egg-info/top_level.txt
 beswarm/agents/chatgroup.py
 beswarm/agents/planact.py
 beswarm/aient/main.py
-beswarm/aient/setup.py
-beswarm/aient/src/aient/__init__.py
-beswarm/aient/src/aient/core/__init__.py
-beswarm/aient/src/aient/core/log_config.py
-beswarm/aient/src/aient/core/models.py
-beswarm/aient/src/aient/core/request.py
-beswarm/aient/src/aient/core/response.py
-beswarm/aient/src/aient/core/utils.py
-beswarm/aient/src/aient/core/test/test_base_api.py
-beswarm/aient/src/aient/core/test/test_geminimask.py
-beswarm/aient/src/aient/core/test/test_image.py
-beswarm/aient/src/aient/core/test/test_payload.py
-beswarm/aient/src/aient/models/__init__.py
-beswarm/aient/src/aient/models/audio.py
-beswarm/aient/src/aient/models/base.py
-beswarm/aient/src/aient/models/chatgpt.py
-beswarm/aient/src/aient/models/claude.py
-beswarm/aient/src/aient/models/duckduckgo.py
-beswarm/aient/src/aient/models/gemini.py
-beswarm/aient/src/aient/models/groq.py
-beswarm/aient/src/aient/models/vertex.py
-beswarm/aient/src/aient/plugins/__init__.py
-beswarm/aient/src/aient/plugins/arXiv.py
-beswarm/aient/src/aient/plugins/config.py
-beswarm/aient/src/aient/plugins/excute_command.py
-beswarm/aient/src/aient/plugins/get_time.py
-beswarm/aient/src/aient/plugins/image.py
-beswarm/aient/src/aient/plugins/list_directory.py
-beswarm/aient/src/aient/plugins/read_file.py
-beswarm/aient/src/aient/plugins/read_image.py
-beswarm/aient/src/aient/plugins/readonly.py
-beswarm/aient/src/aient/plugins/registry.py
-beswarm/aient/src/aient/plugins/run_python.py
-beswarm/aient/src/aient/plugins/websearch.py
-beswarm/aient/src/aient/plugins/write_file.py
-beswarm/aient/src/aient/utils/__init__.py
-beswarm/aient/src/aient/utils/prompt.py
-beswarm/aient/src/aient/utils/scripts.py
-beswarm/aient/test/chatgpt.py
-beswarm/aient/test/claude.py
-beswarm/aient/test/test.py
-beswarm/aient/test/test_API.py
-beswarm/aient/test/test_Deepbricks.py
+beswarm/aient/aient/__init__.py
+beswarm/aient/aient/core/__init__.py
+beswarm/aient/aient/core/log_config.py
+beswarm/aient/aient/core/models.py
+beswarm/aient/aient/core/request.py
+beswarm/aient/aient/core/response.py
+beswarm/aient/aient/core/utils.py
+beswarm/aient/aient/core/test/test_base_api.py
+beswarm/aient/aient/core/test/test_geminimask.py
+beswarm/aient/aient/core/test/test_image.py
+beswarm/aient/aient/core/test/test_payload.py
+beswarm/aient/aient/models/__init__.py
+beswarm/aient/aient/models/audio.py
+beswarm/aient/aient/models/base.py
+beswarm/aient/aient/models/chatgpt.py
+beswarm/aient/aient/plugins/__init__.py
+beswarm/aient/aient/plugins/arXiv.py
+beswarm/aient/aient/plugins/config.py
+beswarm/aient/aient/plugins/excute_command.py
+beswarm/aient/aient/plugins/get_time.py
+beswarm/aient/aient/plugins/image.py
+beswarm/aient/aient/plugins/list_directory.py
+beswarm/aient/aient/plugins/read_file.py
+beswarm/aient/aient/plugins/read_image.py
+beswarm/aient/aient/plugins/readonly.py
+beswarm/aient/aient/plugins/registry.py
+beswarm/aient/aient/plugins/run_python.py
+beswarm/aient/aient/plugins/websearch.py
+beswarm/aient/aient/plugins/write_file.py
+beswarm/aient/aient/utils/__init__.py
+beswarm/aient/aient/utils/prompt.py
+beswarm/aient/aient/utils/scripts.py
 beswarm/aient/test/test_Web_crawler.py
-beswarm/aient/test/test_aiwaves.py
-beswarm/aient/test/test_aiwaves_arxiv.py
-beswarm/aient/test/test_ask_gemini.py
-beswarm/aient/test/test_class.py
-beswarm/aient/test/test_claude.py
-beswarm/aient/test/test_claude_zh_char.py
 beswarm/aient/test/test_ddg_search.py
-beswarm/aient/test/test_download_pdf.py
-beswarm/aient/test/test_gemini.py
-beswarm/aient/test/test_get_token_dict.py
 beswarm/aient/test/test_google_search.py
-beswarm/aient/test/test_jieba.py
-beswarm/aient/test/test_json.py
-beswarm/aient/test/test_logging.py
 beswarm/aient/test/test_ollama.py
 beswarm/aient/test/test_plugin.py
-beswarm/aient/test/test_py_run.py
-beswarm/aient/test/test_requests.py
 beswarm/aient/test/test_search.py
-beswarm/aient/test/test_tikitoken.py
-beswarm/aient/test/test_token.py
 beswarm/aient/test/test_url.py
 beswarm/aient/test/test_whisper.py
-beswarm/aient/test/test_wildcard.py
 beswarm/aient/test/test_yjh.py
 beswarm/bemcp/bemcp/__init__.py
 beswarm/bemcp/bemcp/decorator.py

{beswarm-0.2.50 → beswarm-0.2.51}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.2.50"
+version = "0.2.51"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"

{beswarm-0.2.50 → beswarm-0.2.51}/test/test_graph.py

@@ -203,6 +203,67 @@ class KnowledgeGraphManager:
         self._save_graph()
         return f"✅ 成功将节点 '{source_path}' 移动到 '{target_parent_path}' 下。"
 
+    def _get_path_by_node_id(self, node_id: str) -> str:
+        """通过节点ID获取其在图中的完整路径。"""
+        if node_id == "root":
+            return "."
+
+        path_segments = []
+        current_id = node_id
+        while current_id != "root":
+            if current_id not in self.graph:
+                return None
+
+            predecessors = list(self.graph.predecessors(current_id))
+            if not predecessors:
+                return None
+
+            parent_id = predecessors[0]
+            node_name = self.graph.nodes[current_id].get('name')
+            if node_name is None:
+                return None
+
+            path_segments.append(node_name)
+            current_id = parent_id
+
+        return "/".join(reversed(path_segments))
+
+    def get_node_details(self, node_path: str) -> str:
+        """获取指定路径节点的所有详细信息。"""
+        node_id = self._get_node_id_by_path(node_path)
+        if node_id is None:
+            path_segments = [s for s in node_path.strip('/').split('/') if s and s != '.']
+            if not path_segments:
+                return f"❌ 错误：路径 '{node_path}' 不存在。"
+
+            target_name = path_segments[-1]
+            suggestions = []
+            for n_id, data in self.graph.nodes(data=True):
+                if n_id != 'root' and target_name in data.get('name', ''):
+                    full_path = self._get_path_by_node_id(n_id)
+                    if full_path:
+                        suggestions.append(full_path)
+
+            if suggestions:
+                suggestions = sorted(list(set(suggestions)))
+                message = f"❌ 错误：路径 '{node_path}' 不存在。\n\n您是不是要找：\n"
+                message += "\n".join([f"- {s}" for s in suggestions])
+                return message
+            else:
+                return f"❌ 错误：路径 '{node_path}' 不存在。"
+
+        node_data = self.graph.nodes[node_id]
+        description = node_data.get('description', '无描述。')
+        tags = self._get_tags(node_id)
+
+        details = [f"节点: {node_path}", f"描述: {description}"]
+        if tags:
+            details.append(f"标签: {' '.join([f'#{t}' for t in tags])}")
+        else:
+            details.append("标签: 无")
+
+        return "\n".join(details)
+
     def render_tree(self, show_source_tags=False) -> str:
         """渲染整个知识图谱为树状结构的文本。"""
         if not self.graph or "root" not in self.graph:
@@ -326,5 +387,47 @@ def main_test_loop():
 
         input("\n按回车键继续...")
 
+def run_automated_tests():
+    """运行自动化测试，验证核心功能。"""
+    print("\n" + "="*50)
+    print("🚀 开始运行自动化测试...")
+    print("="*50)
+
+    test_file = "test_graph.graphml"
+    if os.path.exists(test_file):
+        os.remove(test_file)
+
+    kgm = KnowledgeGraphManager(storage_path=test_file)
+
+    # 1. 测试节点建议功能
+    print("\n" + colors.HEADER + "1. 测试 'get_node_details' 的建议功能..." + colors.ENDC)
+    kgm.add_node(".", "技术")
+    kgm.add_node("技术", "深度学习")
+    kgm.add_node("技术/深度学习", "挑战")
+    kgm.add_node(".", "应用")
+    kgm.add_node("应用", "挑战")
+
+    # 查找一个不存在但有明确建议的节点
+    result = kgm.get_node_details("挑战")
+    expected_suggestions = ["技术/深度学习/挑战", "应用/挑战"]
+
+    print(f"查询 '挑战' 的结果:\n{result}")
+
+    all_found = all(suggestion in result for suggestion in expected_suggestions)
+    if "您是不是要找" in result and all_found:
+        print(colors.OKGREEN + "✅ 测试通过: 成功提供了正确的建议。" + colors.ENDC)
+    else:
+        print(colors.FAIL + "❌ 测试失败: 未能提供正确的建议。" + colors.ENDC)
+
+    # 2. 清理测试文件
+    if os.path.exists(test_file):
+        os.remove(test_file)
+    print("\n" + "🧹 清理测试文件完成。")
+    print("\n" + "="*50)
+    print("✅ 自动化测试结束。")
+    print("="*50)
+
+
 if __name__ == "__main__":
+    run_automated_tests()
     main_test_loop()

beswarm-0.2.50/beswarm/aient/main.py

@@ -1,50 +0,0 @@
-import os
-from datetime import datetime
-
-from src.aient.utils import prompt
-from src.aient.models import chatgpt, claude3, gemini, groq
-LANGUAGE = os.environ.get('LANGUAGE', 'Simplified Chinese')
-GPT_ENGINE = os.environ.get('GPT_ENGINE', 'gpt-4-turbo-2024-04-09')
-
-API = os.environ.get('API', None)
-API_URL = os.environ.get('API_URL', None)
-
-CLAUDE_API = os.environ.get('CLAUDE_API', None)
-GOOGLE_AI_API_KEY = os.environ.get('GOOGLE_AI_API_KEY', None)
-GROQ_API_KEY = os.environ.get('GROQ_API_KEY', None)
-
-current_date = datetime.now()
-Current_Date = current_date.strftime("%Y-%m-%d")
-
-message = "https://arxiv.org/abs/2404.02041 这篇论文讲了啥？"
-systemprompt = os.environ.get('SYSTEMPROMPT', prompt.chatgpt_system_prompt)
-# systemprompt = os.environ.get('SYSTEMPROMPT', prompt.system_prompt.format(LANGUAGE, Current_Date))
-# systemprompt = (
-#     "你是一位旅行专家。你可以规划旅行行程，如果用户有预算限制，还需要查询机票价格。结合用户的出行时间，给出合理的行程安排。"
-#     "在规划行程之前，必须先查找旅行攻略搜索景点信息，即使用 get_city_tarvel_info 查询景点信息。查询攻略后，你需要分析用户个性化需求给出合理的行程安排。充分考虑用户的年龄，情侣，家庭，朋友，儿童，独自旅行等情况。"
-#     "你需要根据用户给出的地点和预算，给出真实准确的行程，包括游玩时长、景点之间的交通方式和移动距离，每天都要给出总的游玩时间。"
-#     "给用户介绍景点的时候，根据查到的景点介绍结合你自己的知识，景点介绍尽量丰富精彩，吸引用户眼球，不要直接复述查到的景点介绍。"
-#     "尽量排满用户的行程，不要有太多空闲时间。"
-#     "你还可以根据用户的需求，给出一些旅行建议。"
-# )
-bot = chatgpt(api_key=API, api_url=API_URL , engine=GPT_ENGINE, system_prompt=systemprompt)
-# bot = claude3(api_key=CLAUDE_API, engine=GPT_ENGINE, system_prompt=systemprompt)
-# bot = gemini(api_key=GOOGLE_AI_API_KEY, engine=GPT_ENGINE, system_prompt=systemprompt)
-# bot = groq(api_key=GROQ_API_KEY, engine=GPT_ENGINE, system_prompt=systemprompt)
-for text in bot.ask_stream(message):
-# for text in bot.ask_stream("今天的微博热搜有哪些？"):
-# for text in bot.ask_stream("250m usd = cny"):
-# for text in bot.ask_stream("我在广州市，想周一去香港，周四早上回来，是去游玩，请你帮我规划整个行程。包括细节，如交通，住宿，餐饮，价格，等等，最好细节到每天各个部分的时间，花费，等等，尽量具体，用户一看就能直接执行的那种"):
-# for text in bot.ask_stream("英伟达最早支持杜比视界的显卡是哪一代"):
-# for text in bot.ask_stream("100个斐波纳切数列的和是多少"):
-# for text in bot.ask_stream("上海有哪些好玩的地方？"):
-# for text in bot.ask_stream("https://arxiv.org/abs/2404.02041 这篇论文讲了啥？"):
-# for text in bot.ask_stream("今天伊朗总统目前的情况怎么样？"):
-# for text in bot.ask_stream("我不是很懂y[..., 2]，y[..., 2] - y[:, 0:1, 0:1, 2]，y[:, 0:1, 0:1, 2]这些对张量的slice操作，给我一些练习demo代码，专门给我巩固这些张量复杂操作。让我从易到难理解透彻所有这样类型的张量操作。"):
-# for text in bot.ask_stream("just say test"):
-# for text in bot.ask_stream("画一只猫猫"):
-# for text in bot.ask_stream("我在上海想去重庆旅游，我只有2000元预算，我想在重庆玩一周，你能帮我规划一下吗？"):
-# for text in bot.ask_stream("我在上海想去重庆旅游，我有一天的时间。你能帮我规划一下吗？"):
-    print(text, end="")
-
-# print("\n bot tokens usage", bot.tokens_usage)