这是之前做的一个小项目,主要是用LangGraph实现了一个简单的聊天机器人。它能够记住之前的对话内容,这样聊天就能保持连贯性了。
我把核心代码放在了src/agent/graph.py里,还算比较简单明了。基本上就是接收用户消息,然后根据历史记录生成回复。
简单来说就是:
- 接收用户消息
- 记住我们的对话历史
- 根据上下文生成回复
- 把新的对话加入历史记录中
代码结构很简单,我觉得以后要是想加功能也挺方便的。
前提是你已经装了LangGraph Studio,然后:
- 复制个.env文件
- 把需要的API密钥填进去
- 按需修改一下代码
- 用LangGraph Studio打开就可以了
如果你想改动的话,主要有这几个地方:
- 系统提示词在configuration.py里,可以改一下让机器人有不同的性格
- 默认用的是Claude 3 Sonnet,不过也可以换成别的,比如gpt-4什么的
- 如果想改对话流程,就去修改graph.py
我还想过以后可以加些功能,比如:
- 加点自定义工具让它能做更多事
- 加点逻辑处理特定的问题
- 可能再接入一些外部API