e6a0f37a66
- Updated UserOrGuestID method to return guest status alongside user ID. - Introduced UploadOption struct to encapsulate additional upload parameters, including public visibility and collection ID. - Enhanced handleUpload method to validate collection ID for knowledge storage and improved error handling. - Updated GetHeader function to include file size in the header processing. - Implemented content fingerprinting for file deduplication and added support for upload status tracking. - Enhanced README documentation to reflect new features and usage examples for file uploads and deduplication.
89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
package neo
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/yaoapp/gou/session"
|
|
"github.com/yaoapp/yao/neo/assistant"
|
|
chatctx "github.com/yaoapp/yao/neo/context"
|
|
)
|
|
|
|
// Answer reply the message
|
|
func (neo *DSL) Answer(ctx chatctx.Context, question string, c *gin.Context) error {
|
|
var err error
|
|
var ast assistant.API = Neo.Assistant
|
|
if ctx.AssistantID != "" {
|
|
ast, err = neo.Select(ctx.AssistantID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
_, err = ast.Execute(c, ctx, question, nil)
|
|
return err
|
|
}
|
|
|
|
// Select select an assistant
|
|
func (neo *DSL) Select(id string) (assistant.API, error) {
|
|
if id == "" {
|
|
return Neo.Assistant, nil
|
|
}
|
|
return assistant.Get(id)
|
|
}
|
|
|
|
// UserID get the user id from the session
|
|
func (neo *DSL) UserID(sid string) (interface{}, error) {
|
|
fieldID := neo.AuthSetting.SessionFields.ID
|
|
return session.Global().ID(sid).Get(fieldID)
|
|
}
|
|
|
|
// GuestID get the guest id from the session
|
|
func (neo *DSL) GuestID(sid string) (interface{}, error) {
|
|
fieldGuest := neo.AuthSetting.SessionFields.Guest
|
|
return session.Global().ID(sid).Get(fieldGuest)
|
|
}
|
|
|
|
// UserRoles get the user roles from the session
|
|
func (neo *DSL) UserRoles(sid string) (interface{}, error) {
|
|
fieldRoles := neo.AuthSetting.SessionFields.Roles
|
|
return session.Global().ID(sid).Get(fieldRoles)
|
|
}
|
|
|
|
// UserOrGuestID get the user id or guest id from the session
|
|
func (neo *DSL) UserOrGuestID(sid string) (interface{}, bool, error) {
|
|
userID, err := neo.UserID(sid)
|
|
if err != nil {
|
|
guestID, err := neo.GuestID(sid)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
return guestID, true, nil
|
|
}
|
|
return userID, false, nil
|
|
}
|
|
|
|
// Download downloads a file
|
|
func (neo *DSL) Download(ctx chatctx.Context, c *gin.Context) (*assistant.FileResponse, error) {
|
|
// Get file_id from query string
|
|
fileID := c.Query("file_id")
|
|
if fileID == "" {
|
|
return nil, fmt.Errorf("file_id is required")
|
|
}
|
|
|
|
// Get assistant_id from context or query
|
|
// res, err := neo.HookCreate(ctx, []map[string]interface{}{}, c)
|
|
// if err != nil {
|
|
// return nil, err
|
|
// }
|
|
|
|
// Select Assistant
|
|
ast, err := neo.Select(neo.Use.Default)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Download file using the assistant
|
|
// return ast.Download(ctx.Context, fileID)
|
|
fmt.Println(ast)
|
|
return nil, nil
|
|
}
|