fix: 1.调整api直连输出 2.增加知识库loading输出
This commit is contained in:
parent
b184dce3ea
commit
14a5fe5974
|
|
@ -9,7 +9,6 @@ import (
|
||||||
"ai_scheduler/internal/data/model"
|
"ai_scheduler/internal/data/model"
|
||||||
"ai_scheduler/internal/entitys"
|
"ai_scheduler/internal/entitys"
|
||||||
"ai_scheduler/internal/gateway"
|
"ai_scheduler/internal/gateway"
|
||||||
"ai_scheduler/internal/pkg"
|
|
||||||
"ai_scheduler/internal/pkg/l_request"
|
"ai_scheduler/internal/pkg/l_request"
|
||||||
"ai_scheduler/internal/pkg/mapstructure"
|
"ai_scheduler/internal/pkg/mapstructure"
|
||||||
"ai_scheduler/internal/tools"
|
"ai_scheduler/internal/tools"
|
||||||
|
|
@ -269,7 +268,7 @@ func (r *Handle) handleApiTask(ctx context.Context, requireData *entitys.Require
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
entitys.ResJson(requireData.Ch, "", pkg.JsonStringIgonErr(res.Text))
|
entitys.ResJson(requireData.Ch, "", res.Text)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -59,6 +59,7 @@ func (k *KnowledgeBaseTool) Definition() entitys.ToolDefinition {
|
||||||
|
|
||||||
// Execute 执行知识库查询
|
// Execute 执行知识库查询
|
||||||
func (k *KnowledgeBaseTool) Execute(ctx context.Context, requireData *entitys.RequireData) error {
|
func (k *KnowledgeBaseTool) Execute(ctx context.Context, requireData *entitys.RequireData) error {
|
||||||
|
entitys.ResLoading(requireData.Ch, k.Name(), "正在为您搜索相关信息")
|
||||||
|
|
||||||
return k.chat(requireData)
|
return k.chat(requireData)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue