666c1e8d74
- Updated the Vision function and associated handlers to accept a forceUses parameter, allowing explicit control over the use of external tools regardless of model capabilities. - Modified the Assistant's BuildContent method to set the AssistantID in the context for better file tracking. - Enhanced the applyCreateResponseOptions method to merge and prioritize Uses and ForceUses configurations from the createResponse. - Added comprehensive tests to validate the adjustment of Uses and ForceUses configurations through hooks, ensuring correct application in content processing.
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package content
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/gou/connector/openai"
|
|
agentContext "github.com/yaoapp/yao/agent/context"
|
|
)
|
|
|
|
// Registry holds all registered content handlers
|
|
type Registry struct {
|
|
handlers []Handler
|
|
}
|
|
|
|
// NewRegistry creates a new handler registry with default handlers
|
|
func NewRegistry() *Registry {
|
|
return &Registry{
|
|
handlers: []Handler{
|
|
&ImageHandler{},
|
|
&AudioHandler{},
|
|
&PDFHandler{},
|
|
&WordHandler{},
|
|
&ExcelHandler{},
|
|
&TextHandler{},
|
|
},
|
|
}
|
|
}
|
|
|
|
// GetHandler finds the appropriate handler for the given content
|
|
func (r *Registry) GetHandler(contentType string, fileType FileType) Handler {
|
|
for _, handler := range r.handlers {
|
|
if handler.CanHandle(contentType, fileType) {
|
|
return handler
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Handle processes content using the appropriate handler
|
|
func (r *Registry) Handle(ctx *agentContext.Context, info *Info, capabilities *openai.Capabilities, uses *agentContext.Uses, forceUses bool) (*Result, error) {
|
|
handler := r.GetHandler(info.ContentType, info.FileType)
|
|
if handler == nil {
|
|
return nil, fmt.Errorf("no handler found for content type: %s, file type: %s", info.ContentType, info.FileType)
|
|
}
|
|
|
|
return handler.Handle(ctx, info, capabilities, uses, forceUses)
|
|
}
|