feat: introduce Customizable Base URLs for API Providers

basilisk/config/__init__.py

@@ -1,6 +1,11 @@
 """Configuration module for Basilisk."""
 
-from .account_config import Account, AccountManager, AccountOrganization
+from .account_config import (
+	CUSTOM_BASE_URL_PATTERN,
+	Account,
+	AccountManager,
+	AccountOrganization,
+)
 from .account_config import get_account_config as accounts
 from .config_enums import (
 	AccountSource,
@@ -27,6 +32,8 @@
 	"conf",
 	"ConversationProfile",
 	"conversation_profiles",
+	"CUSTOM_BASE_URL_PATTERN",
+	"get_account_source_labels",
 	"KeyStorageMethodEnum",
 	"LogLevelEnum",
 	"ReleaseChannelEnum",

basilisk/config/account_config.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import logging
+import re
 from functools import cache, cached_property
 from os import getenv
 from typing import Annotated, Any, Iterable, Optional, Union
@@ -38,6 +39,10 @@
 
 log = logging.getLogger(__name__)
 
+CUSTOM_BASE_URL_PATTERN = re.compile(
+	r"^https?://[\w.-]+(?::\d{1,5})?(?:/[\w-]+)*/?$"
+)
+
 
 class AccountOrganization(BaseModel):
 	"""Manage organization key for an account."""
@@ -131,6 +136,11 @@ class Account(BaseModel):
 	organizations: Optional[list[AccountOrganization]] = Field(default=None)
 	active_organization_id: Optional[UUID4] = Field(default=None)
 	source: AccountSource = Field(default=AccountSource.CONFIG, exclude=True)
+	custom_base_url: Optional[str] = Field(
+		default=None,
+		pattern=CUSTOM_BASE_URL_PATTERN,
+		description="Custom base URL for the API provider. Must be a valid HTTP/HTTPS URL.",
+	)
 
 	def __init__(self, **data: Any):
 		"""Initialize an account instance. If an error occurs, log the error and raise an exception."""
@@ -186,6 +196,8 @@ def validate_api_key(
 			if not value:
 				raise ValueError("API key not found in keyring")
 			return SecretStr(value)
+		elif not data["provider"].require_api_key and value is None:
+			return None
 		else:
 			raise ValueError("Invalid API key storage method")