704c0331b1
- Added permission validation in the Assistant's Stream method to ensure user authorization before processing input messages. - Introduced conversation initialization within the Stream method to prepare the context for chat interactions. - Removed the history.go file, which previously contained a placeholder method for handling chat history, streamlining the Assistant's codebase. - Updated the Knowledge Base API integration in collection management, ensuring all collection operations utilize the new API structure for improved consistency and error handling.
79 lines
2 KiB
Go
79 lines
2 KiB
Go
package assistant
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/yao/agent/context"
|
|
"github.com/yaoapp/yao/trace/types"
|
|
)
|
|
|
|
// WithHistory merges the input messages with chat history and traces it
|
|
// This method can be overridden or extended to implement actual history loading
|
|
func (ast *Assistant) WithHistory(ctx *context.Context, input []context.Message, agentNode types.Node, options ...*context.Options) ([]context.Message, error) {
|
|
|
|
// TODO: Implement actual history loading logic here
|
|
// For now, just simulate a check and return the input messages as is
|
|
|
|
// Simulate error check (this is where actual history loading would happen)
|
|
// if some_condition {
|
|
// ast.traceAgentFail(agentNode, err)
|
|
// return nil, err
|
|
// }
|
|
|
|
fullMessages := input
|
|
|
|
// Log the chat history
|
|
ast.traceAgentHistory(ctx, agentNode, fullMessages)
|
|
|
|
return fullMessages, nil
|
|
}
|
|
|
|
// initializeConversation initialize the conversation
|
|
func (ast *Assistant) initializeConversation(ctx *context.Context, input []context.Message, options ...*context.Options) error {
|
|
|
|
var opts *context.Options
|
|
if len(options) > 0 && options[0] != nil {
|
|
opts = options[0]
|
|
} else {
|
|
opts = &context.Options{}
|
|
}
|
|
|
|
// SKIP: History (for internal calls like title/prompt etc.)
|
|
if opts.Skip != nil && opts.Skip.History {
|
|
return nil
|
|
}
|
|
|
|
chatid := ctx.ChatID
|
|
teamid := ctx.Authorized.TeamID
|
|
userid := ctx.Authorized.UserID
|
|
fmt.Printf(">>> initializeChat: chatid=%s, teamid=%s, userid=%s\n", chatid, teamid, userid)
|
|
|
|
// Prepare kb collection (optional)
|
|
err := ast.prepareKBCollection(ctx, input, opts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Save chat
|
|
err = ast.saveChat(ctx, input, opts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Prepare kb collection (optional)
|
|
func (ast *Assistant) prepareKBCollection(ctx *context.Context, input []context.Message, opts *context.Options) error {
|
|
_ = ctx
|
|
_ = opts
|
|
_ = input
|
|
return nil
|
|
}
|
|
|
|
func (ast *Assistant) saveChat(ctx *context.Context, input []context.Message, opts *context.Options) error {
|
|
_ = ctx
|
|
_ = input
|
|
_ = opts
|
|
return nil
|
|
}
|