jarvis-ai-assistant 0.7.8__py3-none-any.whl → 1.0.2__py3-none-any.whl

jarvis/jarvis_utils/globals.py

@@ -7,12 +7,19 @@
 - 带有自定义主题的控制台配置
 - 环境初始化
 """
+
 import os
 
 # 全局变量：保存消息历史
-from typing import Any, Dict, List, Optional
+from typing import Any
+from typing import Dict
+from typing import List
+from typing import Optional
 
 message_history: List[str] = []
+
+# 全局模型组
+_global_model_group: Optional[str] = None
 MAX_HISTORY_SIZE = 50
 
 # 短期记忆存储
@@ -29,7 +36,8 @@ colorama.init()
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 # 全局代理管理
 global_agents: Dict[str, Any] = {}
-current_agent_name: str = ""
+# 使用栈结构跟踪嵌套的agent运行状态，支持agent调用其他agent的场景
+running_agent_stack: List[str] = []  # 正在运行的agent栈（最顶层是当前正在运行的agent）
 # 表示与大模型交互的深度(>0表示正在交互)
 g_in_chat: int = 0
 # 表示是否接收到中断信号
@@ -56,7 +64,12 @@ console = Console(theme=custom_theme)
 
 def make_agent_name(agent_name: str) -> str:
     """
-    通过附加后缀生成唯一的代理名称（如果必要）。
+    基于现有 global_agents 生成唯一的代理名称。
+
+    约定：
+    - 如果该名称尚未被占用，直接返回原始名称；
+    - 如果已存在同名 Agent，则按 `name_1`、`name_2`... 的形式依次查找可用名称；
+    - 已存在的带后缀名称会被一并考虑，例如 `name`、`name_1`、`name_3` 已存在时，返回 `name_2`。
 
     参数：
         agent_name: 基础代理名称
@@ -64,12 +77,17 @@ def make_agent_name(agent_name: str) -> str:
     返回：
         str: 唯一的代理名称
     """
-    if agent_name in global_agents:
-        i = 1
-        while f"{agent_name}_{i}" in global_agents:
-            i += 1
-        return f"{agent_name}_{i}"
-    return agent_name
+    # 如果名称未被占用，直接返回原始名称
+    if agent_name not in global_agents:
+        return agent_name
+
+    # 已存在同名 Agent，按 name_1、name_2... 形式查找第一个未被占用的名称
+    index = 1
+    while True:
+        candidate = f"{agent_name}_{index}"
+        if candidate not in global_agents:
+            return candidate
+        index += 1
 
 
 def get_agent(agent_name: str) -> Any:
@@ -85,44 +103,90 @@ def get_agent(agent_name: str) -> Any:
     return global_agents.get(agent_name)
 
 
-def set_agent(agent_name: str, agent: Any) -> None:
+def get_current_agent() -> Any:
+    """
+    获取当前正在运行的代理实例。
+
+    重要：只返回正在运行的agent，确保不会返回已完成的agent。
+    这保证返回的agent一定是正在运行中的，可以安全地修改其内部状态。
+
+    如果没有正在运行的agent，返回None（可以为空，但一定不能错）。
+
+    返回：
+        Any: 正在运行的代理实例，如果没有正在运行的agent则返回None
+    """
+    # 只返回正在运行的agent，确保不会返回已完成的agent
+    global running_agent_stack
+    if running_agent_stack:
+        current_agent_name = running_agent_stack[-1]
+        if current_agent_name in global_agents:
+            return global_agents.get(current_agent_name)
+    return None
+
+
+def get_current_agent_name() -> str:
+    """
+    获取当前代理名称（最后开始运行的agent，可能已完成）。
+    注意：这返回的是"最后开始运行的agent"，可能已经完成运行。
+
+    返回：
+        str: 当前代理名称，如果不存在则返回空字符串
     """
-    设置当前代理并将其添加到全局代理集合中。
+    agent = get_current_agent()
+    if agent:
+        return agent.name
+    return ""
+
+
+def set_current_agent(agent_name: str, agent: Any) -> None:
+    """
+    设置当前正在运行的代理名称（推入栈）。
+    应该在agent开始运行时调用。
+    支持嵌套调用：当agent调用其他agent时，状态会被正确保存。
+
+    注意：同一个agent可以多次运行（非嵌套），每次运行都会独立推入栈。
 
     参数：
         agent_name: 代理名称
