langroid 0.59.3__tar.gz → 0.59.5__tar.gz

{langroid-0.59.3 → langroid-0.59.5}/PKG-INFO

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

{langroid-0.59.3 → langroid-0.59.5}/langroid/agent/chat_agent.py

@@ -264,12 +264,17 @@ class ChatAgent(Agent):
         Revisit later.
         """
         agent_cls = type(self)
-        config_copy = copy.deepcopy(self.config)
+        # Use model_copy to preserve Pydantic subclass types (like MockLMConfig)
+        # instead of deepcopy which loses subclass information
+        config_copy = self.config.model_copy(deep=True)
         config_copy.name = f"{config_copy.name}-{i}"
         new_agent = agent_cls(config_copy)
         new_agent.system_tool_instructions = self.system_tool_instructions
         new_agent.system_tool_format_instructions = self.system_tool_format_instructions
         new_agent.llm_tools_map = self.llm_tools_map
+        new_agent.llm_tools_known = self.llm_tools_known
+        new_agent.llm_tools_handled = self.llm_tools_handled
+        new_agent.llm_tools_usable = self.llm_tools_usable
         new_agent.llm_functions_map = self.llm_functions_map
         new_agent.llm_functions_handled = self.llm_functions_handled
         new_agent.llm_functions_usable = self.llm_functions_usable

{langroid-0.59.3 → langroid-0.59.5}/langroid/language_models/openai_gpt.py

@@ -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.3 → langroid-0.59.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langroid"
-version = "0.59.3"
+version = "0.59.5"
 authors = [
     {name = "Prasad Chalasani", email = "pchalasani@gmail.com"},
 ]