v3.1: Register custom 'anthropic' provider in DataDesigner

app.py

@@ -10,7 +10,7 @@ from models import (
     HealthResponse, ZaiModel
 )
 
-# z.ai als Anthropic-Endpunkt für LiteLLM konfigurieren
+# z.ai API Konfiguration
 ZAI_API_KEY = os.environ.get("ZAI_API_KEY", "")
 ZAI_BASE_URL = "https://api.z.ai/api/anthropic"
 
@@ -19,20 +19,35 @@ os.environ["ANTHROPIC_API_KEY"] = ZAI_API_KEY
 os.environ["ANTHROPIC_BASE_URL"] = ZAI_BASE_URL
 
 data_designer = None
+zai_provider = None
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    global data_designer
+    global data_designer, zai_provider
     from data_designer.interface import DataDesigner
-    data_designer = DataDesigner(artifact_path=tempfile.gettempdir())
+    from data_designer.config.models import ModelProvider
+    
+    # Custom "anthropic" Provider registrieren
+    # provider_type="openai" ist Pflicht, aber LiteLLM erkennt "anthropic/" Prefix
+    zai_provider = ModelProvider(
+        name="anthropic",
+        endpoint=ZAI_BASE_URL,
+        provider_type="openai",
+        api_key="ZAI_API_KEY",
+    )
+    
+    data_designer = DataDesigner(
+        artifact_path=tempfile.gettempdir(),
+        model_providers=[zai_provider]
+    )
     yield
 
 
 app = FastAPI(
     title="NeMo DataDesigner API",
     description="Synthetic data generation with DataDesigner + z.ai",
-    version="3.0.0",
+    version="3.1.0",
     lifespan=lifespan
 )
 
@@ -73,7 +88,7 @@ def build_config(request):
                 )
             )
 
-    # LiteLLM erkennt "anthropic/" Prefix und nutzt ANTHROPIC_BASE_URL
+    # "anthropic/" Prefix sagt LiteLLM, dass es das Anthropic-Protokoll nutzen soll
     model_config = ModelConfig(
         alias="zai-model",
         model=f"anthropic/{model_id}",