CometAgent./XCometAgent.PY at Master · CUI-hyg/CometAgent. · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import os
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).parent))
from CometLoop.MetaAgent import MetaAgent

class CometAgent:
    def __init__(self):
        self.current_dir = os.path.dirname(os.path.abspath(__file__))

    def print_orange(self, text):
        print(f"\033[38;2;255;140;0m{text}\033[0m")

    def print_dim(self, text):
        print(f"\033[90m{text}\033[0m")

    def welcome(self):
        self.print_orange("\n┌" + "─" * 78)
        self.print_orange("│")
        self.print_orange("│    CometAgent - 自由、高效的Agent")
        self.print_orange("│")
        self.print_orange("│  Version 0.1.23-R1 - 2025.10.10-BuildD1")
        self.print_orange("│")
        self.print_orange("└" + "─" * 78)

    def display_help(self):
        self.print_orange("\n┌" + "─" * 50)
        self.print_orange("│  可用命令")
        self.print_orange("├" + "─" * 50)
        self.print_orange("│  help    - 显示此帮助信息")
        self.print_orange("│  normal  - 使用普通模式运行Agent")
        self.print_orange("│  quit    - 退出系统")
        self.print_orange("│")
        self.print_orange("│  使用示例:")
        self.print_orange("│  • 直接输入问题 - 使用Agent模式")
        self.print_orange("│  • normal 问题 - 使用普通模式")
        self.print_orange("└" + "─" * 50)

    def run(self):
        self.welcome()
        self.display_help()
        self.print_dim("输入 'help' 查看命令；'quit' 退出")
        while True:
            try:
                self.print_orange("\n┌" + "─" * 50)
                user_input = input("│ 你> ").strip()
                self.print_orange("└" + "─" * 50)
                if not user_input:
                    continue
                if user_input.lower() in ['quit', 'exit', 'q']:
                    self.print_orange("\nCometAgent 已安全退出")
                    break
                if user_input.lower() == 'help':
                    self.display_help()
                    continue
                parts = user_input.split(maxsplit=1)
                if len(parts) == 1:
                    query = parts[0]
                    asys = "yes"
                else:
                    command, query = parts if len(parts) == 2 else (parts[0], "")
                    if command.lower() == 'normal':
                        asys = "no"
                    else:
                        query = user_input
                        asys = "yes"
                if not query or not query.strip():
                    self.print_orange("请输入有效的问题！")
                    continue
                self.print_orange("\n正在处理...")
                self.print_dim(f"   问题: {query}")
                try:
                    ma = MetaAgent(query)
                    result = ma.run()
                    self.print_orange("\n处理完成!")
                    out = result.get("final") if isinstance(result, dict) else result
                    if out:
                        print(out)
                except Exception as e:
                    self.print_orange("\n处理出错")
                    self.print_dim(str(e))
            except KeyboardInterrupt:
                self.print_orange("\n\n用户中断")
                self.print_orange("CometAgent 已安全退出")
                break
            except Exception as e:
                self.print_orange(f"\n发生错误: {str(e)}")
                self.print_orange("输入 'help' 获取帮助")

def main():
    capp = CometAgent()
    capp.run()

if __name__ == "__main__":
    main()