fix(agent): restore end_turn in run_conversation finally
Keep relay turn teardown when clearing activity labels after a turn exits.
This commit is contained in:
+7
-3
@@ -7563,6 +7563,13 @@ class AIAgent:
|
||||
finish_task_run(**task_context, error=exc)
|
||||
raise
|
||||
finally:
|
||||
try:
|
||||
if relay_turn is not None:
|
||||
relay_runtime.SESSION_COORDINATOR.end_turn(
|
||||
relay_turn,
|
||||
outcome=relay_outcome,
|
||||
)
|
||||
finally:
|
||||
try:
|
||||
if relay_lease is not None:
|
||||
relay_runtime.SESSION_COORDINATOR.release_conversation(
|
||||
@@ -7581,9 +7588,6 @@ class AIAgent:
|
||||
reset_accounting_context(acct_token)
|
||||
if token is not None:
|
||||
reset_conversation_context(token)
|
||||
reset_accounting_context(acct_token)
|
||||
if token is not None:
|
||||
reset_conversation_context(token)
|
||||
|
||||
def chat(self, message: str, stream_callback: Optional[callable] = None) -> str:
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user