Improve tool and state isolation in .llm() method (#40)

- Add _clear_tools() and _disable_state() methods to ensure clean tool and state isolation
- Modify .llm() method to create a clean copy of the Promptic instance
- Bump version to 5.3.1 to reflect the isolation improvement
- Add test case to verify tool isolation between parent and child functions

Author: knowsuchagency

promptic.py

@@ -17,7 +17,7 @@
 from litellm import completion as litellm_completion
 from pydantic import BaseModel
 
-__version__ = "5.3.0"
+__version__ = "5.3.1"
 
 SystemPrompt = Optional[Union[str, List[str], List[Dict[str, str]]]]
 
@@ -360,11 +360,22 @@ def decorate(cls, func: Callable = None, **kwargs):
         instance = cls(**kwargs)
         return instance._decorator(func) if func else instance
 
+    def _clear_tools(self):
+        self.tools = {}
+        self.tool_definitions = None
+
+    def _disable_state(self):
+        self.state = None
+
     def llm(self, func: Callable = None, **kwargs):
         """Decorate a function with the Promptic instance."""
         new_instance = copy.copy(self)
+        new_instance._clear_tools()
+        new_instance._disable_state()
+
         for key, value in kwargs.items():
             setattr(new_instance, key, value)
+
         return new_instance._decorator(func) if func else new_instance._decorator
 
     def _decorator(self, func: Callable):

tests/test_promptic.py

@@ -1623,3 +1623,48 @@ def test_completion_method(model, create_completion_fn):
     ):
         response = p_with_state.completion(messages)
         assert "Paris" in response.choices[0].message.content
+
+
+@pytest.mark.parametrize("model", CHEAP_MODELS)
+@pytest.mark.parametrize(
+    "create_completion_fn", [openai_completion_fn, litellm_completion]
+)
+def test_tool_isolation_with_llm_method(model, create_completion_fn):
+    """Test that tools don't leak between parent and child functions when using llm method"""
+    if create_completion_fn == openai_completion_fn and not model.startswith("gpt"):
+        pytest.skip("Non-GPT models are not supported with OpenAI client")
+
+    p = Promptic(
+        model=model,
+        temperature=0,
+        timeout=5,
+        create_completion_fn=create_completion_fn,
+    )
+
+    @retry(
+        wait=wait_exponential(multiplier=1, min=4, max=10),
+        retry=retry_if_exception_type(ERRORS),
+    )
+    @p.llm
+    def parent_function(command):
+        """Respond to: {command}"""
+
+    # Add a tool to the parent function
+    @parent_function.tool
+    def get_time():
+        """Get the current time"""
+        return "12:00 PM"
+
+    # Create a child function using the parent's llm method
+    @p.llm
+    def child_function(command):
+        """Respond to: {command}"""
+
+    # Verify parent function has access to the tool
+    result = parent_function("What time is it?")
+    assert "12:00" in result
+
+    # Verify child function does not have access to the tool
+    # It should respond without using the tool
+    result = child_function("What time is it?")
+    assert "12:00" not in result