langroid 0.52.2__py3-none-any.whl → 0.52.4__py3-none-any.whl

langroid/parsing/file_attachment.py

@@ -209,39 +209,38 @@ class FileAttachment(BaseModel):
         Returns:
             Dictionary with file data
         """
-        if "gemini" in model.lower():
-            return dict(type="image_url", image_url=dict(url=self.to_data_uri()))
-        else:
-            # optimistically try this: some API proxies like litellm
-            # support this, and others may not.
-            # For OpenAI models, handle images differently than other files
-            # For OpenAI models, handle images differently than other files
-            if self.mime_type and self.mime_type.startswith("image/"):
-                image_url_dict = {}
-
-                # If we have a URL and it's a full http/https URL, use it directly
-                if self.url and (
-                    self.url.startswith("http://") or self.url.startswith("https://")
-                ):
-                    image_url_dict["url"] = self.url
-                # Otherwise use base64 data URI
-                else:
-                    image_url_dict["url"] = self.to_data_uri()
-
-                # Add detail parameter if specified
-                if self.detail:
-                    image_url_dict["detail"] = self.detail
-
-                return dict(
-                    type="image_url",
-                    image_url=image_url_dict,
-                )
+        if (
+            self.mime_type
+            and self.mime_type.startswith("image/")
+            or "gemini" in model.lower()
+        ):
+            # for gemini models, we use `image_url` for both pdf-files and images
+
+            image_url_dict = {}
+
+            # If we have a URL and it's a full http/https URL, use it directly
+            if self.url and (
+                self.url.startswith("http://") or self.url.startswith("https://")
+            ):
+                image_url_dict["url"] = self.url
+            # Otherwise use base64 data URI
             else:
-                # For non-image files
-                return dict(
-                    type="file",
-                    file=dict(
-                        filename=self.filename,
-                        file_data=self.to_data_uri(),
-                    ),
-                )
+                image_url_dict["url"] = self.to_data_uri()
+
+            # Add detail parameter if specified
+            if self.detail:
+                image_url_dict["detail"] = self.detail
+
+            return dict(
+                type="image_url",
+                image_url=image_url_dict,
+            )
+        else:
+            # For non-image files
+            return dict(
+                type="file",
+                file=dict(
+                    filename=self.filename,
+                    file_data=self.to_data_uri(),
+                ),
+            )

langroid/parsing/pdf_utils.py

@@ -17,12 +17,12 @@ if fitz is None:
 
 
 def pdf_split_pages(
-    input_pdf: Union[BytesIO, BinaryIO],
+    input_pdf: Union[BytesIO, BinaryIO, str],
 ) -> Tuple[List[Path], TemporaryDirectory[Any]]:
     """Splits a PDF into individual pages in a temporary directory.
 
     Args:
-        input_pdf: Input PDF file in bytes or binary mode
+        input_pdf: Input PDF file in bytes, binary mode, or a file path
         max_workers: Maximum number of concurrent workers for parallel processing
 
     Returns:
@@ -36,7 +36,10 @@ def pdf_split_pages(
         tmp_dir.cleanup()  # Clean up temp files when done
     """
     tmp_dir = tempfile.TemporaryDirectory()
-    doc = fitz.open(stream=input_pdf, filetype="pdf")
+    if isinstance(input_pdf, str):
+        doc = fitz.open(input_pdf)
+    else:
+        doc = fitz.open(stream=input_pdf, filetype="pdf")
     paths = []
 
     for page_num in range(len(doc)):

{langroid-0.52.2.dist-info → langroid-0.52.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.52.2
+Version: 0.52.4
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT

{langroid-0.52.2.dist-info → langroid-0.52.4.dist-info}/RECORD

@@ -83,12 +83,12 @@ langroid/parsing/__init__.py,sha256=2oUWJJAxIavq9Wtw5RGlkXLq3GF3zgXeVLLW4j7yeb8,
 langroid/parsing/agent_chats.py,sha256=sbZRV9ujdM5QXvvuHVjIi2ysYSYlap-uqfMMUKulrW0,1068
 langroid/parsing/code_parser.py,sha256=5ze0MBytrGGkU69pA_bJDjRm6QZz_QYfPcIwkagUa7U,3796
 langroid/parsing/document_parser.py,sha256=cUcp4JKS_LpsjX7OqnGBhHorDHx7FG5pvKGjRBkQoMw,57685
-langroid/parsing/file_attachment.py,sha256=eGW9IEneN1NG-xb8OvM07blvdGNLKI2XlFM7pEIYrt4,7709
+langroid/parsing/file_attachment.py,sha256=ryJVhVFOhINrfkf9Z0vWTTwCnm80z2qzXgp20_miP98,7362
 langroid/parsing/md_parser.py,sha256=JUgsUpCaeAuBndmtDaJR9HMZaje1gmtXtaLXJHst3i8,21340
 langroid/parsing/para_sentence_split.py,sha256=AJBzZojP3zpB-_IMiiHismhqcvkrVBQ3ZINoQyx_bE4,2000
 langroid/parsing/parse_json.py,sha256=aADo38bAHQhC8on4aWZZzVzSDy-dK35vRLZsFI2ewh8,4756
 langroid/parsing/parser.py,sha256=uaAITarcGI2504zcP_dLhp3LjNdh9A6R_yS-o_VcaH8,15599
-langroid/parsing/pdf_utils.py,sha256=rmNJ9UzuBgXTAYwj1TtRJcD8h53x7cizhgyYHKO88I4,1513
+langroid/parsing/pdf_utils.py,sha256=9HnwhbZvpBUhW8WjY9OpGPKaIt3oe_a1AuqhWKqNQ6s,1616
 langroid/parsing/repo_loader.py,sha256=NpysuyzRHvgL3F4BB_wGo5sCUnZ3FOlVCJmZ7CaUdbs,30202
 langroid/parsing/routing.py,sha256=-FcnlqldzL4ZoxuDwXjQPNHgBe9F9-F4R6q7b_z9CvI,1232
 langroid/parsing/search.py,sha256=0NJ5-Rou_BbrHAD7O9b20bKjZJnbadjObvGm4Zq8Kis,9818
@@ -130,7 +130,7 @@ langroid/vector_store/pineconedb.py,sha256=otxXZNaBKb9f_H75HTaU3lMHiaR2NUp5MqwLZ
 langroid/vector_store/postgres.py,sha256=wHPtIi2qM4fhO4pMQr95pz1ZCe7dTb2hxl4VYspGZoA,16104
 langroid/vector_store/qdrantdb.py,sha256=O6dSBoDZ0jzfeVBd7LLvsXu083xs2fxXtPa9gGX3JX4,18443
 langroid/vector_store/weaviatedb.py,sha256=Yn8pg139gOy3zkaPfoTbMXEEBCiLiYa1MU5d_3UA1K4,11847
-langroid-0.52.2.dist-info/METADATA,sha256=Xf03bEX9NqSbKKmbTqzbvNk9uKouRLRniUxlsnDNXfg,63642
-langroid-0.52.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.52.2.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.52.2.dist-info/RECORD,,
+langroid-0.52.4.dist-info/METADATA,sha256=8q70bM9X5i9DkmZ9jg3zAEn5Lk-vUKaJRF46A2EnSvA,63642
+langroid-0.52.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.52.4.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.52.4.dist-info/RECORD,,