From 5ec380decf1850d9a113efdad07c030fdaeb6cef Mon Sep 17 00:00:00 2001 From: Pao Sheng Date: Mon, 3 Mar 2025 18:07:54 +0800 Subject: [PATCH] chore: remove alias from model additional params --- wren-ai-service/src/providers/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wren-ai-service/src/providers/__init__.py b/wren-ai-service/src/providers/__init__.py index 8260f33c04..b9c3ff4b41 100644 --- a/wren-ai-service/src/providers/__init__.py +++ b/wren-ai-service/src/providers/__init__.py @@ -70,7 +70,7 @@ def llm_processor(entry: dict) -> dict: for model in entry.get("models", []): model_name = f"{entry.get('provider')}.{model.get('alias', model.get('model'))}" model_additional_params = { - k: v for k, v in model.items() if k not in ["model", "kwargs"] + k: v for k, v in model.items() if k not in ["model", "kwargs", "alias"] } returned[model_name] = { "provider": entry["provider"], @@ -124,7 +124,7 @@ def embedder_processor(entry: dict) -> dict: for model in entry["models"]: identifier = f"{entry['provider']}.{model.get('alias', model.get('model'))}" model_additional_params = { - k: v for k, v in model.items() if k not in ["model", "kwargs"] + k: v for k, v in model.items() if k not in ["model", "kwargs", "alias"] } returned[identifier] = { "provider": entry["provider"],