-        agent: 代理对象
     """
+    global running_agent_stack, global_agents
     global_agents[agent_name] = agent
-    global current_agent_name
-    current_agent_name = agent_name
+    running_agent_stack.append(agent_name)
 
 
-def get_agent_list() -> str:
+def clear_current_agent() -> None:
     """
-    获取表示当前代理状态的格式化字符串。
+    清除正在运行的代理名称（从栈中弹出）。
+    应该在agent运行结束时调用。
+    支持嵌套调用：当子agent结束时，会恢复父agent的运行状态。
 
-    返回：
-        str: 包含代理数量和当前代理名称的格式化字符串
+    重要：只从栈顶移除agent，确保嵌套调用的正确性。
+    如果agent不在栈顶，说明状态不一致（可能是重复调用或异常），
+    会记录警告但不会修改栈，避免破坏嵌套结构。
+
+    参数：
+        agent_name: 要清除的代理名称（用于验证，确保清除的是正确的agent）
     """
-    return (
-        "[" + str(len(global_agents)) + "]" + current_agent_name
-        if global_agents
-        else ""
-    )
+    global running_agent_stack
+    if not running_agent_stack:
+        # 栈为空，可能是重复调用clear或状态不一致
+        return
+    current_agent_name = running_agent_stack.pop()
+    if current_agent_name in global_agents:
+        del global_agents[current_agent_name]
 
 
-def delete_agent(agent_name: str) -> None:
+def get_agent_list() -> str:
     """
-    从全局代理集合中删除一个代理。
+    获取表示当前代理状态的格式化字符串。
 
-    参数：
-        agent_name: 要删除的代理名称
+    返回：
+        str: 包含代理数量和当前代理名称的格式化字符串
     """
-    if agent_name in global_agents:
-        del global_agents[agent_name]
-        global current_agent_name
-        current_agent_name = ""
+    global global_agents
+    if not global_agents:
+        return ""
+    return "[" + str(len(global_agents)) + "]" + ", ".join(global_agents.keys())
 
 
 def set_in_chat(status: bool) -> None:
@@ -265,6 +329,25 @@ def clear_short_term_memories() -> None:
     short_term_memories.clear()
 
 
+def set_global_model_group(model_group: Optional[str]) -> None:
+    """设置全局模型组
+
+    参数:
+        model_group: 模型组名称
+    """
+    global _global_model_group
+    _global_model_group = model_group
+
+
+def get_global_model_group() -> Optional[str]:
+    """获取全局模型组
+
+    返回:
+        Optional[str]: 全局模型组名称，如果未设置则返回None
+    """
+    return _global_model_group
+
+
 def get_all_memory_tags() -> Dict[str, List[str]]:
     """
     获取所有记忆类型中的标签集合。
@@ -273,9 +356,10 @@ def get_all_memory_tags() -> Dict[str, List[str]]:
     返回:
         Dict[str, List[str]]: 按记忆类型分组的标签列表
     """
-    from pathlib import Path
     import json
     import random
+    from pathlib import Path
+
     from jarvis.jarvis_utils.config import get_data_dir
 
     tags_by_type: Dict[str, List[str]] = {

jarvis/jarvis_utils/http.py

@@ -1,7 +1,11 @@
 # -*- coding: utf-8 -*-
 
-import requests  # type: ignore[import-untyped]
-from typing import Any, Dict, Optional, Generator
+from typing import Any
+from typing import Dict
+from typing import Generator
+from typing import Optional
+
+import requests
 
 
 def get_requests_session() -> requests.Session: