ankur-bohra commited on
Commit
e0f70b2
·
1 Parent(s): 5e08b8e

Hint towards Pydantic formats

Browse files
Files changed (1) hide show
  1. categories/travel_cab/__init__.py +2 -1
categories/travel_cab/__init__.py CHANGED
@@ -23,7 +23,8 @@ model = ChatOpenAI(
23
  # Build categorizing chain
24
  system_message_prompt = SystemMessagePromptTemplate.from_template(
25
  "You are an information extraction engine that outputs details from OCR processed "
26
- "documents such as date/time/place of departure and arrival.\n"
 
27
  "{format_instructions}"
28
  )
29
  human_message_prompt = HumanMessagePromptTemplate.from_template("{text}")
 
23
  # Build categorizing chain
24
  system_message_prompt = SystemMessagePromptTemplate.from_template(
25
  "You are an information extraction engine that outputs details from OCR processed "
26
+ "documents such as date/time/place of departure and arrival. Your outputs should "
27
+ "conform to Pydantic's acceptable format e.g. datetime.\n"
28
  "{format_instructions}"
29
  )
30
  human_message_prompt = HumanMessagePromptTemplate.from_template("{text}")