Changes in embedding.py
Another try, this time trusting pydantic_tensor to fix it
This commit is contained in:
1 parent
b5137abe11
commit
71206a03fd
1 file changed
+1
-24
@@ -22,27 +22,4 @@ class Embedding:
|
||||
# Fallback or error if not implemented for other types (e.g., torch.Tensor)
|
||||
raise NotImplementedError(
|
||||
"Comparison must be implemented for all supported data types."
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def __get_pydantic_core_schema__(cls, source_type: Any, handler):
|
||||
def serialize(value: "Embedding") -> dict:
|
||||
serialized_data = handler(source_type=value.data.__class__, value=value.data)
|
||||
return {
|
||||
'data': serialized_data
|
||||
}
|
||||
|
||||
return core_schema.json_or_python_schema(
|
||||
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'
|
||||
),
|
||||
)
|
||||
)
|
||||
Reference in new issue
Block a user