m00913563 commited on
Commit
9187cc8
·
1 Parent(s): ff56225

fix pydantic parsing

Browse files
Files changed (1) hide show
  1. extractor_llm.py +3 -1
extractor_llm.py CHANGED
@@ -114,5 +114,7 @@ def predict(input):
114
  ],
115
  response_format=CVExtracted,
116
  )
117
- completion = client.chat.completions.create(**model_parameters)
 
118
  return completion.choices[0].message.parsed
 
 
114
  ],
115
  response_format=CVExtracted,
116
  )
117
+ completion = client.chat.completions.create(**model_parameters.model_dump()) # For Pydantic v2
118
+ #completion = client.chat.completions.create(**model_parameters)
119
  return completion.choices[0].message.parsed
120
+