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(