ccoliu commented on code in PR #76201:
URL: https://github.com/apache/airflow/pull/76201#discussion_r4438157420
##########
providers/common/ai/src/airflow/providers/common/ai/durable/strands.py:
##########
@@ -132,15 +132,9 @@ def research(question: str) -> str:
)
storage = TaskStateStoreStorage(context["task_state_store"])
if session_id is None:
- ti = context["ti"]
# A hash: short (task state store keys are at most 512 characters) and
free of the
# characters Strands refuses in a session id, whatever the Dag, run
and task ids hold.
- session_id = compute_identity_key(
- dag_id=ti.dag_id,
- task_id=ti.task_id,
- run_id=ti.run_id,
- map_index=ti.map_index if ti.map_index is not None else -1,
- )
+ session_id = compute_identity_key_for_context(context)
Review Comment:
Got it, thanks! Fixing these.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
