Skip to content

Commit 76aeb44

Browse files
committed
fix: add iteration counter reset on agent init
Handle 'init_event_loop' event to reset iteration count, ensuring proper limit tracking for new agent sessions.
1 parent 19bab2f commit 76aeb44

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

packages/cdk/lambda-python/generic-agent-core-runtime/src/agent.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,10 @@ def __init__(self):
3434
def set_session_info(self, session_id: str, trace_id: str):
3535
"""Set session and trace IDs"""
3636
self.tool_manager.set_session_info(session_id, trace_id)
37-
self.iteration_count = 0
3837

3938
def iteration_limit_handler(self, **ev):
39+
if ev.get("init_event_loop"):
40+
self.iteration_count = 0
4041
if ev.get("start_event_loop"):
4142
self.iteration_count += 1
4243
if self.iteration_count > self.max_iterations:

packages/cdk/test/__snapshots__/generative-ai-use-cases.test.ts.snap

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4992,7 +4992,7 @@ exports[`GenerativeAiUseCases matches the snapshot (closed network mode) 4`] = `
49924992
"AgentCoreRuntimeName": "GenericAgentCoreRuntime",
49934993
"CustomConfig": {
49944994
"containerImageUri": {
4995-
"Fn::Sub": "123456890123.dkr.ecr.us-east-1.\${AWS::URLSuffix}/cdk-hnb659fds-container-assets-123456890123-us-east-1:498ade13deb7f88e2c00806dbfebfcfcb6dd51e45fd26075fa88a8e2f33064f5",
4995+
"Fn::Sub": "123456890123.dkr.ecr.us-east-1.\${AWS::URLSuffix}/cdk-hnb659fds-container-assets-123456890123-us-east-1:89f9ea87f3ee318598f0b63b57f12a3762e5abe60ad4c925b317d2cf2c737d0d",
49964996
},
49974997
"environmentVariables": {
49984998
"FILE_BUCKET": {
@@ -26205,7 +26205,7 @@ exports[`GenerativeAiUseCases matches the snapshot 4`] = `
2620526205
"AgentCoreRuntimeName": "GenericAgentCoreRuntime",
2620626206
"CustomConfig": {
2620726207
"containerImageUri": {
26208-
"Fn::Sub": "123456890123.dkr.ecr.us-east-1.\${AWS::URLSuffix}/cdk-hnb659fds-container-assets-123456890123-us-east-1:498ade13deb7f88e2c00806dbfebfcfcb6dd51e45fd26075fa88a8e2f33064f5",
26208+
"Fn::Sub": "123456890123.dkr.ecr.us-east-1.\${AWS::URLSuffix}/cdk-hnb659fds-container-assets-123456890123-us-east-1:89f9ea87f3ee318598f0b63b57f12a3762e5abe60ad4c925b317d2cf2c737d0d",
2620926209
},
2621026210
"environmentVariables": {
2621126211
"FILE_BUCKET": {

0 commit comments

Comments
 (0)