From 9fd50e0272c011d375e290d6f8dc200fdaaba789 Mon Sep 17 00:00:00 2001
From: siddharthsambharia-portkey <siddhath.s@portkey.ai>
Date: Fri, 20 Dec 2024 18:14:27 +0530
Subject: [PATCH] portkey integration v4

---
 llama_stack/providers/remote/inference/portkey/config.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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]: