sofhiaazzhr commited on
Commit
f34af71
·
1 Parent(s): efc301b

fix planner prompt template parsing

Browse files
Files changed (1) hide show
  1. src/query/planner/service.py +2 -1
src/query/planner/service.py CHANGED
@@ -11,6 +11,7 @@ from __future__ import annotations
11
 
12
  from pathlib import Path
13
 
 
14
  from langchain_core.prompts import ChatPromptTemplate
15
  from langchain_core.runnables import Runnable
16
  from langchain_openai import AzureChatOpenAI
@@ -47,7 +48,7 @@ def _build_default_chain() -> Runnable:
47
  )
48
  prompt = ChatPromptTemplate.from_messages(
49
  [
50
- ("system", _load_prompt_text()),
51
  ("human", "{human_content}"),
52
  ]
53
  )
 
11
 
12
  from pathlib import Path
13
 
14
+ from langchain_core.messages import SystemMessage
15
  from langchain_core.prompts import ChatPromptTemplate
16
  from langchain_core.runnables import Runnable
17
  from langchain_openai import AzureChatOpenAI
 
48
  )
49
  prompt = ChatPromptTemplate.from_messages(
50
  [
51
+ SystemMessage(content=_load_prompt_text()),
52
  ("human", "{human_content}"),
53
  ]
54
  )