beb63b77de
- Added support for converting extended types (file, data) to standard LLM types (text, image_url, input_audio) in the Stream method. - Introduced a new agentCallerWrapper to facilitate agent calls from the content package. - Implemented ThreadID management for nested agent calls to improve concurrent stream identification. - Enhanced message handling to include metadata for message_start and message_end events, allowing for better tracking of message states. - Removed deprecated vision capability checks from the Assistant initialization process, streamlining the codebase. - Added new content types (file, data) to the context types for improved message content handling.
48 lines
1.7 KiB
Go
48 lines
1.7 KiB
Go
package content
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/yaoapp/gou/connector/openai"
|
|
agentContext "github.com/yaoapp/yao/agent/context"
|
|
)
|
|
|
|
// ExcelHandler handles Microsoft Excel spreadsheets
|
|
type ExcelHandler struct{}
|
|
|
|
// CanHandle checks if this handler can handle the content type
|
|
func (h *ExcelHandler) CanHandle(contentType string, fileType FileType) bool {
|
|
return fileType == FileTypeExcel ||
|
|
contentType == "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet" ||
|
|
contentType == "application/vnd.ms-excel" ||
|
|
strings.Contains(contentType, "excel") ||
|
|
strings.Contains(contentType, "spreadsheet")
|
|
}
|
|
|
|
// Handle processes Excel spreadsheet content
|
|
func (h *ExcelHandler) Handle(ctx *agentContext.Context, info *Info, capabilities *openai.Capabilities, uses *agentContext.Uses) (*Result, error) {
|
|
// TODO: Implement Excel handling
|
|
// 1. Extract data from .xlsx or .xls file
|
|
// 2. Convert to text format (e.g., CSV-like or structured text)
|
|
// 3. Handle multiple sheets
|
|
// 4. Return Result with formatted text
|
|
return nil, fmt.Errorf("not implemented")
|
|
}
|
|
|
|
// extractExcelText extracts text from Excel file
|
|
func extractExcelText(data []byte, contentType string) (string, error) {
|
|
// TODO: Implement Excel text extraction
|
|
// Handle both .xls (old format) and .xlsx (new format)
|
|
// Consider using libraries like:
|
|
// - github.com/360EntSecGroup-Skylar/excelize for .xlsx
|
|
// Format output as readable text or CSV
|
|
return "", fmt.Errorf("not implemented")
|
|
}
|
|
|
|
// formatExcelAsText formats Excel data as readable text
|
|
func formatExcelAsText(sheets map[string][][]string) string {
|
|
// TODO: Format multiple sheets into readable text
|
|
// Include sheet names, headers, and data
|
|
return ""
|
|
}
|