Fixed up failing tests

instructor-ai · Oct 5, 2024 · 5ded593 · 5ded593
1 parent 6629e8e
commit 5ded593
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/instructor/multimodal.py b/instructor/multimodal.py
@@ -29,6 +29,9 @@ def from_path(cls, path: str | Path) -> Image:
  if suffix not in ["jpeg", "jpg", "png"]:
  raise ValueError(f"Unsupported image format: {suffix}")
 
+ if path.stat().st_size == 0:
+ raise ValueError("Image file is empty")
+
  media_type = "image/jpeg" if suffix in ["jpeg", "jpg"] else "image/png"
  data = base64.b64encode(path.read_bytes()).decode("utf-8")
  return cls(source=str(path), media_type=media_type, data=data)

diff --git a/instructor/process_response.py b/instructor/process_response.py
@@ -664,7 +664,8 @@ def handle_response_model(
  else:
  raise ValueError(f"Invalid patch mode: {mode}")
 
- new_kwargs["messages"] = convert_messages(new_kwargs["messages"], mode)
+ if "messages" in new_kwargs:
+ new_kwargs["messages"] = convert_messages(new_kwargs["messages"], mode)
 
  logger.debug(
  f"Instructor Request: {mode.value=}, {response_model=}, {new_kwargs=}",