38 lines
761 B
Go
38 lines
761 B
Go
package mq
|
|
|
|
import (
|
|
"github.com/google/wire"
|
|
|
|
"github.com/chaitin/panda-wiki/repo/ipdb"
|
|
"github.com/chaitin/panda-wiki/repo/mq"
|
|
"github.com/chaitin/panda-wiki/repo/pg"
|
|
"github.com/chaitin/panda-wiki/store/rag"
|
|
"github.com/chaitin/panda-wiki/store/s3"
|
|
"github.com/chaitin/panda-wiki/usecase"
|
|
)
|
|
|
|
type MQHandlers struct {
|
|
RAGMQHandler *RAGMQHandler
|
|
RagDocUpdateHandler *RagDocUpdateHandler
|
|
StatCronHandler *CronHandler
|
|
}
|
|
|
|
var ProviderSet = wire.NewSet(
|
|
pg.ProviderSet,
|
|
rag.ProviderSet,
|
|
mq.ProviderSet,
|
|
ipdb.ProviderSet,
|
|
s3.ProviderSet,
|
|
|
|
usecase.NewLLMUsecase,
|
|
usecase.NewStatUseCase,
|
|
usecase.NewNodeUsecase,
|
|
usecase.NewModelUsecase,
|
|
|
|
NewRAGMQHandler,
|
|
NewRagDocUpdateHandler,
|
|
NewCronHandler,
|
|
|
|
wire.Struct(new(MQHandlers), "*"),
|
|
)
|