cfde900ada
- Updated MCP server configuration to support advanced formats, including tools and resources. - Refactored the Assistant's Stream method for improved readability and maintainability, adding clear section comments. - Enhanced error handling and logging for better traceability during streaming operations. - Removed deprecated methods and streamlined MCP server handling in tests to ensure robust functionality and validation.
31 lines
848 B
Go
31 lines
848 B
Go
package assistant
|
|
|
|
import (
|
|
"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,
|
|
inputMessages []context.Message,
|
|
agentNode types.Node,
|
|
) ([]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 := inputMessages
|
|
|
|
// Log the chat history
|
|
ast.traceAgentHistory(ctx, agentNode, fullMessages)
|
|
|
|
return fullMessages, nil
|
|
}
|