cjber commited on
Commit
1c9f9ee
·
1 Parent(s): 4762df6

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 LLM
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 = LLM.with_structured_output(Grade, strict=True)
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()