Mazyar
51a1a6aa82
continuous-integration/drone/push Build is passing
- Updated comments and logging messages in the worker and related files to replace "daily-run" with "auto run" for clarity and consistency. - Adjusted the `WorkerConfig` struct to reflect the new terminology in configuration settings. - Renamed functions and test cases to align with the updated terminology, enhancing code readability and maintainability. This change improves the clarity of the AI pipeline's functionality within the tender management system.
53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package workers
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"tm/pkg/ai_summarizer"
|
|
"tm/pkg/logger"
|
|
)
|
|
|
|
// AIPipelineDailyWorker triggers the scheduled Opplens AI pipeline auto run.
|
|
type AIPipelineDailyWorker struct {
|
|
Logger logger.Logger
|
|
AIClient *ai_summarizer.Client
|
|
}
|
|
|
|
// NewAIPipelineDailyWorker creates a worker that calls POST /pipeline/auto on the AI service.
|
|
func NewAIPipelineDailyWorker(log logger.Logger, aiClient *ai_summarizer.Client) *AIPipelineDailyWorker {
|
|
return &AIPipelineDailyWorker{
|
|
Logger: log,
|
|
AIClient: aiClient,
|
|
}
|
|
}
|
|
|
|
// Run starts the pipeline auto job. A 409 response means a run is already in progress and is returned as an error
|
|
// so the daily job tracker does not mark today as completed before the run actually starts.
|
|
func (w *AIPipelineDailyWorker) Run() error {
|
|
if w.AIClient == nil {
|
|
w.Logger.Warn("AI pipeline auto worker skipped: AI client is not configured", map[string]interface{}{})
|
|
return nil
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Minute)
|
|
defer cancel()
|
|
|
|
w.Logger.Info("AI pipeline auto worker started", map[string]interface{}{})
|
|
|
|
resp, err := w.AIClient.PipelineAuto(ctx)
|
|
if err != nil {
|
|
w.Logger.Error("AI pipeline auto worker failed", map[string]interface{}{
|
|
"error": err.Error(),
|
|
})
|
|
return err
|
|
}
|
|
|
|
w.Logger.Info("AI pipeline auto worker triggered successfully", map[string]interface{}{
|
|
"status": resp.Status,
|
|
"report": resp.Report,
|
|
})
|
|
|
|
return nil
|
|
}
|