Skip to content
4 changes: 4 additions & 0 deletions src/benchmark_core/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -289,6 +289,10 @@ class UsagePolicyProfile(BaseModel):
default=None,
description="Budget limit in currency units",
)
budget_currency: str = Field(
default="USD",
description="Budget currency",
)
ttl_seconds: int | None = Field(
default=None,
description="Key TTL in seconds (time-to-live before expiration)",
Expand Down
19 changes: 14 additions & 5 deletions src/benchmark_core/services/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,19 @@

# Comprehensive SQL-based services (COE-305 implementation)
from benchmark_core.services.benchmark_metadata_service import BenchmarkMetadataService

# Unified credential service (COE-370)
from benchmark_core.services.credential_service import CredentialService
from benchmark_core.services.experiment_service import ExperimentService
from benchmark_core.services.harness_profile_service import HarnessProfileService
from benchmark_core.services.provider_service import ProviderService
Comment thread
kumanday marked this conversation as resolved.

# Sessionless proxy key services
from benchmark_core.services.proxy_key_service import (
LiteLLMAPIError,
ProxyKeyService,
ProxyKeyServiceError,
)
from benchmark_core.services.rendering import (
EnvRenderingService,
EnvSnippet,
Expand All @@ -25,11 +35,6 @@
from benchmark_core.services.task_card_service import TaskCardService
from benchmark_core.services.variant_service import VariantService

# ABC service exports - Note: old services_abc SessionService removed, use services.session_service
from benchmark_core.services_abc import (
CredentialService,
)

__all__ = [
# Core SQL-based services
"SessionService",
Expand All @@ -49,6 +54,10 @@
"RenderingError",
"ProfileValidationError",
"render_env_for_session",
# Sessionless proxy key services
"ProxyKeyService",
"ProxyKeyServiceError",
"LiteLLMAPIError",
# ABC services
"CredentialService",
]
Loading
Loading