611de commited on
Commit
d5146af
·
1 Parent(s): 05b4dd6

feat: Add initial implementation of chatbot and mcp tool with Gradio interface and necessary dependencies

Browse files
Files changed (5) hide show
  1. .gitignore +194 -0
  2. agent.py +27 -0
  3. app.py +59 -0
  4. requirements.txt +10 -0
  5. utils.py +0 -0
.gitignore ADDED
@@ -0,0 +1,194 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # UV
98
+ # Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ #uv.lock
102
+
103
+ # poetry
104
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
105
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
106
+ # commonly ignored for libraries.
107
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
108
+ #poetry.lock
109
+
110
+ # pdm
111
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
112
+ #pdm.lock
113
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
114
+ # in version control.
115
+ # https://pdm.fming.dev/latest/usage/project/#working-with-version-control
116
+ .pdm.toml
117
+ .pdm-python
118
+ .pdm-build/
119
+
120
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
121
+ __pypackages__/
122
+
123
+ # Celery stuff
124
+ celerybeat-schedule
125
+ celerybeat.pid
126
+
127
+ # SageMath parsed files
128
+ *.sage.py
129
+
130
+ # Environments
131
+ .env
132
+ .venv
133
+ env/
134
+ venv/
135
+ ENV/
136
+ env.bak/
137
+ venv.bak/
138
+
139
+ # Spyder project settings
140
+ .spyderproject
141
+ .spyproject
142
+
143
+ # Rope project settings
144
+ .ropeproject
145
+
146
+ # mkdocs documentation
147
+ /site
148
+
149
+ # mypy
150
+ .mypy_cache/
151
+ .dmypy.json
152
+ dmypy.json
153
+
154
+ # Pyre type checker
155
+ .pyre/
156
+
157
+ # pytype static type analyzer
158
+ .pytype/
159
+
160
+ # Cython debug symbols
161
+ cython_debug/
162
+
163
+ # PyCharm
164
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
165
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
166
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
167
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
168
+ #.idea/
169
+
170
+ # Abstra
171
+ # Abstra is an AI-powered process automation framework.
172
+ # Ignore directories containing user credentials, local state, and settings.
173
+ # Learn more at https://abstra.io/docs
174
+ .abstra/
175
+
176
+ # Visual Studio Code
177
+ # Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore
178
+ # that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
179
+ # and can be added to the global gitignore or merged into this file. However, if you prefer,
180
+ # you could uncomment the following to ignore the enitre vscode folder
181
+ # .vscode/
182
+
183
+ # Ruff stuff:
184
+ .ruff_cache/
185
+
186
+ # PyPI configuration file
187
+ .pypirc
188
+
189
+ # Cursor
190
+ # Cursor is an AI-powered code editor. `.cursorignore` specifies files/directories to
191
+ # exclude from AI features like autocomplete and code analysis. Recommended for sensitive data
192
+ # refer to https://docs.cursor.com/context/ignore-files
193
+ .cursorignore
194
+ .cursorindexingignore
agent.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_mcp_adapters.client import MultiServerMCPClient
2
+ from langgraph.prebuilt import create_react_agent
3
+ from langchain_community.chat_models import ChatTongyi
4
+
5
+ llm = ChatTongyi(model_name="qwen-turbo")
6
+
7
+ client = MultiServerMCPClient(
8
+ {
9
+ "filesystem": {
10
+ "disabled": False,
11
+ "timeout": 60,
12
+ "command": "npx",
13
+ "args": [
14
+ "-y",
15
+ "@modelcontextprotocol/server-filesystem",
16
+ "/root/project/python_test",
17
+ ],
18
+ "transport": "stdio",
19
+ }
20
+ }
21
+ )
22
+
23
+
24
+ async def get_agent():
25
+ tools = await client.get_tools()
26
+ agent = create_react_agent(llm, tools)
27
+ return agent
app.py ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+ from langchain.schema import HumanMessage, AIMessage
4
+
5
+ from agent import get_agent
6
+
7
+ # 获取模型回复
8
+ agent = None
9
+
10
+ async def initialize_agent_once():
11
+ global agent
12
+ if agent is None:
13
+ agent = await get_agent()
14
+ return agent
15
+
16
+
17
+ async def respond(message, chat_history):
18
+
19
+ messages = []
20
+ for human_msg, ai_msg in chat_history:
21
+ messages.append(HumanMessage(content=human_msg))
22
+ messages.append(AIMessage(content=ai_msg))
23
+ messages.append(HumanMessage(content=message))
24
+
25
+ agent = await initialize_agent_once()
26
+ response = await agent.ainvoke({"messages": messages})
27
+
28
+ chat_history.append((message, response['messages'][-1].content))
29
+ return "", chat_history
30
+
31
+ with gr.Blocks() as app:
32
+ gr.Markdown("# 旅游攻略助手")
33
+
34
+ with gr.Row():
35
+ with gr.Column(scale=3):
36
+ chatbot = gr.Chatbot(height=500)
37
+ with gr.Column(scale=1):
38
+ gr.Markdown("### 说明\n"
39
+ "- 请在下方输入框中输入您的问题\n"
40
+ "- 点击发送或按Enter键提交\n"
41
+ "- 支持多轮对话")
42
+
43
+ with gr.Row():
44
+ msg = gr.Textbox(
45
+ label="您的问题",
46
+ placeholder="请输入您的旅游需求,例如:推荐北京的景点并规划一日游路线...",
47
+ lines=3
48
+ )
49
+
50
+ with gr.Row():
51
+ submit_btn = gr.Button("发送", variant="primary")
52
+ clear_btn = gr.Button("清除")
53
+
54
+ # 设置事件处理
55
+ submit_btn.click(respond, [msg, chatbot], [msg, chatbot])
56
+ msg.submit(respond, [msg, chatbot], [msg, chatbot])
57
+ clear_btn.click(lambda: None, None, chatbot, queue=False)
58
+
59
+ app.launch()
requirements.txt ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ # requirements.txt
2
+ langchain==0.3.25
3
+ openai==0.27.8
4
+ gradio
5
+ requests==2.31.0
6
+ langchain-community==0.3.25
7
+ langchain-community
8
+ langchain-mcp-adapters
9
+ langgraph
10
+ dashscope
utils.py ADDED
File without changes