langroid 0.41.0__tar.gz → 0.41.2__tar.gz

{langroid-0.41.0 → langroid-0.41.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.41.0
+Version: 0.41.2
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT
@@ -12,7 +12,7 @@ Requires-Dist: async-generator<2.0,>=1.10
 Requires-Dist: bs4<1.0.0,>=0.0.1
 Requires-Dist: cerebras-cloud-sdk<2.0.0,>=1.1.0
 Requires-Dist: colorlog<7.0.0,>=6.7.0
-Requires-Dist: docling<3.0.0,>=2.16.0
+Requires-Dist: docling<3.0.0,>=2.20.0
 Requires-Dist: docstring-parser<1.0,>=0.16
 Requires-Dist: duckduckgo-search<7.0.0,>=6.0.0
 Requires-Dist: exa-py>=1.8.7
@@ -96,7 +96,7 @@ Requires-Dist: psycopg2<3.0.0,>=2.9.7; extra == 'db'
 Requires-Dist: pymysql<2.0.0,>=1.1.0; extra == 'db'
 Requires-Dist: sqlalchemy<3.0.0,>=2.0.19; extra == 'db'
 Provides-Extra: doc-chat
-Requires-Dist: docling<3.0.0,>=2.16.0; extra == 'doc-chat'
+Requires-Dist: docling<3.0.0,>=2.20.0; extra == 'doc-chat'
 Requires-Dist: pdf2image<2.0.0,>=1.17.0; extra == 'doc-chat'
 Requires-Dist: pymupdf4llm<0.1.0,>=0.0.17; extra == 'doc-chat'
 Requires-Dist: pymupdf<2.0.0,>=1.23.3; extra == 'doc-chat'

{langroid-0.41.0 → langroid-0.41.2}/langroid/language_models/model_info.py

@@ -320,8 +320,15 @@ MODEL_INFO: Dict[str, ModelInfo] = {
 }
 
 
-def get_model_info(model: str | ModelName) -> ModelInfo:
+def get_model_info(
+    model: str | ModelName,
+    fallback_model: str | ModelName = "",
+) -> ModelInfo:
     """Get model information by name or enum value"""
+    return _get_model_info(model) or _get_model_info(fallback_model) or ModelInfo()
+
+
+def _get_model_info(model: str | ModelName) -> ModelInfo | None:
     if isinstance(model, str):
-        return MODEL_INFO.get(model) or ModelInfo()
-    return MODEL_INFO.get(model.value) or ModelInfo()
+        return MODEL_INFO.get(model)
+    return MODEL_INFO.get(model.value)

{langroid-0.41.0 → langroid-0.41.2}/langroid/language_models/openai_gpt.py

@@ -468,7 +468,10 @@ class OpenAIGPT(LanguageModel):
             self.supports_strict_tools = self.api_base is None
             self.supports_json_schema = (
                 self.api_base is None
-                and get_model_info(self.config.chat_model).has_structured_output
+                and get_model_info(  # look for family/provider-specific then generic
+                    self.chat_model_orig,  # e.g. "gemini/gemini-2.0-flash"
+                    self.config.chat_model,  # e.g. "gemini-2.0-flash"
+                ).has_structured_output
             )
 
         if settings.chat_model != "":
@@ -620,7 +623,10 @@ class OpenAIGPT(LanguageModel):
     def supports_functions_or_tools(self) -> bool:
         return (
             self.is_openai_chat_model()
-            and get_model_info(self.config.chat_model).has_tools
+            and get_model_info(
+                self.chat_model_orig,
+                self.config.chat_model,
+            ).has_tools
         )
 
     def is_openai_completion_model(self) -> bool:
@@ -644,7 +650,10 @@ class OpenAIGPT(LanguageModel):
         """
         List of params that are not supported by the current model
         """
-        model_info = get_model_info(self.config.chat_model)
+        model_info = get_model_info(
+            self.chat_model_orig,
+            self.config.chat_model,
+        )
         unsupported = set(model_info.unsupported_params)
         for param, model_list in OpenAI_API_ParamInfo().params.items():
             if (
@@ -659,7 +668,10 @@ class OpenAIGPT(LanguageModel):
         Map of param name -> new name for specific models.
         Currently main troublemaker is o1* series.
         """
-        return get_model_info(self.config.chat_model).rename_params
+        return get_model_info(
+            self.chat_model_orig,
+            self.config.chat_model,
+        ).rename_params
 
     def chat_context_length(self) -> int:
         """
@@ -671,7 +683,12 @@ class OpenAIGPT(LanguageModel):
             if self.config.use_completion_for_chat
             else self.config.chat_model
         )
-        return get_model_info(model).context_length
+        orig_model = (
+            self.config.completion_model
+            if self.config.use_completion_for_chat
+            else self.chat_model_orig
+        )
+        return get_model_info(orig_model, model).context_length
 
     def completion_context_length(self) -> int:
         """
@@ -683,7 +700,12 @@ class OpenAIGPT(LanguageModel):
             if self.config.use_chat_for_completion
             else self.config.completion_model
         )
-        return get_model_info(model).context_length
+        orig_model = (
+            self.chat_model_orig
+            if self.config.use_chat_for_completion
+            else self.config.completion_model
+        )
+        return get_model_info(orig_model, model).context_length
 
     def chat_cost(self) -> Tuple[float, float]:
         """
@@ -709,7 +731,10 @@ class OpenAIGPT(LanguageModel):
         return (
             self.config.stream
             and settings.stream
-            and get_model_info(self.config.chat_model).allows_streaming
+            and get_model_info(
+                self.chat_model_orig,
+                self.config.chat_model,
+            ).allows_streaming
             and not settings.quiet
         )
 

{langroid-0.41.0 → langroid-0.41.2}/langroid/parsing/url_loader.py

@@ -68,7 +68,11 @@ class URLLoader:
                     docs.extend(new_chunks)
                 else:
                     # Try to detect content type and handle accordingly
-                    headers = requests.head(url).headers
+                    try:
+                        headers = requests.head(url).headers
+                    except Exception as e:
+                        logging.warning(f"Error getting headers for {url}: {e}")
+                        headers = {}
                     content_type = headers.get("Content-Type", "").lower()
                     temp_file_suffix = None
                     if "application/pdf" in content_type:

{langroid-0.41.0 → langroid-0.41.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langroid"
-version = "0.41.0"
+version = "0.41.2"
 authors = [
     {name = "Prasad Chalasani", email = "pchalasani@gmail.com"},
 ]
@@ -15,7 +15,7 @@ dependencies = [
     "bs4<1.0.0,>=0.0.1",
     "cerebras-cloud-sdk<2.0.0,>=1.1.0",
     "colorlog<7.0.0,>=6.7.0",
-    "docling<3.0.0,>=2.16.0",
+    "docling<3.0.0,>=2.20.0",
     "docstring-parser<1.0,>=0.16",
     "duckduckgo-search<7.0.0,>=6.0.0",
     "exa-py>=1.8.7",
@@ -62,7 +62,7 @@ dependencies = [
 
 [project.optional-dependencies]
 doc-chat = [
-    "docling<3.0.0,>=2.16.0",
+    "docling<3.0.0,>=2.20.0",
     "pdf2image<2.0.0,>=1.17.0",
     "pymupdf4llm<0.1.0,>=0.0.17",
     "pymupdf<2.0.0,>=1.23.3",