23 lines
746 B
Python
23 lines
746 B
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from ..services.product_detect import detect_product_name
|
|
from .base import BaseSkill, SkillResult, WorkflowContext
|
|
|
|
|
|
logger = logging.getLogger("review_agent.file_summary.skills.product_detect")
|
|
|
|
|
|
class ProductDetectSkill(BaseSkill):
|
|
name = "product_detect"
|
|
|
|
def run(self, context: WorkflowContext) -> SkillResult:
|
|
logger.info("Product detect started", extra={"batch_id": context.batch.pk})
|
|
product_name = detect_product_name(context.batch)
|
|
logger.info(
|
|
"Product detect finished",
|
|
extra={"batch_id": context.batch.pk, "product_name": product_name},
|
|
)
|
|
return SkillResult(success=True, data={"product_name": product_name})
|