fix type per models

2025-10-05 16:30:48 +02:00
parent a31acb3a0b
commit ac356c3753
2 changed files with 3 additions and 3 deletions
--- a/src/app/agents/pipeline.py
+++ b/src/app/agents/pipeline.py
@@ -30,7 +30,7 @@ class Pipeline:
        model = self.available_models[index]
        self.predictor = model.get_agent(
            PREDICTOR_INSTRUCTIONS,
-            output=PredictorOutput, # type: ignore
+            output_schema=PredictorOutput,
        )

    def choose_style(self, index: int):