Spaces:
Build error
Build error
feat: specify gpt4o for structured output
Browse files
planning_ai/retrievers/theme_retriever.py
CHANGED
@@ -8,7 +8,7 @@ from langchain_core.prompts import PromptTemplate
|
|
8 |
from langchain_openai import OpenAIEmbeddings
|
9 |
from pydantic import BaseModel, Field
|
10 |
|
11 |
-
from planning_ai.llms.llm import
|
12 |
|
13 |
# See: https://consultations.greatercambridgeplanning.org/greater-cambridge-local-plan-preferred-options/supporting-documents
|
14 |
|
@@ -68,7 +68,7 @@ grade_template = PromptTemplate(
|
|
68 |
)
|
69 |
|
70 |
|
71 |
-
SLLM =
|
72 |
grade_chain = grade_template | SLLM
|
73 |
|
74 |
vectorstore = create_db()
|
|
|
8 |
from langchain_openai import OpenAIEmbeddings
|
9 |
from pydantic import BaseModel, Field
|
10 |
|
11 |
+
from planning_ai.llms.llm import GPT4o
|
12 |
|
13 |
# See: https://consultations.greatercambridgeplanning.org/greater-cambridge-local-plan-preferred-options/supporting-documents
|
14 |
|
|
|
68 |
)
|
69 |
|
70 |
|
71 |
+
SLLM = GPT4o.with_structured_output(Grade, strict=True)
|
72 |
grade_chain = grade_template | SLLM
|
73 |
|
74 |
vectorstore = create_db()
|