@brandon_9527/tcode 1.0.2 → 1.0.3

package/dist/python-src/entry.py

@@ -5,11 +5,38 @@
 # if __name__ == "__main__":
 #     main()
 
+
+# ===================== 强制修复 Git Bash 报错：NoConsoleScreenBufferError =====================
+import os
+import sys
+
+# 🔥 核心：强制让 prompt_toolkit 使用兼容模式，彻底关闭 Windows 控制台检测
+os.environ["PROMPT_TOOLKIT_NO_WIN32"] = "1"
+os.environ["PROMPT_TOOLKIT_BASIC_OUTPUT"] = "1"
+os.environ["PROMPT_TOOLKIT_FORCE_VT100"] = "1"
+
+# 🔥 强制设置终端类型，让程序认为是 Linux 终端
+if sys.platform == "win32":
+    os.environ["TERM"] = "xterm-256color"
+
+# ===================== 下面才是你原来的代码 =====================
+
+
+import warnings
+warnings.filterwarnings("ignore", category=DeprecationWarning)
+warnings.filterwarnings("ignore", message="Pydantic serializer warnings")
+
 from main import (
     agent_ui, 
     asingle_agent,
+    team_main,
 )
 
 if __name__ == "__main__":
     import asyncio
-    asyncio.run(asingle_agent())
+    
+    try:
+        asyncio.run(asingle_agent())
+        # asyncio.run(team_main())
+    except Exception as e:
+        pass

package/dist/python-src/main.py

@@ -19,6 +19,12 @@ _ = load_dotenv(find_dotenv())
 from langchain_core.messages import AIMessageChunk
 
 
+import warnings
+warnings.filterwarnings("ignore", category=DeprecationWarning)
+warnings.filterwarnings("ignore", message="Pydantic serializer warnings")
+
+
+
 
 # # ========== 关键修改1：配置logging，让INFO级别日志能输出 ==========
 # import logging
@@ -214,15 +220,388 @@ async def asingle_agent():
         # sandbox=None
     )
     
