From a33fa62012b5fc4f05e0f1f7934b5346321da939 Mon Sep 17 00:00:00 2001 From: ZhuangYumin Date: Thu, 26 Jun 2025 18:53:32 +0800 Subject: [PATCH] able to submit model id --- openrouter_provider.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openrouter_provider.py b/openrouter_provider.py index a54368a..48b57df 100644 --- a/openrouter_provider.py +++ b/openrouter_provider.py @@ -191,7 +191,7 @@ class Pipe: except Exception as e: print(f"Error reporting API call: {e}") - def pipe(self, body: dict, __user__: dict) -> Union[str, Generator, Iterator]: + def pipe(self, body: dict, __user__: dict, __metadata__: dict) -> Union[str, Generator, Iterator]: """Process the request and handle reasoning tokens if supported""" # Clone the body for OpenRouter payload = body.copy() @@ -201,7 +201,7 @@ class Pipe: # Extract user email and model ID for reporting user_email = __user__.get("email", "") if __user__ else "" - model_id = "test_model" + model_id = __metadata__.get("model").get("id", "") if __metadata__ else "" # Make sure the model ID is properly extracted from the pipe format if "model" in payload and payload["model"] and "." in payload["model"]: