Fix: resolve pyright issues with PDF (#1465)

Author: ivanleomk

instructor/multimodal.py

@@ -525,8 +525,8 @@ def to_anthropic(self) -> dict[str, Any]:
             }
         else:
             if not self.data:
-                self.data = requests.get(str(self.source)).content
-                self.data = base64.b64encode(self.data).decode("utf-8")
+                self.data = requests.get(str(self.source)).content  # type: ignore
+                self.data = base64.b64encode(self.data).decode("utf-8")  # type: ignore
 
             return {
                 "type": "document",
@@ -586,7 +586,7 @@ def from_new_genai_file(
         file = client.files.upload(file=file_path)
         while file.state != FileState.ACTIVE:
             time.sleep(retry_delay)
-            file = client.files.get(name=file.name)
+            file = client.files.get(name=file.name)  # type: ignore
             if max_retries > 0:
                 max_retries -= 1
             else:

pyproject.toml

@@ -16,7 +16,7 @@ dependencies = [
     "requests<3.0.0,>=2.32.3",
 ]
 name = "instructor"
-version = "1.7.8"
+version = "1.7.9"
 description = "structured outputs for llm"
 readme = "README.md"
 

tests/llm/test_genai/test_multimodal.py

@@ -17,7 +17,7 @@ class ImageDescription(BaseModel):
 
 pdf_path = os.path.join(curr_file, "../../assets/invoice.pdf")
 
-
+pdf_url = "https://raw.githubusercontent.com/instructor-ai/instructor/main/tests/assets/invoice.pdf"
 pdf_base64 = base64.b64encode(open(pdf_path, "rb").read()).decode("utf-8")
 pdf_base64_string = f"data:application/pdf;base64,{pdf_base64}"
 
@@ -222,7 +222,7 @@ async def test_autodetect_images_async(client, model, mode, autodetect_images):
         messages=[
             {
                 "role": "system",
-                "content": "Return true if you are provided with an image that you can describe visually in your prompt. This does not include image paths or urls that might point to URLs. ",
+                "content": "Do you see any blueberries in the context that you've been provided?. ",
             },
             {
                 "role": "user",
@@ -243,7 +243,7 @@ class Invoice(BaseModel):
 
 @pytest.mark.parametrize("model", models)
 @pytest.mark.parametrize("mode", modes)
-@pytest.mark.parametrize("pdf_source", [pdf_path, pdf_base64_string])
+@pytest.mark.parametrize("pdf_source", [pdf_path, pdf_base64_string, pdf_url])
 def test_local_pdf(client, model, mode, pdf_source):
     client = instructor.from_genai(client, mode=mode)
     response = client.chat.completions.create(
@@ -315,7 +315,7 @@ def test_upload_file_genai_pdf_integration(client, model, mode):
 
 @pytest.mark.parametrize("model", models)
 @pytest.mark.parametrize("mode", modes)
-@pytest.mark.parametrize("pdf_source", [pdf_path, pdf_base64_string])
+@pytest.mark.parametrize("pdf_source", [pdf_path, pdf_base64_string, pdf_url])
 def test_local_pdf_with_genai_file(client, model, mode, pdf_source):
     client = instructor.from_genai(client, mode=mode)
     response = client.chat.completions.create(