+    tui = LiveChatUI(
+        agent=agent, 
+        saver=saver, 
+        workspace=WORKSPACE, 
+        context=context,
+        interrupt_tools=[
+            "bash",
+            "read_file",
+            "write_file",
+            "glob",
+            "grep",
+            "edit",
+            "list_dir",
+            "web_search",
+            "web_fetch"   
+        ],
+    )
+    
+    await tui.run_async()
     
     
-    tui = LiveChatUI(agent=agent, saver=saver, workspace=WORKSPACE, context=context)
+async def team_main():
+    """ 加载开发团队 """
+    from dataclasses import dataclass, field
+    from pathlib import Path
+    
+    from src.core.deepagents import create_deep_agent, get_default_model
+    from src.managers.manager_context import ContextManager
+    from src.utils.prompt import apply_template, apply_prompt
+    from src.managers.sandbox import Container
+    from src.prompts.prompts import leader
+    
+    from src.tools.tools import (
+        SkillAgentContext,
+        shell, 
+        bash, 
+        write_file,
+        read_file, 
+        list_dir,
+        glob, 
+        grep, 
+        edit
+    )
+    
+    from src.tools.web import (
+        web_search,
+        web_fetch
+    )
+    
+    from langgraph.checkpoint.memory import MemorySaver
+    from langchain_core.tools import BaseTool
+    
+    
+    def _prepare(workspace, run_mode, session_id):
+        saver = MemorySaver()
+        
+        # 1. 构建上下文管理器
+        context_manager = ContextManager(
+            dockerfile_path=os.path.join(os.getcwd(), "sources", "docker", "Dockerfile.sandbox"),
+            mode=run_mode
+        )
+        
+        context_manager.create_environment(
+            session_id=session_id,
+            workspace_path=workspace,
+        )
+        
+        # 2. 指令管理器
+        instruction_manager = context_manager.get_instruction_manager(
+            session_id=session_id
+        )
+        
+        # 3. 代理管理器
+        agent_manager = context_manager.get_agent_manager(
+            session_id=session_id
+        )
+        
+        # 4. mcp 管理器
+        # mcp_manager = context_manager.get_mcp_manager(
+        #     session_id=session_id
+        # )
+        
+        # 5. 沙箱
+        sandbox = context_manager.get_container(
+            session_id=session_id
+        )
+    
+        # 6. 获取配置
+        agents_conf = agent_manager.get_all_conf()
+        
+        toolkits = {
+            "filetools": [ shell, read_file, write_file, glob, grep, edit, list_dir],
+            "web": [web_search, web_fetch]
+        } if run_mode == "sandbox" else {
+            "filetools": [ bash, read_file, write_file, glob, grep, edit, list_dir],
+            "web": [web_search, web_fetch]
+        }
+        
+        context = SkillAgentContext(
+            working_directory=workspace,
+            # skill_loader=None,
+            sandbox=sandbox
+        )
+        
+        return agents_conf, toolkits, context, saver, instruction_manager
+    
+    def _build_team(agents_conf: Dict[str, Any], domain: str, llm: ChatOpenAI, toolkits: Dict[str, List[BaseTool]], workspace: str, saver=None, store=None, recursion_limit=1000):
+        """ """
+        team_conf = agents_conf[domain]
+        
+        # for mname, mconf in team_conf['members'].items():
+        #     print(f"member_name: {mname}, member: {mconf}")
+            
+        agents = []
+        for member_name, member in team_conf["members"].items():
+            print(f"member_name: {member_name}, member: {member}")
+            
+            tools = []
+            for tool_name in member["tools"]:
+                tools.extend(toolkits.get(tool_name, []))
+                
+            llm_conf = member.get("llm", None)
+            model = ChatOpenAI(**llm_conf) if llm_conf else get_default_model(streaming=False)
+            
+            agent = {
+                "name": member_name,
+                "description": member["description"],
+                "system_prompt": apply_template(
+                    agents_config=agents_conf,
+                    domain=domain,
+                    prompt_name=member_name,
+                    WORKSPACE=workspace
+                ),
+                "context_schema": SkillAgentContext,
+                "tools": tools,
+                "model": model
+            }
+            
+            agents.append(agent)
+            
+        tools = []
+        for tool_name in ["filetools", "web"]:
+            tools.extend([tool for tool in toolkits.get(tool_name, [])])
+            
+        team = create_deep_agent(
+            model=llm,
+            subagents=agents,
+            tools=tools,
+            context_schema=SkillAgentContext,
+            # system_prompt=apply_template(
+            #     agents_config=agents_conf,
+            #     domain="default",
+            #     prompt_name="leader_",
+            #     WORKSPACE=workspace
+            # ),
+            system_prompt=apply_prompt(leader, WORKSPACE=workspace),
+            checkpointer=saver
+        ).with_config({"recursion_limit": recursion_limit})
+        
+        return team
+    
+    # workspace = os.path.join(os.getcwd(), "workspace")
+    workspace = os.getcwd()
+    run_mode = "local"
+    session_id = "1"
+    
+    # agents_conf, toolkits, context, saver, instruction_manager
+    agents_config, toolkits, context, saver, instruction_manager = _prepare(workspace, run_mode, session_id)   
+    agent = _build_team(
+        agents_conf=agents_config,
+        domain="coding",
+        llm=llm,
+        toolkits=toolkits,
+        workspace=workspace,
+        saver=saver,
+        store=None,
+        recursion_limit=1000
+    )
+    
+    ui = LiveChatUI(agent=agent, saver=saver, workspace=workspace, context=context, instruction_manager=instruction_manager)
+    
+    await ui.run_async()
     
-    await tui.run_async()
     
