Changes in embedding.py

Manually defining schema
This commit is contained in:
Christoph Eder committed 2025-12-12 16:56:44 +01:00
1 parent 2cf4f2912a
commit b5137abe11
1 file changed
+9 -1
+9 -1
View File
@@ -33,7 +33,15 @@ class Embedding:
}
return core_schema.json_or_python_schema(
python_schema=handler(cls),
python_schema=core_schema.dataclass_schema(
schema=handler.generate_schema(cls),
fields={
'data': core_schema.model_field(
schema=handler.generate_schema(cls.data),
required=True
),
}
),
json_schema=core_schema.plain_serializer_function_ser_schema(
serialize, when_used='json'
),