Skip to content

fix BUG-api_pipelines\text2vecsql_pipeline_gen#338

Merged
SunnyHaze merged 2 commits intoOpenDCAI:mainfrom
yaodongwen:main
Oct 25, 2025
Merged

fix BUG-api_pipelines\text2vecsql_pipeline_gen#338
SunnyHaze merged 2 commits intoOpenDCAI:mainfrom
yaodongwen:main

Conversation

@yaodongwen
Copy link
Contributor

修复.BUG-api_pipelines\text2vecsql_pipeline_gen,初始化第三步算子抛出类型错误

Copy link
Collaborator

@SunnyHaze SunnyHaze left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个应该是把你这个算子注册到对应算子的prompt_restrict函数中,这样是对的。
参考15行和24行。

@prompt_restrict(Text2SQLQuestionGeneratorPrompt)
@OPERATOR_REGISTRY.register()
class Text2SQLQuestionGenerator(OperatorABC):
def __init__(self,
llm_serving: LLMServingABC,
embedding_serving: LLMServingABC,
database_manager: DatabaseManager,
question_candidates_num: int = 5,
prompt_template = Text2SQLQuestionGeneratorPrompt

@SunnyHaze
Copy link
Collaborator

以及官方的Prompt应该还是由Prompt基类实现,只有用户自定义的才需要集成DIYPrompt

Copy link
Collaborator

@SunnyHaze SunnyHaze left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, sincerely thanks!

@SunnyHaze SunnyHaze merged commit 8608fed into OpenDCAI:main Oct 25, 2025
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

Comments