fix: py tests

Author: enitrat

python/tests/unit/test_generation_program.py

@@ -374,10 +374,10 @@ async def test_forward_retry_logic(self, call_method, generation_program):
         # Mock the generation_program to raise AdapterParseError
         side_effect = [
             AdapterParseError(
-                "Parse error 1", CairoCodeGeneration(), "", "test response", None
+                "Parse error 1", CairoCodeGeneration, "", "test response", None
             ),
             AdapterParseError(
-                "Parse error 2", CairoCodeGeneration(), "", "test response", None
+                "Parse error 2", CairoCodeGeneration, "", "test response", None
             ),
             dspy.Prediction(answer="Success"),
         ]
@@ -399,16 +399,16 @@ async def test_forward_max_retries_exceeded(self, call_method, generation_progra
         # Mock the generation_program to always raise AdapterParseError
         side_effect = [
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "", "test response", None
+                "Parse error", CairoCodeGeneration, "", "test response", None
             ),
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "", "test response", None
+                "Parse error", CairoCodeGeneration, "", "test response", None
             ),
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "", "test response", None
+                "Parse error", CairoCodeGeneration, "", "test response", None
             ),
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "", "test response", None
+                "Parse error", CairoCodeGeneration, "", "test response", None
             ),
         ]
         getattr(generation_program.generation_program, call_method).side_effect = side_effect
@@ -446,13 +446,13 @@ async def test_should_extract_code_before_raising(self, generation_program, call
         # Mock the generation_program to raise AdapterParseError
         side_effect = [
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "", "test response", None
+                "Parse error", CairoCodeGeneration, "", "test response", None
             ),
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "", "test response", None
+                "Parse error", CairoCodeGeneration, "", "test response", None
             ),
             AdapterParseError(
-                "Parse error", CairoCodeGeneration(), "```cairo\nfn main() {}\n```", "test response", None
+                "Parse error", CairoCodeGeneration, "```cairo\nfn main() {}\n```", "test response", None
             ),
         ]
         generation_program.generation_program.aforward.side_effect = side_effect

python/tests/unit/test_rag_pipeline.py

@@ -8,6 +8,7 @@
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
+import dspy
 
 from cairo_coder.core.rag_pipeline import (
     RagPipeline,
@@ -123,7 +124,7 @@ def mock_mcp_generation_program(self):
 
 Storage variables use #[storage] attribute.
 """
-        program.forward.return_value = mock_res
+        program.forward.return_value = dspy.Prediction(answer=mock_res)
         return program
 
     @pytest.fixture