iterative event loop

Author: pgrayy

.github/workflows/integration-test.yml

@@ -3,20 +3,20 @@ name: Secure Integration test
 on:
   pull_request_target:
     branches: main
-  
+
 jobs:
   authorization-check:
     permissions: read-all
     runs-on: ubuntu-latest
-    outputs: 
+    outputs:
       approval-env: ${{ steps.collab-check.outputs.result }}
     steps:
       - name: Collaborator Check
         uses: actions/github-script@v7
         id: collab-check
         with:
           result-encoding: string
-          script: | 
+          script: |
             try {
               const permissionResponse = await github.rest.repos.getCollaboratorPermissionLevel({
                 owner: context.repo.owner,
@@ -31,7 +31,7 @@ jobs:
               } else {
                 console.log(`Verifed ${context.payload.pull_request.user.login} has write access. Auto Approving PR Checks.`)
                 return "auto-approve"
-              }             
+              }
             } catch (error) {
               console.log(`${context.payload.pull_request.user.login} does not have write access. Requiring Manual Approval to run PR Checks.`)
               return "manual-approval"
@@ -45,9 +45,9 @@ jobs:
       pull-requests: read
       contents: read
     steps:
-      - name: Configure Credentials 
+      - name: Configure Credentials
         uses: aws-actions/configure-aws-credentials@v4
-        with: 
+        with:
          role-to-assume: ${{ secrets.STRANDS_INTEG_TEST_ROLE }}
          aws-region: us-east-1
          mask-aws-account-id: true
@@ -70,5 +70,3 @@ jobs:
         id: tests
         run: |
           hatch test tests-integ
-      
-    

pyproject.toml

@@ -125,8 +125,8 @@ extra-dependencies = [
     "pytest-xdist>=3.0.0,<4.0.0",
 ]
 extra-args = [
-    "-n",
-    "auto",
+    # "-n",
+    # "auto",
     "-vv",
 ]
 
@@ -156,7 +156,7 @@ python = ["3.13", "3.12", "3.11", "3.10"]
 [tool.hatch.envs.hatch-test.scripts]
 run = [
     # excluding due to A2A and OTEL http exporter dependency conflict
-    "pytest{env:HATCH_TEST_ARGS:} {args} --ignore=tests/multiagent/a2a"
+    "pytest{env:HATCH_TEST_ARGS:} {args} tests-integ/test_model_litellm.py::test_agent --ignore=tests/multiagent/a2a -s -vv"
 ]
 run-cov = [
     # excluding due to A2A and OTEL http exporter dependency conflict

src/strands/agent/agent.py

@@ -546,7 +546,7 @@ def _execute_event_loop_cycle(self, callback_handler: Callable[..., Any], kwargs
 
         try:
             # Execute the main event loop cycle
-            stop_reason, message, metrics, state = event_loop_cycle(
+            events = event_loop_cycle(
                 model=model,
                 system_prompt=system_prompt,
                 messages=messages,  # will be modified by event_loop_cycle
@@ -559,6 +559,11 @@ def _execute_event_loop_cycle(self, callback_handler: Callable[..., Any], kwargs
                 event_loop_parent_span=self.trace_span,
                 **kwargs,
             )
+            for event in events:
+                if "callback" in event:
+                    callback_handler(**event["callback"])
+
+            stop_reason, message, metrics, state = event["stop"]
 
             return AgentResult(stop_reason, message, metrics, state)
 

src/strands/event_loop/__init__.py

@@ -4,6 +4,6 @@
 iterative manner.
 """
 
-from . import error_handler, event_loop, message_processor
+from . import event_loop, message_processor
 
-__all__ = ["error_handler", "event_loop", "message_processor"]
+__all__ = ["event_loop", "message_processor"]