Complete Migration

agent/models.py

@@ -23,6 +23,16 @@ class CategoryEvaluation(BaseModel):
             )
         return score
 
+class FinalResult(BaseModel):
+    """Pydantic model for final tweet optimization result."""
+    initial_text: str = Field(description="The initial tweet text")
+    final_tweet: str = Field(description="The final optimized tweet")
+    best_score: float = Field(description="The best score for the final tweet")
+    iterations_run: int = Field(description="The number of iterations run")
+    early_stopped: bool = Field(description="Whether the optimization early stopped")
+    scores_history: List[List[int]] = Field(description="The history of scores")
+    improvement_count: int = Field(description="The number of improvements found")
+
 
 class EvaluationResult(BaseModel):
     """Pydantic model for tweet evaluation results."""