Spaces:
Runtime error
Runtime error
m00913563 commited on
Commit ·
9187cc8
1
Parent(s): ff56225
fix pydantic parsing
Browse files- 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 |
+
|