+async def teminal_chat():
+    """ """
+    from dataclasses import dataclass, field
+    from pathlib import Path
+    
+    from src.core.deepagents import create_deep_agent, get_default_model
+    from src.managers.manager_context import ContextManager
+    from src.utils.prompt import apply_template, apply_prompt
+    from src.managers.sandbox import Container
+    from src.prompts.prompts import leader
+    from src.stream.handler import ainput, anormal_handler
+    
+    from src.tools.tools import (
+        SkillAgentContext,
+        shell, 
+        bash, 
+        write_file,
+        read_file, 
+        list_dir,
+        glob, 
+        grep, 
+        edit
+    )
+    
+    from src.tools.web import (
+        web_search,
+        web_fetch
+    )
+    
+    from langgraph.checkpoint.memory import MemorySaver
+    from langchain_core.messages import HumanMessage
+    from langchain_core.tools import BaseTool
+    
+    
+    def _prepare(workspace, run_mode, session_id):
+        saver = MemorySaver()
+        
+        # 1. 构建上下文管理器
+        context_manager = ContextManager(
+            dockerfile_path=os.path.join(os.getcwd(), "sources", "docker", "Dockerfile.sandbox"),
+            mode=run_mode
+        )
+        
+        context_manager.create_environment(
+            session_id=session_id,
+            workspace_path=workspace,
+        )
+        
+        # 2. 指令管理器
+        instruction_manager = context_manager.get_instruction_manager(
+            session_id=session_id
+        )
+        
+        # 3. 代理管理器
+        agent_manager = context_manager.get_agent_manager(
+            session_id=session_id
+        )
+        
+        # 4. mcp 管理器
+        # mcp_manager = context_manager.get_mcp_manager(
+        #     session_id=session_id
+        # )
+        
+        # 5. 沙箱
+        sandbox = context_manager.get_container(
+            session_id=session_id
+        )
+    
+        # 6. 获取配置
+        agents_conf = agent_manager.get_all_conf()
+        
+        toolkits = {
+            "filetools": [ shell, read_file, write_file, glob, grep, edit, list_dir],
+            "web": [web_search, web_fetch]
+        } if run_mode == "sandbox" else {
+            "filetools": [ bash, read_file, write_file, glob, grep, edit, list_dir],
+            "web": [web_search, web_fetch]
+        }
+        
+        context = SkillAgentContext(
+            working_directory=workspace,
+            # skill_loader=None,
+            sandbox=sandbox
+        )
+        
+        return agents_conf, toolkits, context, saver, instruction_manager
+    
+    def _build_team(agents_conf: Dict[str, Any], domain: str, llm: ChatOpenAI, toolkits: Dict[str, List[BaseTool]], workspace: str, saver=None, store=None, recursion_limit=1000):
+        """ """
+        team_conf = agents_conf[domain]
+        
+        # for mname, mconf in team_conf['members'].items():
+        #     print(f"member_name: {mname}, member: {mconf}")
+            
+        agents = []
+        for member_name, member in team_conf["members"].items():
+            print(f"member_name: {member_name}, member: {member}")
+            
+            tools = []
+            for tool_name in member["tools"]:
+                tools.extend(toolkits.get(tool_name, []))
+                
+            llm_conf = member.get("llm", None)
+            model = ChatOpenAI(**llm_conf) if llm_conf else get_default_model(streaming=False)
+            
+            agent = {
+                "name": member_name,
+                "description": member["description"],
+                "system_prompt": apply_template(
+                    agents_config=agents_conf,
+                    domain=domain,
+                    prompt_name=member_name,
+                    WORKSPACE=workspace
+                ),
+                "context_schema": SkillAgentContext,
+                "tools": tools,
+                "model": model
+            }
+            
+            agents.append(agent)
+            
+        tools = []
+        for tool_name in ["filetools", "web"]:
+            tools.extend([tool for tool in toolkits.get(tool_name, [])])
+            
+        team = create_deep_agent(
+            model=llm,
+            subagents=agents,
+            tools=tools,
+            context_schema=SkillAgentContext,
+            # system_prompt=apply_template(
+            #     agents_config=agents_conf,
+            #     domain="default",
+            #     prompt_name="leader_",
+            #     WORKSPACE=workspace
+            # ),
+            system_prompt=apply_prompt(leader, WORKSPACE=workspace),
+            checkpointer=saver
+        ).with_config({"recursion_limit": recursion_limit})
+        
+        return team
+    
+    # workspace = os.path.join(os.getcwd(), "workspace")
+    workspace = os.getcwd()
+    run_mode = "local"
+    session_id = "1"
+    
+    # agents_conf, toolkits, context, saver, instruction_manager
+    agents_config, toolkits, context, saver, instruction_manager = _prepare(workspace, run_mode, session_id)   
+    agent = _build_team(
+        agents_conf=agents_config,
+        domain="coding",
+        llm=llm,
+        toolkits=toolkits,
+        workspace=workspace,
+        saver=saver,
+        store=None,
+        recursion_limit=1000
+    )
+    
+    
+    try:
+        while True:
+            query = await ainput("user>")
+            
+            if query == "/comands":
+                for instruction in instruction_manager.list_instructions():
+                    # print(instruction)
+                    print(f"command: {instruction.name}, description: {instruction.description}")
+                continue
+            
+            instruction_result = instruction_manager.parse(query)
+            executed_instruction, message = instruction_result["executed_instruction"], instruction_result['message']
+            
+            new_query = f"\n [注意]: 执行用户请求必须严格遵循如下准则:\n{executed_instruction}\n\n 用户请求:\n{message}" if executed_instruction else message
+            
+            stream = agent.astream(
+                {
+                    "messages":[HumanMessage(content=new_query)]
+                },
+                config = {"configurable": {"thread_id": 1}},
+                stream_mode=["updates", "custom"],
+                context=context
+            ) # .with_config({"recursion_limit": 1000})
+            
+            await anormal_handler(stream, detail=True)
+            
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+    finally:
+        pass
+            
+        
    
 if __name__ == "__main__":
-    pass
+    # pass
+    asyncio.run(teminal_chat())
     # asyncio.run(agent_ui())
     # asyncio.run(team_main())
     # asyncio.run(asingle_agent())

package/dist/python-src/pyproject.toml

