Stage 4

backend/igny8_core/ai/engine.py

@@ -6,6 +6,7 @@ from typing import Dict, Any, Optional
 from igny8_core.ai.base import BaseAIFunction
 from igny8_core.ai.tracker import StepTracker, ProgressTracker, CostTracker
 from igny8_core.ai.ai_core import AICore
+from igny8_core.ai.settings import get_model_config
 
 logger = logging.getLogger(__name__)
 
@@ -62,7 +63,11 @@ class AIEngine:
             
             # Phase 3: AI_CALL - Provider API Call (25-70%)
             ai_core = AICore(account=self.account)
-            model = fn.get_model(self.account)
+            function_name = fn.get_name()
+            
+            # Get model config from settings
+            model_config = get_model_config(function_name)
+            model = model_config.get('model')
             
             # Track AI call start
             self.step_tracker.add_response_step("AI_CALL", "success", f"Calling {model or 'default'} model...")
@@ -73,9 +78,10 @@ class AIEngine:
                 raw_response = ai_core.run_ai_request(
                     prompt=prompt,
                     model=model,
-                    max_tokens=4000,
-                    temperature=0.7,
-                    function_name=fn.get_name()
+                    max_tokens=model_config.get('max_tokens'),
+                    temperature=model_config.get('temperature'),
+                    response_format=model_config.get('response_format'),
+                    function_name=function_name
                 )
             except Exception as e:
                 error_msg = f"AI call failed: {str(e)}"