added new dspy

app.py

@@ -96,6 +96,7 @@ from src.schemas.query_schema import QueryRequest
 
 from src.utils.logger import Logger
 
+from src.routes.session_routes import apply_model_safeguards
 
 
 # Import deep analysis components directly
@@ -345,22 +346,22 @@ def get_session_lm(session_state):
 
             model_name = model_config.get("model", DEFAULT_MODEL_CONFIG["model"])
 
-            # Get temperature and clamp to valid range for Anthropic (0..1)
-            temp = model_config.get("temperature", DEFAULT_MODEL_CONFIG["temperature"])
-            if provider == "anthropic":
-                temp = min(1.0, max(0.0, float(temp)))
-
-            # Handle special OpenAI models (gpt-5 and o1 series)
-            if ('gpt-5' in model_name or 'o1' in model_name) and provider == 'openai':
-                if 'gpt-5' in model_name:
-                    MODEL_OBJECTS[model_name].__dict__['kwargs']['max_tokens'] = 16_000
-                if 'o1' in model_name:
-                    MODEL_OBJECTS[model_name].__dict__['kwargs']['max_tokens'] = 20_000
-                MODEL_OBJECTS[model_name].__dict__['kwargs']['temperature'] = 1.0
-            else:
-                # All other models
-                MODEL_OBJECTS[model_name].__dict__['kwargs']['max_tokens'] = model_config.get("max_tokens", DEFAULT_MODEL_CONFIG["max_tokens"])
-                MODEL_OBJECTS[model_name].__dict__['kwargs']['temperature'] = temp
+            # Import and apply centralized safeguards (temperature + max_tokens)
+            
+
+            requested_temp = model_config.get("temperature", DEFAULT_MODEL_CONFIG["temperature"])
+            requested_max = model_config.get("max_tokens", DEFAULT_MODEL_CONFIG["max_tokens"])
+
+            safe_params = apply_model_safeguards(
+                model_name=model_name,
+                provider=provider,
+                temperature=requested_temp,
+                max_tokens=requested_max,
+            )
+
+            # Apply the safeguarded parameters
+            MODEL_OBJECTS[model_name].__dict__['kwargs']['max_tokens'] = safe_params["max_tokens"]
+            MODEL_OBJECTS[model_name].__dict__['kwargs']['temperature'] = safe_params["temperature"]
 
 
 

src/routes/session_routes.py

@@ -422,38 +422,27 @@ async def update_model_settings(
         
         # Get session state to update model config
         session_state = app_state.get_session_state(session_id)
-        
-        # Clamp temperature to valid range (0..1 for all providers)
-        clamped_temp = min(1.0, max(0.0, float(settings.temperature)))
+
+        # Apply model-specific safeguards (temperature + max_tokens)
+        safe_params = apply_model_safeguards(
+            model_name=settings.model,
+            provider=settings.provider,
+            temperature=settings.temperature,
+            max_tokens=settings.max_tokens
+        )
+
+        # Create the model config with safe parameters
+        model_config = {
+            "provider": settings.provider,
+            "model": settings.model,
+            "api_key": settings.api_key,
+            "temperature": safe_params["temperature"],
+            "max_tokens": safe_params["max_tokens"]
+        }
+
         
         # Create the model config
-        if 'gpt-5' in str(settings.model):
-            model_config = {
-                "provider": settings.provider,
-                "model": settings.model,
-                "api_key": settings.api_key,
-                "temperature": 1,
-                "max_tokens":16_000
-                # "max_completion_tokens": 2500
-            }
-        elif 'o1' or 'o3' in str(settings.model):
-            model_config = {
-                "provider": settings.provider,
-                "model": settings.model,
-                "api_key": settings.api_key,
-                "temperature": 1.0,  # O-series only supports 1
-                "max_tokens":20_000
-            }
-            
-        
-        else:
-            model_config = {
-                "provider": settings.provider,
-                "model": settings.model,
-                "api_key": settings.api_key,
-                "temperature": clamped_temp,
-                "max_tokens": settings.max_tokens
-            }
+
             
         # Update only the session's model config
         session_state["model_config"] = model_config