Compare commits
1 Commits
main
...
zj/ft/sugg
| Author | SHA1 | Date | |
|---|---|---|---|
| 41a936321c |
@ -77,7 +77,7 @@ func NewRepository(idgen idgen.IDGenerator, db *gorm.DB, redis cache.Cmdable, to
|
|||||||
cpStore einoCompose.CheckPointStore, chatModel cm.BaseChatModel, workflowConfig workflow.WorkflowConfig) (workflow.Repository, error) {
|
cpStore einoCompose.CheckPointStore, chatModel cm.BaseChatModel, workflowConfig workflow.WorkflowConfig) (workflow.Repository, error) {
|
||||||
sg, err := NewSuggester(chatModel)
|
sg, err := NewSuggester(chatModel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
logs.Warnf("[NewRepository] Failed to create suggester: %v", err)
|
||||||
}
|
}
|
||||||
return &RepositoryImpl{
|
return &RepositoryImpl{
|
||||||
IDGenerator: idgen,
|
IDGenerator: idgen,
|
||||||
@ -103,6 +103,13 @@ func NewRepository(idgen idgen.IDGenerator, db *gorm.DB, redis cache.Cmdable, to
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (r *RepositoryImpl) Suggest(ctx context.Context, input *vo.SuggestInfo) ([]string, error) {
|
||||||
|
if r.Suggester == nil {
|
||||||
|
return []string{}, nil
|
||||||
|
}
|
||||||
|
return r.Suggester.Suggest(ctx, input)
|
||||||
|
}
|
||||||
|
|
||||||
func (r *RepositoryImpl) CreateMeta(ctx context.Context, meta *vo.Meta) (int64, error) {
|
func (r *RepositoryImpl) CreateMeta(ctx context.Context, meta *vo.Meta) (int64, error) {
|
||||||
id, err := r.GenID(ctx)
|
id, err := r.GenID(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
Reference in New Issue
Block a user