This is an automated email from the ASF dual-hosted git repository. vikramkoka pushed a commit to branch aip99-langchain-example in repository https://gitbox.apache.org/repos/asf/airflow.git
commit e98dfb48c257c72596b3f79b0a2d1fd468fa7503 Author: Vikram Koka <[email protected]> AuthorDate: Tue Jun 2 18:11:47 2026 -0700 Updated example_langchain_10k Updated this example in common AI provider to fix bug caused by the change in XCom data passing introduced by PR 67644 --- .../providers/common/ai/example_dags/example_langchain_10k.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/providers/common/ai/src/airflow/providers/common/ai/example_dags/example_langchain_10k.py b/providers/common/ai/src/airflow/providers/common/ai/example_dags/example_langchain_10k.py index d7c7b8a4f6c..4335f7bdcb5 100644 --- a/providers/common/ai/src/airflow/providers/common/ai/example_dags/example_langchain_10k.py +++ b/providers/common/ai/src/airflow/providers/common/ai/example_dags/example_langchain_10k.py @@ -461,7 +461,9 @@ def example_langchain_10k_analysis(): # [END 10k_langchain_decompose] @task - def extract_sub_questions(decomposed: dict) -> list[dict]: + def extract_sub_questions(decomposed) -> list[dict]: + if hasattr(decomposed, "model_dump"): + decomposed = decomposed.model_dump() return decomposed["sub_questions"] sub_questions = extract_sub_questions(decomposed) @@ -565,7 +567,9 @@ Cite specific data points and scores. # AnalysisReport); this task renders it as clean prose. # ------------------------------------------------------------------ @task - def format_report(report: dict) -> str: + def format_report(report) -> str: + if hasattr(report, "model_dump"): + report = report.model_dump() lines = [f"# Executive Summary\n\n{report['executive_summary']}"] if report.get("company_findings"):
