atomic-agents 2.7.3__tar.gz → 2.7.4__tar.gz

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomic-agents
-Version: 2.7.3
+Version: 2.7.4
 Summary: A versatile framework for creating and managing intelligent agents.
 Project-URL: Homepage, https://github.com/BrainBlend-AI/atomic-agents
 Project-URL: Repository, https://github.com/BrainBlend-AI/atomic-agents
@@ -321,6 +321,10 @@ For a complete list of examples, see the [examples directory](/atomic-examples/)
 
 ## 🚀 Version 2.0 Released!
 
+[![What's New in V2.0](./.assets/whats-new-v2-thumbnail.png)](https://www.youtube.com/watch?v=ro-pM2bPmr0)
+
+[**Watch: What's New in Atomic Agents V2.0**](https://www.youtube.com/watch?v=ro-pM2bPmr0)
+
 **Atomic Agents v2.0 is here with major improvements!** This release includes breaking changes that significantly improve the developer experience:
 
 ### Key Changes in v2.0:

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/README.md

@@ -290,6 +290,10 @@ For a complete list of examples, see the [examples directory](/atomic-examples/)
 
 ## 🚀 Version 2.0 Released!
 
+[![What's New in V2.0](./.assets/whats-new-v2-thumbnail.png)](https://www.youtube.com/watch?v=ro-pM2bPmr0)
+
+[**Watch: What's New in Atomic Agents V2.0**](https://www.youtube.com/watch?v=ro-pM2bPmr0)
+
 **Atomic Agents v2.0 is here with major improvements!** This release includes breaking changes that significantly improve the developer experience:
 
 ### Key Changes in v2.0:

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/atomic-agents/atomic_agents/__init__.py

@@ -6,8 +6,10 @@ Atomic Agents - A modular framework for building AI agents.
 from .agents.atomic_agent import AtomicAgent, AgentConfig, BasicChatInputSchema, BasicChatOutputSchema
 from .base import BaseIOSchema, BaseTool, BaseToolConfig
 
-# Version info
-__version__ = "2.0.0"
+# Version info - read from pyproject.toml via package metadata
+from importlib.metadata import version as _version
+
+__version__ = _version("atomic-agents")
 
 __all__ = [
     "AtomicAgent",

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/atomic-agents/atomic_agents/agents/atomic_agent.py

@@ -249,6 +249,19 @@ class AtomicAgent[InputSchema: BaseIOSchema, OutputSchema: BaseIOSchema]:
         self.messages = self._build_system_messages()
         self.messages += self.history.get_history()
 
+    def _get_completion_kwargs(self) -> Dict[str, Any]:
+        """
+        Build kwargs for Instructor completion calls.
+
+        Instructor defaults `strict=True`, which forces enum fields to receive enum
+        instances instead of allowing Pydantic's normal coercion from strings. We
+        default to `strict=None` here so the output schema's own Pydantic behavior
+        applies unless callers explicitly override it via `model_api_parameters`.
+        """
+        completion_kwargs = dict(self.model_api_parameters)
+        completion_kwargs.setdefault("strict", None)
+        return completion_kwargs
+
     def _build_tools_definition(self) -> Optional[List[Dict[str, Any]]]:
         """
         Build the tools definition that Instructor sends for TOOLS mode.
@@ -443,7 +456,7 @@ class AtomicAgent[InputSchema: BaseIOSchema, OutputSchema: BaseIOSchema]:
             messages=self.messages,
             model=self.model,
             response_model=self.output_schema,
-            **self.model_api_parameters,
+            **self._get_completion_kwargs(),
         )
         self.history.add_message(self.assistant_role, response)
         self._prepare_messages()
@@ -477,18 +490,20 @@ class AtomicAgent[InputSchema: BaseIOSchema, OutputSchema: BaseIOSchema]:
             model=self.model,
             messages=self.messages,
             response_model=self.output_schema,
-            **self.model_api_parameters,
+            **self._get_completion_kwargs(),
             stream=True,
         )
 
+        last_response = None
         for partial_response in response_stream:
+            last_response = partial_response
             yield partial_response
 
-        full_response_content = self.output_schema(**partial_response.model_dump())
-        self.history.add_message(self.assistant_role, full_response_content)
-        self._prepare_messages()
-
-        return full_response_content
+        if last_response:
+            full_response_content = self.output_schema(**last_response.model_dump())
+            self.history.add_message(self.assistant_role, full_response_content)
+            self._prepare_messages()
+            return full_response_content
 
     async def run_async(self, user_input: Optional[InputSchema] = None) -> OutputSchema:
         """
@@ -513,7 +528,7 @@ class AtomicAgent[InputSchema: BaseIOSchema, OutputSchema: BaseIOSchema]:
         self._prepare_messages()
 
         response = await self.client.chat.completions.create(
-            model=self.model, messages=self.messages, response_model=self.output_schema, **self.model_api_parameters
+            model=self.model, messages=self.messages, response_model=self.output_schema, **self._get_completion_kwargs()
         )
 
         self.history.add_message(self.assistant_role, response)
@@ -542,7 +557,7 @@ class AtomicAgent[InputSchema: BaseIOSchema, OutputSchema: BaseIOSchema]:
             model=self.model,
             messages=self.messages,
             response_model=self.output_schema,
-            **self.model_api_parameters,
+            **self._get_completion_kwargs(),
             stream=True,
         )
 

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/atomic-agents/atomic_agents/context/chat_history.py

@@ -204,12 +204,12 @@ class ChatHistory:
         """
         return self.current_turn_id
 
-    def delete_turn_id(self, turn_id: int):
+    def delete_turn_id(self, turn_id: str):
         """
         Delete messages from the history by its turn ID.
 
         Args:
-            turn_id (int): The turn ID of the message to delete.
+            turn_id (str): The turn ID of the message to delete.
 
         Returns:
             str: A success message with the deleted turn ID.

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/atomic-agents/tests/agents/test_atomic_agent.py

@@ -1,6 +1,8 @@
 import pytest
 from unittest.mock import Mock, call, patch
+from enum import Enum
 from pydantic import BaseModel, Field
+from pydantic import ValidationError
 import instructor
 from atomic_agents import (
     BaseIOSchema,
@@ -148,6 +150,73 @@ def test_initialization_without_max_tokens(mock_instructor, mock_history, mock_s
     assert agent.model_api_parameters["max_tokens"] == 1024
 
 
+def test_run_uses_pydantic_default_strictness_for_enum_output(mock_history, mock_system_prompt_generator):
+    class Topic(Enum):
+        FOOD = "food"
+        OTHER = "other"
+
+    class EnumOutputSchema(BaseIOSchema):
+        """Output schema with an enum field for validation tests."""
+
+        topic: Topic = Field(...)
+
+    enum_instructor = Mock(spec=instructor.Instructor)
+    enum_instructor.chat = Mock()
+    enum_instructor.chat.completions = Mock()
+
+    def mock_create(*args, **kwargs):
+        return kwargs["response_model"].model_validate({"topic": "food"}, strict=kwargs["strict"])
+
+    enum_instructor.chat.completions.create.side_effect = mock_create
+
+    config = AgentConfig(
+        client=enum_instructor,
+        model="gpt-5-mini",
+        history=mock_history,
+        system_prompt_generator=mock_system_prompt_generator,
+    )
+    agent = AtomicAgent[BasicChatInputSchema, EnumOutputSchema](config)
+
+    result = agent.run()
+
+    assert result.topic is Topic.FOOD
+    assert enum_instructor.chat.completions.create.call_args.kwargs["strict"] is None
+
+
+def test_run_respects_explicit_strict_override_for_enum_output(mock_history, mock_system_prompt_generator):
+    class Topic(Enum):
+        FOOD = "food"
+        OTHER = "other"
+
+    class EnumOutputSchema(BaseIOSchema):
+        """Output schema with an enum field for validation tests."""
+
+        topic: Topic = Field(...)
+
+    enum_instructor = Mock(spec=instructor.Instructor)
+    enum_instructor.chat = Mock()
+    enum_instructor.chat.completions = Mock()
+
+    def mock_create(*args, **kwargs):
+        return kwargs["response_model"].model_validate({"topic": "food"}, strict=kwargs["strict"])
+
+    enum_instructor.chat.completions.create.side_effect = mock_create
+
+    config = AgentConfig(
+        client=enum_instructor,
+        model="gpt-5-mini",
+        history=mock_history,
+        system_prompt_generator=mock_system_prompt_generator,
+        model_api_parameters={"strict": True},
+    )
+    agent = AtomicAgent[BasicChatInputSchema, EnumOutputSchema](config)
+
+    with pytest.raises(ValidationError):
+        agent.run()
+
+    assert enum_instructor.chat.completions.create.call_args.kwargs["strict"] is True
+
+
 def test_initialization_system_role_equals_developer(mock_instructor, mock_history, mock_system_prompt_generator):
     config = AgentConfig(
         client=mock_instructor,

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "atomic-agents"
-version = "2.7.3"
+version = "2.7.4"
 description = "A versatile framework for creating and managing intelligent agents."
 readme = "README.md"
 license = "MIT"

{atomic_agents-2.7.3 → atomic_agents-2.7.4}/uv.lock

@@ -194,7 +194,7 @@ wheels = [
 
 [[package]]
 name = "atomic-agents"
-version = "2.7.3"
+version = "2.7.4"
 source = { editable = "." }
 dependencies = [
     { name = "gitpython" },