Spaces:
Build error
Build error
fix: use gpt4o not o3-mini
Browse files
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
|
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 |
|
15 |
|
16 |
|
17 |
reduce_prompt_final = ChatPromptTemplate([("system", reduce_template_final)])
|
18 |
-
reduce_chain_final = reduce_prompt_final |
|
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__":
|