From 150e84ed185157b16f99ad77dfef59a16a2900a4 Mon Sep 17 00:00:00 2001 From: Fangyin Cheng Date: Fri, 25 Jul 2025 17:13:35 +0800 Subject: [PATCH] fix(awel): Fix HO context key error (#2862) --- packages/dbgpt-app/src/dbgpt_app/operators/datasource.py | 1 + packages/dbgpt-app/src/dbgpt_app/operators/rag.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/packages/dbgpt-app/src/dbgpt_app/operators/datasource.py b/packages/dbgpt-app/src/dbgpt_app/operators/datasource.py index 702c981e7..c47b709b0 100644 --- a/packages/dbgpt-app/src/dbgpt_app/operators/datasource.py +++ b/packages/dbgpt-app/src/dbgpt_app/operators/datasource.py @@ -148,6 +148,7 @@ _PARAMETER_CONTEXT_KEY = Parameter.build_from( type=str, optional=True, default="context", + alias=["context"], description=_("The key of the context, it will be used in building the prompt"), ) _INPUTS_QUESTION = IOField.build_from( diff --git a/packages/dbgpt-app/src/dbgpt_app/operators/rag.py b/packages/dbgpt-app/src/dbgpt_app/operators/rag.py index 18144bc09..33140da87 100644 --- a/packages/dbgpt-app/src/dbgpt_app/operators/rag.py +++ b/packages/dbgpt-app/src/dbgpt_app/operators/rag.py @@ -33,10 +33,11 @@ def _load_space_name() -> List[OptionValue]: _PARAMETER_CONTEXT_KEY = Parameter.build_from( _("Context Key"), - "context", + "context_key", type=str, optional=True, default="context", + alias=["context"], description=_("The key of the context, it will be used in building the prompt"), ) _PARAMETER_TOP_K = Parameter.build_from(