@@ -23,7 +23,6 @@ dependencies = [
   "croniter>=6.0.0",
   "apscheduler>=3.11.2",
   "whatthepatch>=1.0.7",
-  "ddgs>=9.10.0",
   "requests>=2.32.5",
   "beautifulsoup4>=4.14.3",
   "langchain-dashscope>=0.1.8",
@@ -32,6 +31,7 @@ dependencies = [
   "dashscope>=1.25.12",
   "langchain-core==1.2.13",
   "python-minifier>=3.2.0",
+  "ddgs>=9.11.4",
 ]
 
 #[project.scripts]

package/dist/python-src/src/ai_tcode.egg-info/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.4
+Name: ai-tcode
+Version: 1.0.0
+Requires-Python: >=3.11
+Requires-Dist: prompt-toolkit>=3.0.43
+Requires-Dist: rich>=13.0.0
+Requires-Dist: pyyaml>=6.0.0
+Requires-Dist: click>=8.0.0
+Requires-Dist: numpy>=1.24.0
+Requires-Dist: python-dateutil>=2.8.0
+Requires-Dist: langchain>=1.0.0
+Requires-Dist: langchain-openai>=1.0.0
+Requires-Dist: langgraph>=1.0.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: python-dotenv>=1.0
+Requires-Dist: lark-oapi>=1.5.3
+Requires-Dist: langchain-mcp-adapters>=0.2.1
+Requires-Dist: docker>=7.1.0
+Requires-Dist: croniter>=6.0.0
+Requires-Dist: apscheduler>=3.11.2
+Requires-Dist: whatthepatch>=1.0.7
+Requires-Dist: ddgs>=9.10.0
+Requires-Dist: requests>=2.32.5
+Requires-Dist: beautifulsoup4>=4.14.3
+Requires-Dist: langchain-dashscope>=0.1.8
+Requires-Dist: langchain-alibaba>=0.1.1
+Requires-Dist: langchain-community>=0.4.1
+Requires-Dist: dashscope>=1.25.12
+Requires-Dist: langchain-core==1.2.13
+Requires-Dist: python-minifier>=3.2.0

package/dist/python-src/src/ai_tcode.egg-info/SOURCES.txt

@@ -0,0 +1,48 @@
+README.md
+pyproject.toml
+src/__init__.py
+src/agents/__init__.py
+src/ai_tcode.egg-info/PKG-INFO
+src/ai_tcode.egg-info/SOURCES.txt
+src/ai_tcode.egg-info/dependency_links.txt
+src/ai_tcode.egg-info/requires.txt
+src/ai_tcode.egg-info/top_level.txt
+src/core/__init__.py
+src/core/deepagents.py
+src/core/prompts.py
+src/managers/__init__.py
+src/managers/sandbox.py
+src/middlewares/__init__.py
+src/middlewares/dynamic_prompt.py
+src/middlewares/hitl.py
+src/middlewares/patch_tool_calls.py
+src/middlewares/subagents.py
+src/middlewares/summary.py
+src/stream/__init__.py
+src/stream/emitter.py
+src/stream/file_write_parser.py
+src/stream/formatter.py
+src/stream/handler.py
+src/stream/handler_with_tracker.py
+src/stream/tracker.py
+src/stream/utils.py
+src/tools/__init__.py
+src/tools/security_hook.py
+src/tools/skill_loader.py
+src/tools/tools.py
+src/tools/tools_.py
+src/tools/web.py
+src/tui/chatui.py
+src/tui/config.py
+src/tui/components/__init__.py
+src/tui/components/live_spinner.py
+src/tui/components/tdiff.py
+src/tui/components/tdisplay.py
+src/tui/components/tlist.py
+src/tui/components/tscroll_panel.py
+src/tui/utils/__init__.py
+src/tui/utils/render.py
+src/tui/utils/trender.py
+src/utils/__init__.py
+src/utils/prompt.py
+src/utils/utils.py

package/dist/python-src/src/ai_tcode.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

package/dist/python-src/src/ai_tcode.egg-info/requires.txt

@@ -0,0 +1,26 @@
+prompt-toolkit>=3.0.43
+rich>=13.0.0
+pyyaml>=6.0.0
+click>=8.0.0
+numpy>=1.24.0
+python-dateutil>=2.8.0
+langchain>=1.0.0
+langchain-openai>=1.0.0
+langgraph>=1.0.0
+pyyaml>=6.0
+python-dotenv>=1.0
+lark-oapi>=1.5.3
+langchain-mcp-adapters>=0.2.1
+docker>=7.1.0
+croniter>=6.0.0
+apscheduler>=3.11.2
+whatthepatch>=1.0.7
+ddgs>=9.10.0
+requests>=2.32.5
+beautifulsoup4>=4.14.3
+langchain-dashscope>=0.1.8
+langchain-alibaba>=0.1.1
+langchain-community>=0.4.1
+dashscope>=1.25.12
+langchain-core==1.2.13
+python-minifier>=3.2.0

package/dist/python-src/src/ai_tcode.egg-info/top_level.txt

@@ -0,0 +1,9 @@
+__init__
+agents
+core
+managers
+middlewares
+stream
+tools
+tui
+utils