https://github.com/wangle201210/chat-history
chat-history
https://github.com/wangle201210/chat-history
chat-history chat-history-database eino llm
Last synced: 17 days ago
JSON representation
chat-history
- Host: GitHub
- URL: https://github.com/wangle201210/chat-history
- Owner: wangle201210
- License: apache-2.0
- Created: 2025-04-02T07:28:09.000Z (22 days ago)
- Default Branch: main
- Last Pushed: 2025-04-02T10:47:26.000Z (21 days ago)
- Last Synced: 2025-04-02T11:26:34.983Z (21 days ago)
- Topics: chat-history, chat-history-database, eino, llm
- Language: Go
- Homepage:
- Size: 21.5 KB
- Stars: 2
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# chat-history
聊天历史记录管理
## 为什么需要
目前eino提供了快捷接入大模型的方法但是聊天历史记录的功能没有提供，导致没法进行对轮对话，及对话管理
## 怎么接入
可以参照 [example](./example)
1. 安装依赖 `go get github.com/wangle201210/chat-history`
2. 初始化 EinoHistory
```go
var eh = eino.NewEinoHistory("root:123456@tcp(127.0.0.1:3306)/chat_history")
```
3. 在原来获取messages的地方添加 add start -- add end 之间的代码
```go
func createMessagesFromTemplate(ctx context.Context, convID, question string) (messages []*schema.Message, err error) {
template := createTemplate(ctx)
/* add start */
chatHistory, err := eh.GetHistory(convID, 100)
if err != nil {
return
}
// 插入一条用户数据
err = eh.SaveMessage(&schema.Message{
Role: schema.User,
Content: question,
}, convID)
if err != nil {
return
}
/* add end */
// 使用模板生成消息
messages, err = template.Format(context.Background(), map[string]any{
"role": "程序员鼓励师",
"style": "积极、温暖且专业",
"question": question,
"chat_history": chatHistory,
})
if err != nil {
return
}
return
}
```
4. 在返回消息的地方添加 add start -- add end 之间的代码
```go
messages, err := createMessagesFromTemplate(ctx, convID, s)
if err != nil {
log.Fatalf("create messages failed: %v", err)
return
}
result := generate(ctx, cm, messages)
/* add start */
err = eh.SaveMessage(result, convID)
if err != nil {
log.Fatalf("save assistant message err: %v", err)
return
}
/* add end */
log.Printf("result: %+v\n\n", result)
```