langroid 0.41.0__py3-none-any.whl → 0.41.2__py3-none-any.whl

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/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/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.dist-info → langroid-0.41.2.dist-info}/METADATA

@@ -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.dist-info → langroid-0.41.2.dist-info}/RECORD

@@ -71,8 +71,8 @@ langroid/language_models/azure_openai.py,sha256=zNQzzsERxNestq-hFfQZbvTzK43G2vjR
 langroid/language_models/base.py,sha256=mN6HAjLgF2xpHObz5uPZ3JDID7jdTiRLEkoGgGrqLM8,25177
 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=GWqfo9nxUYM1qjuPMln7PPpvyjBFN_3qpFBoZJIc72M,11190
-langroid/language_models/openai_gpt.py,sha256=Xul-vN59x53HPJPFUH5HvzL3iekhjq5xllJ66aAkzfU,77887
+langroid/language_models/model_info.py,sha256=yKAaKoCanPoqaoHCzRVNPjg-M9a4S2Vm2AJGnwMeO-M,11360
+langroid/language_models/openai_gpt.py,sha256=hC6knKQ9s9-bNixPWduxLRhLxs3y1zyfW7hU8oWWtHQ,78681
 langroid/language_models/utils.py,sha256=L4_CbihDMTGcsg0TOG1Yd5JFEto46--h7CX_14m89sQ,5016
 langroid/language_models/prompt_formatter/__init__.py,sha256=2-5cdE24XoFDhifOLl8yiscohil1ogbP1ECkYdBlBsk,372
 langroid/language_models/prompt_formatter/base.py,sha256=eDS1sgRNZVnoajwV_ZIha6cba5Dt8xjgzdRbPITwx3Q,1221
@@ -91,7 +91,7 @@ langroid/parsing/routing.py,sha256=-FcnlqldzL4ZoxuDwXjQPNHgBe9F9-F4R6q7b_z9CvI,1
 langroid/parsing/search.py,sha256=YPCwezM0c4PWbNUMEmQ5RrJBtvX4aWZ1CMCJFs4sqFo,9806
 langroid/parsing/spider.py,sha256=hAVM6wxh1pQ0EN4tI5wMBtAjIk0T-xnpi-ZUzWybhos,3258
 langroid/parsing/table_loader.py,sha256=qNM4obT_0Y4tjrxNBCNUYjKQ9oETCZ7FbolKBTcz-GM,3410
-langroid/parsing/url_loader.py,sha256=JK48KktLRDBfjrt4nsUfy92M6yGdEeicAqOum2MdULM,4656
+langroid/parsing/url_loader.py,sha256=MPJFhAdMl4LYVtL9f8r1BOtTkDOg2-hKkANUBUoXCG0,4846
 langroid/parsing/urls.py,sha256=86omykgxo4hg2jyF10Ef-FJa9n6MgXdSXy2mImqgo5c,8076
 langroid/parsing/utils.py,sha256=ZWMS7oG04GUY9EAIwnFN6KKo_ePCKhqk_H8jW6TDT0s,12805
 langroid/parsing/web_search.py,sha256=wWSmV0METFTGPhHJIs-M4tog2Aur_75Pxr4a49cKDkU,7042
@@ -128,7 +128,7 @@ langroid/vector_store/pineconedb.py,sha256=otxXZNaBKb9f_H75HTaU3lMHiaR2NUp5MqwLZ
 langroid/vector_store/postgres.py,sha256=DQHd6dt-OcV_QVNm-ymn28rlTfhI6hqgcpLTPCsm0jI,15990
 langroid/vector_store/qdrantdb.py,sha256=v7TAsIoj_vxeKDYS9tpwJLBZA8fuTweTYxHo0X_uawM,17949
 langroid/vector_store/weaviatedb.py,sha256=ONEr2iGS0Ii73oMe7tRk6bB-BEXQUa70fYSrdI8d3yo,11481
-langroid-0.41.0.dist-info/METADATA,sha256=jCbP1nZgmhcN4XJE7eh8SBMbFEwNVV8yVoZqrmV8pCQ,61259
-langroid-0.41.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.41.0.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.41.0.dist-info/RECORD,,
+langroid-0.41.2.dist-info/METADATA,sha256=FWd7a2bi9c79wldH9KrVvq2U7Jtu_XeEWI9XcF17FpE,61259
+langroid-0.41.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.41.2.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.41.2.dist-info/RECORD,,