fumiama/go-onebot-agent
1
0
Fork 0
mirror of https://github.com/fumiama/go-onebot-agent.git synced 2026-06-05 02:00:23 +08:00
Code Activity

feat: add view image status management

Browse Source
This commit is contained in:
源文雨
2025-09-24 23:37:27 +08:00
parent a655747fa2
commit 0cdc929590
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
agent.go
View File

@@ -28,6 +28,7 @@ type Agent struct {
chars string chars string
perm *Perm perm *Perm
manualaddreq bool manualaddreq bool
hasimageapi bool
} }
// NewAgent 创建一个 Agent 实例。 // NewAgent 创建一个 Agent 实例。
@@ -59,6 +60,11 @@ func (ag *Agent) AddRequest(grp int64, req *zero.APIRequest) {
ag.log.Add(grp, req, true) ag.log.Add(grp, req, true)
} }
// CanViewImage will be true if SetViewImageAPI is called
func (ag *Agent) CanViewImage() bool {
return ag.hasimageapi
}
// SetViewImageAPI 为 agent 增加识图功能, 需要模型支持视觉 // SetViewImageAPI 为 agent 增加识图功能, 需要模型支持视觉
func (ag *Agent) SetViewImageAPI(api deepinfra.API, p model.Protocol) { func (ag *Agent) SetViewImageAPI(api deepinfra.API, p model.Protocol) {
ag.log.SetPreModelize(func(s *fmt.Stringer) { ag.log.SetPreModelize(func(s *fmt.Stringer) {
@@ -93,6 +99,13 @@ func (ag *Agent) SetViewImageAPI(api deepinfra.API, p model.Protocol) {
} }
} }
}) })
ag.hasimageapi = true
}
// ClearViewImageAPI ...
func (ag *Agent) ClearViewImageAPI() {
ag.log.SetPreModelize(nil)
ag.hasimageapi = false
} }
// GetAction get OneBot CallAction from LLM and add it to context. // GetAction get OneBot CallAction from LLM and add it to context.