selfhostyourtech/cubbi
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/cubbi.git synced 2025-12-20 12:19:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: cubbi configure not working when configuring other provider (#32)

Browse Source
This commit is contained in:
Mathieu Virbel
2025-08-08 10:24:45 -06:00
committed by GitHub
parent 3a7b9213b0
commit 310149dc34
1 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
cubbi/configure.py
View File

@@ -75,6 +75,10 @@ class ProviderConfigurator:
# Add existing providers
for name, config in providers.items():
provider_type = config.get("type", "unknown")
base_url = config.get("base_url")
if base_url:
choices.append(f"{name} ({provider_type}) - {base_url}")
else:
choices.append(f"{name} ({provider_type})")
# Add separator and options
@@ -93,12 +97,13 @@ class ProviderConfigurator:
self._add_new_provider()
else:
# Extract provider name from the choice
# Format: "provider_name (provider_type)"
# Format: "provider_name (provider_type)" or "provider_name (provider_type) - base_url"
provider_name = choice.split(" (")[0]
self._edit_provider(provider_name)
def _add_new_provider(self) -> None:
"""Add a new provider configuration."""
OTHER = "Other (openai compatible)"
# Ask for provider type
provider_type = questionary.select(
"Select provider type:",
@@ -107,7 +112,7 @@ class ProviderConfigurator:
"OpenAI",
"Google",
"OpenRouter",
"OpenAI-compatible (custom)",
OTHER,
],
).ask()
@@ -120,13 +125,13 @@ class ProviderConfigurator:
"OpenAI": "openai",
"Google": "google",
"OpenRouter": "openrouter",
"Other (openai compatible)": "openai",
OTHER: "openai",
}
internal_type = type_mapping[provider_type]
# Ask for provider name
if provider_type == "OpenAI-compatible (custom)":
if provider_type == OTHER:
provider_name = questionary.text(
"Enter a name for this provider (e.g., 'litellm', 'local-llm'):",
validate=lambda name: len(name.strip()) > 0
@@ -194,7 +199,7 @@ class ProviderConfigurator:
return
base_url = None
if internal_type == "openai" and provider_type == "OpenAI-compatible (custom)":
if internal_type == "openai" and provider_type == OTHER:
base_url = questionary.text(
"Base URL for API calls:",
validate=lambda url: url.startswith("http")
@@ -268,6 +273,10 @@ class ProviderConfigurator:
provider_type = provider_config.get("type", "unknown")
has_key = bool(provider_config.get("api_key"))
if has_key:
base_url = provider_config.get("base_url")
if base_url:
choices.append(f"{provider_name} ({provider_type}) - {base_url}")
else:
choices.append(f"{provider_name} ({provider_type})")
if not choices:
@@ -316,7 +325,11 @@ class ProviderConfigurator:
console.print("\n[bold]Providers[/bold]")
providers = self.user_config.list_providers()
if providers:
for name in providers.keys():
for name, config in providers.items():
base_url = config.get("base_url")
if base_url:
console.print(f" - {name} ({base_url})")
else:
console.print(f" - {name}")
else:
console.print(" (no providers configured)")