Update ui_components.py

ui_components.py

@@ -129,7 +129,7 @@ class UIComponents:
                 choices=[],
                 label="🤖 Model",
                 value=None,
-                info="Select GPT OSS model variant"
+                info="Select model"
             )
             
             # Status display
@@ -147,8 +147,13 @@ class UIComponents:
                 self.presence_penalty = gr.Slider(minimum=-2.0, maximum=2.0, value=0.0, step=0.01, label="Presence penalty")
             self.stop_sequences = gr.Textbox(label="Stop sequences (comma-separated)", placeholder="e.g. \n\n, User:")
 
-            # Reasoning effort (GPT-OSS)
-            self.reasoning_effort = gr.Radio(choices=["low", "medium", "high"], value=AppConfig.DEFAULT_REASONING_EFFORT, label="Reasoning effort (GPT‑OSS)")
+            # Reasoning effort (GPT-OSS only)
+            with gr.Group(visible=True) as self.reasoning_group:
+                self.reasoning_effort = gr.Radio(
+                    choices=["low", "medium", "high"],
+                    value=AppConfig.DEFAULT_REASONING_EFFORT,
+                    label="Reasoning effort (GPT‑OSS)"
+                )
 
             # Response format controls
             with gr.Row():
@@ -187,10 +192,17 @@ class UIComponents:
                     "presence_penalty": float(presence_penalty) if presence_penalty is not None else None,
                     # stop: list[str]
                     "stop": [s.strip() for s in stop_sequences.split(",") if s.strip()] if stop_sequences else None,
-                    # GPT-OSS specific control stored separately; mcp_client merges it
-                    "reasoning_effort": reasoning_effort,
                 }
 
+                # Only include reasoning_effort for GPT-OSS models
+                try:
+                    current_model = self.mcp_client.current_model
+                    if current_model and AppConfig.is_gpt_oss_model(current_model):
+                        params["reasoning_effort"] = reasoning_effort
+                except Exception:
+                    # If any issue, omit reasoning
+                    pass
+
                 # response_format
                 if response_format == "json_object":
                     params["response_format"] = {"type": "json_object"}
@@ -389,7 +401,7 @@ class UIComponents:
         # Provider selection with auto-model loading
         def handle_provider_change(provider_id):
             if not provider_id:
-                return gr.Dropdown(choices=[], value=None), "⚪ Select provider first"
+                return gr.Dropdown(choices=[], value=None), "⚪ Select provider first", gr.Group(visible=False)
             
             available_models = AppConfig.get_available_models_for_provider(provider_id)
             model_choices = [(AppConfig.AVAILABLE_MODELS[model]["name"], model) for model in available_models]
@@ -404,16 +416,18 @@ class UIComponents:
                 status_msg = f"✅ Provider selected, model auto-selected ({context_length:,} token context)"
             else:
                 status_msg = "✅ Provider selected, please select a model"
-            
+            # Reasoning UI visibility based on whether model is GPT-OSS
+            show_reasoning = AppConfig.is_gpt_oss_model(default_model) if default_model else False
             return (
                 gr.Dropdown(choices=model_choices, value=default_model, label="🤖 Model"),
-                status_msg
+                status_msg,
+                gr.Group(visible=show_reasoning)
             )
         
         # Model selection
         def handle_model_change(provider_id, model_id):
             if not provider_id or not model_id:
-                return "⚪ Select both provider and model"
+                return "⚪ Select both provider and model", gr.Group(visible=False)
             
             self.mcp_client.set_model_and_provider(provider_id, model_id)
             
@@ -423,9 +437,13 @@ class UIComponents:
             active_params = model_info.get("active_params", "N/A")
             
             if self.mcp_client.hf_client:
-                return f"✅ Ready! Using {active_params} active params, {context_length:,} token context"
+                status = f"✅ Ready! Using {active_params} active params, {context_length:,} token context"
             else:
-                return "❌ Please add your Hugging Face API token"
+                status = "❌ Please add your Hugging Face API token"
+
+            # Toggle reasoning UI by model family
+            show_reasoning = AppConfig.is_gpt_oss_model(model_id)
+            return status, gr.Group(visible=show_reasoning)
         
         # Chat handlers
         def submit_message(message, history):
@@ -564,7 +582,7 @@ class UIComponents:
         # Connect provider/model dropdowns with auto-selection on load
         demo.load(
             fn=lambda: handle_provider_change("cerebras"),
-            outputs=[self.model_dropdown, self.api_status]
+            outputs=[self.model_dropdown, self.api_status, self.reasoning_group]
         )
 
         # Initialise default mcp server load
@@ -576,13 +594,13 @@ class UIComponents:
         self.provider_dropdown.change(
             handle_provider_change,
             inputs=[self.provider_dropdown],
-            outputs=[self.model_dropdown, self.api_status]
+            outputs=[self.model_dropdown, self.api_status, self.reasoning_group]
         )
         
         self.model_dropdown.change(
             handle_model_change,
             inputs=[self.provider_dropdown, self.model_dropdown],
-            outputs=[self.api_status]
+            outputs=[self.api_status, self.reasoning_group]
         )
         
         # Connect chat