Update model name in orchestrator and demo functions to "gpt-4.1"; remove outdated comments and documentation

mai_dx/__init__.py

@@ -1,20 +1,6 @@
-"""
-MAI Diagnostic Orchestrator (MAI-DxO)
-
-An AI-powered diagnostic system that simulates a virtual panel of physician-agents
-for medical diagnosis. Based on Microsoft Research's "Sequential Diagnosis with 
-Language Models" paper.
-
-Key Components:
-- MaiDxOrchestrator: Main orchestrator class
-- DiagnosisResult: Result container with diagnosis, accuracy, and cost
-- Action: Action model for panel decisions
-- AgentRole: Enumeration of physician agent roles
-"""
-
 from typing import Any, Optional
 
-from .main import (
+from mai_dx.main import (
     MaiDxOrchestrator,
     DiagnosisResult,
     Action,
@@ -24,7 +10,6 @@ from .main import (
 
 __version__ = "1.0.0"
 __author__ = "The Swarm Corporation"
-__email__ = "[email protected]"
 __description__ = "AI-powered diagnostic system with virtual physician panels"
 
 # Main exports

mai_dx/main.py

@@ -18,7 +18,7 @@ Key Features:
 
 Example Usage:
     # Standard MAI-DxO usage
-    orchestrator = MaiDxOrchestrator(model_name="gemini/gemini-2.5-flash")
+    orchestrator = MaiDxOrchestrator(model_name="gpt-4.1")
     result = orchestrator.run(initial_case_info, full_case_details, ground_truth)
 
     # Budget-constrained variant
@@ -131,7 +131,7 @@ class MaiDxOrchestrator:
 
     def __init__(
         self,
-        model_name: str = "gemini/gemini-2.5-flash",
+        model_name: str = "gpt-4.1",
         max_iterations: int = 10,
         initial_budget: int = 10000,
         mode: str = "no_budget",  # "instant", "question_only", "budgeted", "no_budget", "ensemble"
@@ -1387,11 +1387,11 @@ def run_mai_dxo_demo(
                 orchestrator = MaiDxOrchestrator.create_variant(
                     variant,
                     budget=3000,
-                    model_name="gemini/gemini-2.5-flash",
+                    model_name="gpt-4.1",
                 )
             else:
                 orchestrator = MaiDxOrchestrator.create_variant(
-                    variant, model_name="gemini/gemini-2.5-flash"
+                    variant, model_name="gpt-4.1"
                 )
 
             result = orchestrator.run(
@@ -1466,13 +1466,13 @@ def run_mai_dxo_demo(
 #                 orchestrator = MaiDxOrchestrator.create_variant(
 #                     variant_name,
 #                     budget=3000,
-#                     model_name="gemini/gemini-2.5-flash",
+#                     model_name="gpt-4.1",
 #                     max_iterations=5,
 #                 )
 #             else:
 #                 orchestrator = MaiDxOrchestrator.create_variant(
 #                     variant_name,
-#                     model_name="gemini/gemini-2.5-flash",
+#                     model_name="gpt-4.1",
 #                     max_iterations=5,
 #                 )
 
@@ -1504,7 +1504,7 @@ def run_mai_dxo_demo(
 
 #         ensemble_orchestrator = MaiDxOrchestrator.create_variant(
 #             "ensemble",
-#             model_name="gemini/gemini-2.5-flash",
+#             model_name="gpt-4.1",
 #             max_iterations=3,  # Shorter iterations for ensemble
 #         )