updating tools version

tests/test_flow_persistence.py

@@ -7,7 +7,7 @@ import pytest
 from pydantic import BaseModel
 
 from crewai.flow.flow import Flow, FlowState, start
-from crewai.flow.persistence import FlowPersistence, persist
+from crewai.flow.persistence import persist
 from crewai.flow.persistence.sqlite import SQLiteFlowPersistence
 
 
@@ -21,20 +21,20 @@ def test_persist_decorator_saves_state(tmp_path):
     """Test that @persist decorator saves state in SQLite."""
     db_path = os.path.join(tmp_path, "test_flows.db")
     persistence = SQLiteFlowPersistence(db_path)
-    
+
     class TestFlow(Flow[Dict[str, str]]):
         initial_state = dict()  # Use dict instance as initial state
-        
+
         @start()
         @persist(persistence)
         def init_step(self):
             self.state["message"] = "Hello, World!"
             self.state["id"] = "test-uuid"  # Ensure we have an ID for persistence
-    
+
     # Run flow and verify state is saved
     flow = TestFlow(persistence=persistence)
     flow.kickoff()
-    
+
     # Load state from DB and verify
     saved_state = persistence.load_state(flow.state["id"])
     assert saved_state is not None
@@ -45,20 +45,20 @@ def test_structured_state_persistence(tmp_path):
     """Test persistence with Pydantic model state."""
     db_path = os.path.join(tmp_path, "test_flows.db")
     persistence = SQLiteFlowPersistence(db_path)
-    
+
     class StructuredFlow(Flow[TestState]):
         initial_state = TestState
-        
+
         @start()
         @persist(persistence)
         def count_up(self):
             self.state.counter += 1
             self.state.message = f"Count is {self.state.counter}"
-    
+
     # Run flow and verify state changes are saved
     flow = StructuredFlow(persistence=persistence)
     flow.kickoff()
-    
+
     # Load and verify state
     saved_state = persistence.load_state(flow.state.id)
     assert saved_state is not None
@@ -70,46 +70,46 @@ def test_flow_state_restoration(tmp_path):
     """Test restoring flow state from persistence with various restoration methods."""
     db_path = os.path.join(tmp_path, "test_flows.db")
     persistence = SQLiteFlowPersistence(db_path)
-    
+
     # First flow execution to create initial state
     class RestorableFlow(Flow[TestState]):
         initial_state = TestState
-        
+
         @start()
         @persist(persistence)
         def set_message(self):
             self.state.message = "Original message"
             self.state.counter = 42
-    
+
     # Create and persist initial state
     flow1 = RestorableFlow(persistence=persistence)
     flow1.kickoff()
     original_uuid = flow1.state.id
-    
+
     # Test case 1: Restore using restore_uuid with field override
     flow2 = RestorableFlow(
         persistence=persistence,
         restore_uuid=original_uuid,
         counter=43,  # Override counter
     )
-    
+
     # Verify state restoration and selective field override
     assert flow2.state.id == original_uuid
     assert flow2.state.message == "Original message"  # Preserved
     assert flow2.state.counter == 43  # Overridden
-    
+
     # Test case 2: Restore using kwargs['id']
     flow3 = RestorableFlow(
         persistence=persistence,
         id=original_uuid,
         message="Updated message",  # Override message
     )
-    
+
     # Verify state restoration and selective field override
     assert flow3.state.id == original_uuid
     assert flow3.state.counter == 42  # Preserved
     assert flow3.state.message == "Updated message"  # Overridden
-    
+
     # Test case 3: Verify error on conflicting IDs
     with pytest.raises(ValueError) as exc_info:
         RestorableFlow(
@@ -118,7 +118,7 @@ def test_flow_state_restoration(tmp_path):
             id="different-id",  # Conflict with restore_uuid
         )
     assert "Conflicting IDs provided" in str(exc_info.value)
-    
+
     # Test case 4: Verify error on non-existent restore_uuid
     with pytest.raises(ValueError) as exc_info:
         RestorableFlow(
@@ -126,7 +126,7 @@ def test_flow_state_restoration(tmp_path):
             restore_uuid="non-existent-uuid",
         )
     assert "No state found" in str(exc_info.value)
-    
+
     # Test case 5: Allow new state creation with kwargs['id']
     new_uuid = "new-flow-id"
     flow4 = RestorableFlow(
@@ -135,7 +135,7 @@ def test_flow_state_restoration(tmp_path):
         message="New message",
         counter=100,
     )
-    
+
     # Verify new state creation with provided ID
     assert flow4.state.id == new_uuid
     assert flow4.state.message == "New message"
@@ -146,25 +146,25 @@ def test_multiple_method_persistence(tmp_path):
     """Test state persistence across multiple method executions."""
     db_path = os.path.join(tmp_path, "test_flows.db")
     persistence = SQLiteFlowPersistence(db_path)
-    
+
     class MultiStepFlow(Flow[TestState]):
         initial_state = TestState
-        
+
         @start()
         @persist(persistence)
         def step_1(self):
             self.state.counter = 1
             self.state.message = "Step 1"
-        
+
         @start()
         @persist(persistence)
         def step_2(self):
             self.state.counter = 2
             self.state.message = "Step 2"
-    
+
     flow = MultiStepFlow(persistence=persistence)
     flow.kickoff()
-    
+
     # Load final state
     final_state = persistence.load_state(flow.state.id)
     assert final_state is not None
@@ -176,20 +176,20 @@ def test_persistence_error_handling(tmp_path):
     """Test error handling in persistence operations."""
     db_path = os.path.join(tmp_path, "test_flows.db")
     persistence = SQLiteFlowPersistence(db_path)
-    
+
     class InvalidFlow(Flow[TestState]):
         # Missing id field in initial state
         class InvalidState(BaseModel):
             value: str = ""
-            
+
         initial_state = InvalidState
-        
+
         @start()
         @persist(persistence)
         def will_fail(self):
             self.state.value = "test"
-    
+
     with pytest.raises(ValueError) as exc_info:
         flow = InvalidFlow(persistence=persistence)
-    
+
     assert "must have an 'id' field" in str(exc_info.value)