langroid 0.59.2__py3-none-any.whl → 0.59.4__py3-none-any.whl

langroid/language_models/model_info.py

@@ -74,10 +74,10 @@ class GeminiModel(ModelName):
     GEMINI_1_5_PRO = "gemini-1.5-pro"
     GEMINI_2_5_PRO = "gemini-2.5-pro"
     GEMINI_2_5_FLASH = "gemini-2.5-flash"
-    GEMINI_2_5_FLASH_LITE_PREVIEW = "gemini-2.5-flash-lite-preview-06-17"
+    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite"
     GEMINI_2_PRO = "gemini-2.0-pro-exp-02-05"
     GEMINI_2_FLASH = "gemini-2.0-flash"
-    GEMINI_2_FLASH_LITE = "gemini-2.0-flash-lite-preview"
+    GEMINI_2_FLASH_LITE = "gemini-2.0-flash-lite"
     GEMINI_2_FLASH_THINKING = "gemini-2.0-flash-thinking-exp"
 
 
@@ -90,6 +90,9 @@ class OpenAI_API_ParamInfo(BaseModel):
     params: Dict[str, List[str]] = dict(
         reasoning_effort=[
             OpenAIChatModel.O3_MINI.value,
+            GeminiModel.GEMINI_2_5_PRO.value,
+            GeminiModel.GEMINI_2_5_FLASH.value,
+            GeminiModel.GEMINI_2_5_FLASH_LITE.value,
         ],
     )
     # model-specific params in extra_body
