@brandon_9527/tcode 1.0.2 → 1.0.6

package/dist/python-src/entry.py

@@ -5,11 +5,80 @@
 # 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")
+
+
+import typer
+import asyncio
+
+
 from main import (
     agent_ui, 
     asingle_agent,
+    team_main,
+    run_once,
+    teminal_chat
 )
 
+app = typer.Typer()
+interact_app = typer.Typer()
+app.add_typer(interact_app)
+
+
+@app.callback(invoke_without_command=True, no_args_is_help=False)
+def root(
+    ctx: typer.Context,
+    prompt: str = typer.Option(None, "-p", "--prompt", help="用户输入请求内容"),
+    verbose: bool = typer.Option(False, "-v", "--verbose", help="显示agent执行详细过程"),
+    mode: str = typer.Option("team", "-m", "--mode", help="运行模式，single或team"),
+):
+    if ctx.invoked_subcommand is None:
+        if not prompt:
+            if mode == "single":
+                asyncio.run(asingle_agent())
+            elif mode == "team":
+                asyncio.run(team_main())
+            
+        # 同步环境执行异步任务
+        asyncio.run(run_once(prompt, verbose))
+        # typer.echo(result)
+        # raise typer.Exit()
+    
+
+@interact_app.command("interact")
+def interact_mode(
+    verbose: bool = typer.Option(False, "-v", "--verbose", help="显示agent执行详细过程"),
+):
+    # 执行交互模式
+    asyncio.run(teminal_chat(verbose))
+
+
+
 if __name__ == "__main__":
-    import asyncio
-    asyncio.run(asingle_agent())
+    app()
+    # import asyncio
+    
+    # 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,446 @@ 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
     
-    await tui.run_async()
+    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()
+    
+    
+def build_team():
+    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 src.middlewares.dynamic_content import DynamicContentMiddleware
+
+    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.setdefault(domain, {"members":{}})
+        
+        # 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
+            # ),
+            middleware=[
+                DynamicContentMiddleware(),
+            ],
+            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
+    )
+    
+    return agent, context, instruction_manager
+    
+    
+async def teminal_chat(verbose=True):
+    """ """
+    from langchain_core.messages import HumanMessage
+    from src.stream.handler import ainput, anormal_handler
+    
+    agent, context, instruction_manager = build_team()
+    
+    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
+            
+            
+            print(f"【new_query】:\n {new_query}")
+            
+            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=verbose)
+            
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+    finally:
+        pass
+            
+            
+async def run_once(prompt:str = None, verbose: bool=True):
+    from langchain_core.messages import HumanMessage
+    from src.stream.handler import ainput, anormal_handler
+    
+    agent, context, instruction_manager = build_team()
+    
+    query = prompt
+    
+    try:
+        if query == "/comands":
+            for instruction in instruction_manager.list_instructions():
+                # print(instruction)
+                print(f"command: {instruction.name}, description: {instruction.description}")
+        
+        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
+        
+        
+        if verbose:
+            print(f"user> {new_query}")
+        
+        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=verbose)
+        
+    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,8 @@ dependencies = [
   "dashscope>=1.25.12",
   "langchain-core==1.2.13",
   "python-minifier>=3.2.0",
+  "ddgs>=9.11.4",
+  "typer>=0.24.1",
 ]
 
 #[project.scripts]
@@ -56,7 +57,7 @@ skip = [
     "resources",
     "_workspace",
     ".autodev",
-    "tools"
+    "tools",
 ]
 
 exclude = [

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

package/dist/python-src/src/core/deepagents.py

@@ -20,7 +20,7 @@ from src.middlewares.hitl import HumanInTheLoopMiddleware
 from dotenv import find_dotenv,load_dotenv
 import os
 _=load_dotenv(find_dotenv())
-def get_default_model(streaming=False):return ChatOpenAI(model_name=os.getenv('DEFAULT_MODEL'),base_url=os.getenv('OPENAI_API_BASE'),api_key=os.getenv('OPENAI_API_KEY'),streaming=streaming,model_kwargs={'stream_options':{'include_usage':True}})
+def get_default_model(streaming=False):D='OPENAI_API_KEY';C='OPENAI_API_BASE';B='DEFAULT_MODEL';A=streaming;return ChatOpenAI(model_name=os.getenv(B),base_url=os.getenv(C),api_key=os.getenv(D),streaming=A,model_kwargs={'stream_options':{'include_usage':True}})if A else ChatOpenAI(model_name=os.getenv(B),base_url=os.getenv(C),api_key=os.getenv(D),streaming=A)
 def create_deep_agent(model=_A,tools=_A,*,system_prompt=_A,middleware=(),subagents=_A,response_format=_A,context_schema=_A,checkpointer=_A,store=_A,interrupt_on=_A,debug=False,name=_A,cache=_A,recursion_limit=1000):
 	G=subagents;F=middleware;E=system_prompt;D=tools;B=interrupt_on;A=model
 	if A is _A:A=get_default_model(streaming=True)