fix arg-type error reported by mypy

This commit is contained in:
italojohnny 2024-06-20 17:22:36 -03:00
commit 6903eca138

View file

@ -1,5 +1,5 @@
from datetime import datetime, timezone from datetime import datetime, timezone
from typing import Annotated, Any, AsyncIterator, Iterator, Optional from typing import Annotated, Any, AsyncIterator, Iterator, Optional, List
from fastapi.encoders import jsonable_encoder from fastapi.encoders import jsonable_encoder
from langchain_core.load import load from langchain_core.load import load
@ -42,7 +42,7 @@ class Message(Data):
return value return value
def model_post_init(self, __context: Any) -> None: def model_post_init(self, __context: Any) -> None:
new_files = [] new_files: List[Any] = []
for file in self.files or []: for file in self.files or []:
if is_image_file(file): if is_image_file(file):
new_files.append(Image(path=file)) new_files.append(Image(path=file))