@@ -438,7 +441,7 @@ MODEL_INFO: Dict[str, ModelInfo] = {
         input_cost_per_million=0.075,
         output_cost_per_million=0.30,
         rename_params={"max_tokens": "max_completion_tokens"},
-        description="Gemini 2.0 Flash Lite Preview",
+        description="Gemini 2.0 Flash Lite",
     ),
     GeminiModel.GEMINI_1_5_FLASH.value: ModelInfo(
         name=GeminiModel.GEMINI_1_5_FLASH.value,
@@ -503,8 +506,8 @@ MODEL_INFO: Dict[str, ModelInfo] = {
         rename_params={"max_tokens": "max_completion_tokens"},
         description="Gemini 2.5 Flash",
     ),
-    GeminiModel.GEMINI_2_5_FLASH_LITE_PREVIEW.value: ModelInfo(
-        name=GeminiModel.GEMINI_2_5_FLASH_LITE_PREVIEW.value,
+    GeminiModel.GEMINI_2_5_FLASH_LITE.value: ModelInfo(
+        name=GeminiModel.GEMINI_2_5_FLASH_LITE.value,
         provider=ModelProvider.GOOGLE,
         context_length=65_536,
         max_output_tokens=65_536,
@@ -512,7 +515,7 @@ MODEL_INFO: Dict[str, ModelInfo] = {
         cached_cost_per_million=0.025,
         output_cost_per_million=0.40,
         rename_params={"max_tokens": "max_completion_tokens"},
-        description="Gemini 2.5 Flash Lite Preview",
+        description="Gemini 2.5 Flash Lite",
     ),
 }
 

langroid/language_models/openai_gpt.py

@@ -215,7 +215,7 @@ class OpenAICallParams(BaseModel):
     top_p: float | None = None
     reasoning_effort: str | None = None  # or "low" or "high" or "medium"
     top_logprobs: int | None = None  # if int, requires logprobs=True
-    n: int = 1  # how many completions to generate (n > 1 is NOT handled now)
+    n: int | None = None  # how many completions to generate (n > 1 is NOT handled now)
     stop: str | List[str] | None = None  # (list of) stop sequence(s)
     seed: int | None = None
     user: str | None = None  # user id for tracking
@@ -323,37 +323,15 @@ class OpenAIGPTConfig(LLMConfig):
         self, *, update: Mapping[str, Any] | None = None, deep: bool = False
     ) -> "OpenAIGPTConfig":
         """
-        Override model_copy to handle unpicklable fields properly.
+        Copy config while preserving nested model instances and subclasses.
 
-        This preserves fields like http_client_factory during normal copying
-        while still allowing exclusion for pickling operations.
+        Important: Avoid reconstructing via `model_dump` as that coerces nested
+        models to their annotated base types (dropping subclass-only fields).
+        Instead, defer to Pydantic's native `model_copy`, which keeps nested
+        `BaseModel` instances (and their concrete subclasses) intact.
         """
-        # Save references to unpicklable fields
-        http_client_factory = self.http_client_factory
-        streamer = self.streamer
-        streamer_async = self.streamer_async
-
-        # Get the current model data, excluding problematic fields
-        data = self.model_dump(
-            exclude={"http_client_factory", "streamer", "streamer_async"}
-        )
-
-        # Apply any updates
-        if update:
-            data.update(update)
-
-        # Create a new instance with the copied data
-        new_instance = self.__class__(**data)
-
-        # Restore the unpicklable fields if they weren't overridden by update
-        if "http_client_factory" not in (update or {}):
-            new_instance.http_client_factory = http_client_factory
-        if "streamer" not in (update or {}):
-            new_instance.streamer = streamer
-        if "streamer_async" not in (update or {}):
-            new_instance.streamer_async = streamer_async
-
-        return new_instance
+        # Delegate to BaseSettings/BaseModel implementation to preserve types
+        return super().model_copy(update=update, deep=deep)  # type: ignore[return-value]
 
     def _validate_litellm(self) -> None:
         """
@@ -441,8 +419,9 @@ class OpenAIGPT(LanguageModel):
         Args:
             config: configuration for openai-gpt model
         """
-        # copy the config to avoid modifying the original
-        config = config.model_copy()
+        # copy the config to avoid modifying the original; deep to decouple
+        # nested models while preserving their concrete subclasses
+        config = config.model_copy(deep=True)
         super().__init__(config)
         self.config: OpenAIGPTConfig = config
         # save original model name such as `provider/model` before

{langroid-0.59.2.dist-info → langroid-0.59.4.dist-info}/METADATA

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

{langroid-0.59.2.dist-info → langroid-0.59.4.dist-info}/RECORD

@@ -77,8 +77,8 @@ langroid/language_models/base.py,sha256=uRi-XtQhsEBN702_lGepxlNQ0e8gaPgaXjULYn2r
 langroid/language_models/client_cache.py,sha256=c8tAUTKxbd-CVGRy9WPYhI1pLtarFWWwX6-qm4ZXaqk,9399
 langroid/language_models/config.py,sha256=rhEZ9Wwq42q1uW_-oRFbilJhIw1vYlTvYqIk_7pHiSY,397
 langroid/language_models/mock_lm.py,sha256=tA9JpURznsMZ59iRhFYMmaYQzAc0D0BT-PiJIV58sAk,4079
-langroid/language_models/model_info.py,sha256=5bq2CzBzNVBSixxxcbNQ1-VIdjBtmbdm5XP4_z5xM88,19236
-langroid/language_models/openai_gpt.py,sha256=K0SSRQYDBb_2ku9TPxGBStsimxBKw-EmmySuagqfgSs,94072
+langroid/language_models/model_info.py,sha256=iJXNriyibD0S-7NbGoUgQWzotNRZ614AfadCtF3ernU,19321
+langroid/language_models/openai_gpt.py,sha256=qz-KN8phkVglZ6GeqXDwW60Dr1--ZhhKdTEaXwftRZA,93543
 langroid/language_models/provider_params.py,sha256=upG4cBrX8fcvAo1g7fcsv-rBbsfypIqcDRRV9m1hohU,4846
 langroid/language_models/utils.py,sha256=n55Oe2_V_4VNGhytvPWLYC-0tFS07RTjN83KWl-p_MI,6032
 langroid/language_models/prompt_formatter/__init__.py,sha256=2-5cdE24XoFDhifOLl8yiscohil1ogbP1ECkYdBlBsk,372
@@ -139,7 +139,7 @@ langroid/vector_store/pineconedb.py,sha256=7V0Bkt4ZrOR3V90tdXvdFmyNGuww7SFdyPq7-
 langroid/vector_store/postgres.py,sha256=TY_VshimwFZglYgKYm7Qn1F-dCSL8GsXRTgmh7VTe9c,16110
 langroid/vector_store/qdrantdb.py,sha256=mqxMOrcLAQpl0opuL8vXhdIt6ppv2zYyAvddHZoEW0Y,19184
 langroid/vector_store/weaviatedb.py,sha256=BS95bxVKNYfQc9VPb85a1HlcgnXfAkgMzjydnjCgRHc,11853
-langroid-0.59.2.dist-info/METADATA,sha256=-v3j3a3T4jAnC6xg0S7SRUgUdg8T2L9y1hSmO5lSy4Q,66517
-langroid-0.59.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.59.2.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.59.2.dist-info/RECORD,,
+langroid-0.59.4.dist-info/METADATA,sha256=5sDrQS2lyGqbo7U02g4Um4SeaoI627fsA19LITqQcxg,66517
+langroid-0.59.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.59.4.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.59.4.dist-info/RECORD,,