refactorings

This commit is contained in:
mudler
2025-03-02 17:45:06 +01:00
parent 3a7e56cdf1
commit 6c83f3d089

View File

@@ -299,16 +299,20 @@ func (a *Agent) knowledgeBaseLookup() {
} }
xlog.Info("[Knowledge Base Lookup] Last user message", "agent", a.Character.Name, "message", userMessage) xlog.Info("[Knowledge Base Lookup] Last user message", "agent", a.Character.Name, "message", userMessage)
if userMessage != "" { if userMessage == "" {
xlog.Info("[Knowledge Base Lookup] No user message found in conversation", "agent", a.Character.Name)
return
}
results, err := a.options.ragdb.Search(userMessage, a.options.kbResults) results, err := a.options.ragdb.Search(userMessage, a.options.kbResults)
if err != nil { if err != nil {
xlog.Info("Error finding similar strings inside KB:", "error", err) xlog.Info("Error finding similar strings inside KB:", "error", err)
// job.Result.Finish(fmt.Errorf("error finding similar strings inside KB: %w", err))
// return
} }
if len(results) != 0 { if len(results) == 0 {
xlog.Info("[Knowledge Base Lookup] No similar strings found in KB", "agent", a.Character.Name)
return
}
formatResults := "" formatResults := ""
for _, r := range results { for _, r := range results {
@@ -327,8 +331,6 @@ func (a *Agent) knowledgeBaseLookup() {
Role: "system", Role: "system",
Content: fmt.Sprintf("Given the user input you have the following in memory:\n%s", formatResults), Content: fmt.Sprintf("Given the user input you have the following in memory:\n%s", formatResults),
}}, a.currentConversation...) }}, a.currentConversation...)
}
}
} }
func (a *Agent) consumeJob(job *Job, role string) { func (a *Agent) consumeJob(job *Job, role string) {