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

langroid/language_models/model_info.py

@@ -31,6 +31,8 @@ class OpenAIChatModel(ModelName):
     O1 = "o1"
     O1_MINI = "o1-mini"
     O3_MINI = "o3-mini"
+    O3 = "o3"
+    O4_MINI = "o4-mini"
     GPT4_1 = "gpt-4.1"
     GPT4_1_MINI = "gpt-4.1-mini"
     GPT4_1_NANO = "gpt-4.1-nano"
@@ -226,6 +228,20 @@ MODEL_INFO: Dict[str, ModelInfo] = {
         has_tools=False,
         description="O1 Reasoning LM",
     ),
+    OpenAIChatModel.O3.value: ModelInfo(
+        name=OpenAIChatModel.O3.value,
+        provider=ModelProvider.OPENAI,
+        context_length=200_000,
+        max_output_tokens=100_000,
+        input_cost_per_million=10.0,
+        output_cost_per_million=40.0,
+        allows_streaming=True,
+        allows_system_message=False,
+        unsupported_params=["temperature"],
+        rename_params={"max_tokens": "max_completion_tokens"},
+        has_tools=False,
+        description="O1 Reasoning LM",
+    ),
     OpenAIChatModel.O1_MINI.value: ModelInfo(
         name=OpenAIChatModel.O1_MINI.value,
         provider=ModelProvider.OPENAI,
@@ -254,6 +270,20 @@ MODEL_INFO: Dict[str, ModelInfo] = {
         has_tools=False,
         description="O3 Mini Reasoning LM",
     ),
+    OpenAIChatModel.O4_MINI.value: ModelInfo(
+        name=OpenAIChatModel.O4_MINI.value,
+        provider=ModelProvider.OPENAI,
+        context_length=200_000,
+        max_output_tokens=100_000,
+        input_cost_per_million=1.10,
+        output_cost_per_million=4.40,
+        allows_streaming=False,
+        allows_system_message=False,
+        unsupported_params=["temperature", "stream"],
+        rename_params={"max_tokens": "max_completion_tokens"},
+        has_tools=False,
+        description="O3 Mini Reasoning LM",
+    ),
     # Anthropic Models
     AnthropicModel.CLAUDE_3_5_SONNET.value: ModelInfo(
         name=AnthropicModel.CLAUDE_3_5_SONNET.value,

langroid/parsing/file_attachment.py

@@ -3,6 +3,7 @@ import mimetypes
 import uuid
 from pathlib import Path
 from typing import Any, BinaryIO, Dict, Optional, Union
+from urllib.parse import urlparse
 
 from langroid.pydantic_v1 import BaseModel
 
@@ -13,6 +14,8 @@ class FileAttachment(BaseModel):
     content: bytes
     filename: Optional[str] = None
     mime_type: str = "application/octet-stream"
+    url: str | None = None
+    detail: str | None = None
 
     def __init__(self, **data: Any) -> None:
         """Initialize with sensible defaults for filename if not provided."""
@@ -23,7 +26,11 @@ class FileAttachment(BaseModel):
         super().__init__(**data)
 
     @classmethod
-    def from_path(cls, file_path: Union[str, Path]) -> "FileAttachment":
+    def _from_path(
+        cls,
+        file_path: Union[str, Path],
+        detail: Optional[str] = None,
+    ) -> "FileAttachment":
         """Create a FileAttachment from a file path.
 
         Args:
@@ -40,7 +47,74 @@ class FileAttachment(BaseModel):
         if mime_type is None:
             mime_type = "application/octet-stream"
 
-        return cls(content=content, filename=path.name, mime_type=mime_type)
+        return cls(
+            content=content,
+            filename=path.name,
+            mime_type=mime_type,
+            detail=detail,
+        )
+
+    @classmethod
+    def _from_url(
+        cls,
+        url: str,
+        content: Optional[bytes] = None,
+        filename: Optional[str] = None,
+        mime_type: Optional[str] = None,
+        detail: Optional[str] = None,
+    ) -> "FileAttachment":
+        """Create a FileAttachment from a URL.
+
+        Args:
+            url: URL to the file
+            content: Optional raw bytes content (if already fetched)
+            filename: Optional name to use for the file
+            mime_type: MIME type of the content, guessed from filename or url
+
+        Returns:
+            FileAttachment instance
+        """
+        if filename is None and url:
+            # Extract filename from URL if possible
+
+            parsed_url = urlparse(url)
+            path = parsed_url.path
+            filename = path.split("/")[-1] if path else None
+
+        if mime_type is None and filename:
+            mime_type, _ = mimetypes.guess_type(filename)
+
+        return cls(
+            content=content or b"",  # Empty bytes if no content provided
+            filename=filename,
+            mime_type=mime_type or "application/octet-stream",
+            url=url,
+            detail=detail,
+        )
+
+    @classmethod
+    def from_path(
+        cls,
+        path: Union[str, Path],
+        detail: str | None = None,
+    ) -> "FileAttachment":
+        """Create a FileAttachment from either a local file path or a URL.
+
+        Args:
+            path_or_url: Path to the file or URL to fetch
+
+        Returns:
+            FileAttachment instance
+        """
+        # Convert to string if Path object
+        path_str = str(path)
+
+        # Check if it's a URL
+        if path_str.startswith(("http://", "https://", "ftp://")):
+            return cls._from_url(url=path_str, detail=detail)
+        else:
+            # Assume it's a local file path
+            return cls._from_path(path_str, detail=detail)
 
     @classmethod
     def from_bytes(
@@ -137,21 +211,37 @@ class FileAttachment(BaseModel):
         """
         if "gemini" in model.lower():
             return dict(type="image_url", image_url=dict(url=self.to_data_uri()))
-        elif "claude" in model.lower():
+        else:
             # optimistically try this: some API proxies like litellm
             # support this, and others may not.
-            return dict(
-                type="file",
-                file=dict(
-                    file_data=self.to_data_uri(),
-                ),
-            )
-        else:
-            # fallback: assume file upload is similar to OpenAI API
-            return dict(
-                type="file",
-                file=dict(
-                    filename=self.filename,
-                    file_data=self.to_data_uri(),
-                ),
-            )
+            # 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,
+                )
+            else:
+                # For non-image files
+                return dict(
+                    type="file",
+                    file=dict(
+                        filename=self.filename,
+                        file_data=self.to_data_uri(),
+                    ),
+                )

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

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

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

@@ -72,7 +72,7 @@ langroid/language_models/azure_openai.py,sha256=SW0Fp_y6HpERr9l6TtF6CYsKgKwjUf_h
 langroid/language_models/base.py,sha256=pfN3t-BktKmN_4K8pwmpjC9OdcHxsytM5s5TmsJ-nPg,28560
 langroid/language_models/config.py,sha256=9Q8wk5a7RQr8LGMT_0WkpjY8S4ywK06SalVRjXlfCiI,378
 langroid/language_models/mock_lm.py,sha256=5BgHKDVRWFbUwDT_PFgTZXz9-k8wJSA2e3PZmyDgQ1k,4022
-langroid/language_models/model_info.py,sha256=0NE1zWNUHJwcM5jhwNxUqGjbpek-Nq7ljGdWpM8R3RQ,13380
+langroid/language_models/model_info.py,sha256=vOaTi-XFKnz-BvHUvgjnt0XfOtl21Apev3Zy7Rhckbw,14458
 langroid/language_models/openai_gpt.py,sha256=KFbG6q143CYt6SP8rU1UEFX2mffn8yXfVOdEyuA5IaY,85854
 langroid/language_models/utils.py,sha256=hC5p61P_Qlrowkm5wMap1A1b5ZUCwK_XhPIzAQk1T1s,5483
 langroid/language_models/prompt_formatter/__init__.py,sha256=2-5cdE24XoFDhifOLl8yiscohil1ogbP1ECkYdBlBsk,372
@@ -83,7 +83,7 @@ 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=iIMTmAkfu-TgV9CKnDEB-BiDDN6WOepH51sIz6-PnFw,4826
+langroid/parsing/file_attachment.py,sha256=eGW9IEneN1NG-xb8OvM07blvdGNLKI2XlFM7pEIYrt4,7709
 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
@@ -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.0.dist-info/METADATA,sha256=xiHZhRlCgRm6s8wRIZ300xBO5gUW1j2wqWZG9rj1npc,63642
-langroid-0.52.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.52.0.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.52.0.dist-info/RECORD,,
+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,,