ATOM Documentation

def test_brennan_has_minimax_2_7_key_not_minimax_key(self):
    # Verify: MINIMAX_2_7_API_KEY exists, MINIMAX_API_KEY does not
    assert minimax_2_7_key is not None  # PASS
    assert minimax_key is None  # PASS (root cause)

def test_byok_handler_initializes_minimax_2_7_but_not_minimax(self):
    # Initialize BYOKHandler for brennan
    clients = byok_handler.clients

    # Check which providers are in clients
    assert has_minimax_2_7  # PASS (key exists)
    assert not has_minimax  # PASS (no key, not initialized)

def test_model_ranking_filters_out_minimax_models(self):
    # Get available providers (what's in self.clients)
    available_providers = list(byok_handler.clients.keys())

    # Test model matching logic from rank_models_by_pricing
    for model_id in ["minimax/gpt-4", "minimax_2_7/gpt-4"]:
        prefix = model_id.split("/")[0]
        if prefix in available_providers:
            # Model can be used
        else:
            # Model is skipped

def test_rank_models_raises_error_when_no_models_pass_filters(self):
    # Call rank_models_by_pricing
    with pytest.raises(ValueError) as exc_info:
        byok_handler.rank_models_by_pricing(...)

    assert "No models available from dynamic pricing" in str(exc_info.value)

def test_fix_add_minimax_api_key_resolves_issue(self):
    # Add MINIMAX_API_KEY to database (same value as MINIMAX_2_7_API_KEY)
    new_setting = TenantSetting(
        setting_key="MINIMAX_API_KEY",
        setting_value=minimax_2_7_key.setting_value
    )
    db.add(new_setting)
    db.commit()

    # Re-initialize BYOKHandler
    byok_handler = BYOKHandler(...)
    clients = byok_handler.clients

    # Check if minimax is now in clients
    assert "minimax" in clients  # PASS (fix works)

# Line 588-612 in core/llm/byok_handler.py
for provider_id, provider_config in self.byok_manager.providers.items():
    if provider_id not in _LLM_PROVIDERS:
        continue

    # Get tenant-specific API key
    api_key = self.byok_manager.get_tenant_api_key(
        self.tenant_id, provider_id, db=db
    )

    if not api_key:
        logger.debug(f"No key found for {provider_id} in {self.workspace_id}")
        continue  # SKIP provider initialization

    # Initialize client for this provider
    self._clients_dict[provider_id] = client

# Line 848 in core/byok_endpoints.py
setting_key = f"{provider_id.upper()}_API_KEY"

# Query database
setting = db.query(TenantSetting).filter(
    TenantSetting.tenant_id == tenant_id,
    TenantSetting.setting_key == setting_key
).first()

if setting and setting.setting_value:
    return setting.setting_value
else:
    logger.warning(f"Setting not found or NULL for {provider_id}")
    return None  # Provider will be skipped

# Line 1178-1200 in core/llm/byok_handler.py
available_providers = list(self.clients.keys())  # Only initialized providers

for model_id, pricing in fetcher.pricing_cache.items():
    # Extract provider from model_id (e.g., "minimax/gpt-4" -> "minimax")
    if "/" in model_id.lower():
        prefix = model_id.lower().split("/")[0]
        if prefix in available_providers:
            active_provider = prefix
        else:
            continue  # SKIP this model

    # ... filter by context, quality, capabilities, etc ...

    candidates.append({
        "provider": active_provider,
        "model": model_id,
        ...
    })

# If no candidates pass all filters
if not ranked_options:
    raise ValueError("No models available from dynamic pricing")

# Via tenant_settings table
INSERT INTO tenant_settings (tenant_id, setting_key, setting_value)
VALUES (
    '31c06fc4-db22-4740-83ea-48ac14f25810',
    'MINIMAX_API_KEY',
    '<same_value_as_MINIMAX_2_7_API_KEY>'
);