소스 검색

Fix: Parameter Extractor Uses Correct Prompt for Prompt Mode in Chat Models (#24636)

Co-authored-by: -LAN- <laipz8200@outlook.com>
tags/2.0.0-beta.2^2
Ding 1 개월 전
부모
커밋
ce2281d31b
No account linked to committer's email address
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2
    1
      api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py

+ 2
- 1
api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py 파일 보기

@@ -52,6 +52,7 @@ from .exc import (
)
from .prompts import (
CHAT_EXAMPLE,
CHAT_GENERATE_JSON_PROMPT,
CHAT_GENERATE_JSON_USER_MESSAGE_TEMPLATE,
COMPLETION_GENERATE_JSON_PROMPT,
FUNCTION_CALLING_EXTRACTOR_EXAMPLE,
@@ -752,7 +753,7 @@ class ParameterExtractorNode(BaseNode):
if model_mode == ModelMode.CHAT:
system_prompt_messages = ChatModelMessage(
role=PromptMessageRole.SYSTEM,
text=FUNCTION_CALLING_EXTRACTOR_SYSTEM_PROMPT.format(histories=memory_str, instruction=instruction),
text=CHAT_GENERATE_JSON_PROMPT.format(histories=memory_str).replace("{{instructions}}", instruction),
)
user_prompt_message = ChatModelMessage(role=PromptMessageRole.USER, text=input_text)
return [system_prompt_messages, user_prompt_message]

Loading…
취소
저장