Commit fecf37f

@@ -249,6 +249,8 @@ async def get_openai_file_id(client: Client, message: Message, openai_params: di
         if base_url.startswith("https://gateway.helicone.ai"):
             helicone_target_url = default_headers.get("helicone-target-url") or ""
             return base_url.replace("https://gateway.helicone.ai", helicone_target_url.rstrip("/"))
+        if base_url == "https://api.portkey.ai/v1":
+            return default_headers.get("x-portkey-custom-host") or ""
         return base_url
 
     if mtype not in ["photo", "video", "document"]:
@@ -265,7 +267,7 @@ async def get_openai_file_id(client: Client, message: Message, openai_params: di
     model_id = openai_params["model_id"]
     key_hash = hashlib.sha256(api_key.encode()).hexdigest()
     tid = get_thread_id(message)
-    r2_key = f"TTL/{cache_day}d/OpenAI/{model_id}/{key_hash}/{message.chat.id}/{message.id}{'/' + str(tid) if tid else ''}-file_id"
+    r2_key = f"TTL/{cache_day}d/OpenAI/{message.chat.id}/{message.id}{'/' + str(tid) if tid else ''}/{model_id}/{key_hash}-file_id"
     r2 = await head_cf_r2(r2_key)
     if file_id := glom(r2, "Metadata.file_id", default=""):
         return file_id