feat(kol): generation service + worker branch
This commit is contained in:
@@ -65,6 +65,14 @@ func main() {
|
||||
app.Config.LLM.MaxOutputTokens,
|
||||
).WithCache(app.Cache)
|
||||
|
||||
kolWorker := generateworker.NewKolGenerationWorker(
|
||||
app.DB,
|
||||
app.LLM,
|
||||
app.Logger,
|
||||
generationCfg,
|
||||
app.Config.LLM.MaxOutputTokens,
|
||||
).WithCache(app.Cache)
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
@@ -73,6 +81,7 @@ func main() {
|
||||
app.RabbitMQ,
|
||||
templateSvc,
|
||||
promptRuleSvc,
|
||||
kolWorker,
|
||||
app.Logger,
|
||||
app.Config.Generation,
|
||||
).Start(ctx)
|
||||
|
||||
Reference in New Issue
Block a user