refactor document pydantic schemas a bit

This commit is contained in:
phernandez
2024-12-21 20:01:09 -06:00
parent ddf2d2fd0e
commit 76322b79c0
3 changed files with 13 additions and 13 deletions
+3 -3
View File
@@ -57,7 +57,7 @@ async def get_document(
"""Get a document by path."""
try:
document, content = await service.read_document(path)
response = DocumentResponse.from_orm(document)
response = DocumentResponse.model_validate(document)
response.content = content
return response
except DocumentNotFoundError:
@@ -77,7 +77,7 @@ async def update_document(
document = await service.update_document(
path=path,
content=doc.content,
metadata=doc.metadata,
metadata=doc.doc_metadata,
)
return DocumentResponse.model_validate(document)
except DocumentNotFoundError:
@@ -104,7 +104,7 @@ async def patch_document(
document = await service.update_document(
path=path,
content=patch.content,
metadata=patch.metadata,
metadata=patch.doc_metadata,
)
return DocumentResponse.from_orm(document)
except DocumentNotFoundError: