4ba62600dc
- Removed the synchronous preparation of the knowledge base (KB) collection from the InitializeConversation method, now initializing it asynchronously after user login. - Introduced a new method, GetDocumentsContent, to retrieve content for multiple documents by their IDs, supporting text-based files and improving document handling. - Updated the API interface to include the new GetDocumentsContent method, enhancing the document management capabilities. - Enhanced locale handling in the login context to support user preferences during KB collection creation.
70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package kb
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/yaoapp/gou/process"
|
|
"github.com/yaoapp/kun/exception"
|
|
"github.com/yaoapp/yao/kb"
|
|
)
|
|
|
|
// ProcessGetDocumentsContent retrieves content for documents by IDs via Yao process
|
|
// Process: kb.documents.getcontents
|
|
//
|
|
// Args[0]: document_ids (string | []string) - Document ID or list of document IDs
|
|
//
|
|
// Returns: []map containing document_id, name, content, content_type, etc. for each document
|
|
//
|
|
// Example:
|
|
//
|
|
// // Single document
|
|
// Process("kb.documents.getcontents", "doc_id_123")
|
|
//
|
|
// // Multiple documents
|
|
// Process("kb.documents.getcontents", ["doc_id_1", "doc_id_2", "doc_id_3"])
|
|
func ProcessGetDocumentsContent(process *process.Process) interface{} {
|
|
process.ValidateArgNums(1)
|
|
|
|
if kb.API == nil {
|
|
exception.New("Knowledge base not initialized", 500).Throw()
|
|
}
|
|
|
|
// Support both single string and array of strings
|
|
var docIDs []string
|
|
arg := process.Args[0]
|
|
|
|
switch v := arg.(type) {
|
|
case string:
|
|
if v == "" {
|
|
exception.New("Document ID is required", 400).Throw()
|
|
}
|
|
docIDs = []string{v}
|
|
case []string:
|
|
docIDs = v
|
|
case []interface{}:
|
|
for _, item := range v {
|
|
if s, ok := item.(string); ok && s != "" {
|
|
docIDs = append(docIDs, s)
|
|
}
|
|
}
|
|
default:
|
|
exception.New("Document IDs must be a string or array of strings", 400).Throw()
|
|
}
|
|
|
|
if len(docIDs) == 0 {
|
|
exception.New("Document IDs are required", 400).Throw()
|
|
}
|
|
|
|
ctx := process.Context
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
|
|
// Call KB API to get documents content
|
|
results, err := kb.API.GetDocumentsContent(ctx, docIDs)
|
|
if err != nil {
|
|
exception.New("Failed to get documents content: "+err.Error(), 500).Throw()
|
|
}
|
|
|
|
return results
|
|
}
|