diff --git a/src/llm_models/utils_model.py b/src/llm_models/utils_model.py index 215b0f739..1f90a730a 100644 --- a/src/llm_models/utils_model.py +++ b/src/llm_models/utils_model.py @@ -216,7 +216,7 @@ class LLMRequest: prompt: str = None, image_base64: str = None, image_format: str = None, - file_bytes: str = None, + file_bytes: bytes = None, file_format: str = None, payload: dict = None, retry_policy: dict = None, @@ -296,7 +296,7 @@ class LLMRequest: prompt: str = None, image_base64: str = None, image_format: str = None, - file_bytes: str = None, + file_bytes: bytes = None, file_format: str = None, payload: dict = None, retry_policy: dict = None,