diff --git a/llama_stack/providers/remote/inference/portkey/config.py b/llama_stack/providers/remote/inference/portkey/config.py index 144fbf6a4..d2add22c9 100644 --- a/llama_stack/providers/remote/inference/portkey/config.py +++ b/llama_stack/providers/remote/inference/portkey/config.py @@ -23,6 +23,14 @@ class PortkeyImplConfig(BaseModel): default=os.environ.get("PORTKEY_API_KEY"), description="Portkey API Key", ) + virtual_key: Optional[str] = Field( + default=os.environ.get("PORTKEY_VIRTUAL_KEY"), + description="Portkey Virtual Key", + ) + config: Optional[str] = Field( + default=os.environ.get("PORTKEY_CONFIG_ID"), + description="Portkey Config ID", + ) @classmethod def sample_run_config(cls, **kwargs) -> Dict[str, Any]: