cjber commited on
Commit
3bfd521
·
1 Parent(s): 8ecd7b9

fix: use gpt4o not o3-mini

Browse files
Files changed (1) hide show
  1. planning_ai/chains/reduce_chain.py +3 -3
planning_ai/chains/reduce_chain.py CHANGED
@@ -2,7 +2,7 @@ from langchain_core.output_parsers import StrOutputParser
2
  from langchain_core.prompts import ChatPromptTemplate
3
 
4
  from planning_ai.common.utils import Paths
5
- from planning_ai.llms.llm import O3Mini
6
 
7
  with open(Paths.PROMPTS / "reduce.txt", "r") as f:
8
  reduce_template = f.read()
@@ -11,11 +11,11 @@ with open(Paths.PROMPTS / "reduce_final.txt", "r") as f:
11
  reduce_template_final = f.read()
12
 
13
  reduce_prompt = ChatPromptTemplate([("system", reduce_template)])
14
- reduce_chain = reduce_prompt | O3Mini | StrOutputParser()
15
 
16
 
17
  reduce_prompt_final = ChatPromptTemplate([("system", reduce_template_final)])
18
- reduce_chain_final = reduce_prompt_final | O3Mini | StrOutputParser()
19
 
20
 
21
  if __name__ == "__main__":
 
2
  from langchain_core.prompts import ChatPromptTemplate
3
 
4
  from planning_ai.common.utils import Paths
5
+ from planning_ai.llms.llm import GPT4o
6
 
7
  with open(Paths.PROMPTS / "reduce.txt", "r") as f:
8
  reduce_template = f.read()
 
11
  reduce_template_final = f.read()
12
 
13
  reduce_prompt = ChatPromptTemplate([("system", reduce_template)])
14
+ reduce_chain = reduce_prompt | GPT4o | StrOutputParser()
15
 
16
 
17
  reduce_prompt_final = ChatPromptTemplate([("system", reduce_template_final)])
18
+ reduce_chain_final = reduce_prompt_final | GPT4o | StrOutputParser()
19
 
20
 
21
  if __name__ == "__main__":