Add raw_generated_gql to intermediate_steps when verify_gql is True

jordimas · jordimas · commit 088755e7c179 · 2025-05-03T14:39:04.000+02:00
diff --git a/src/langchain_google_spanner/graph_qa.py b/src/langchain_google_spanner/graph_qa.py
@@ -301,6 +301,7 @@ def _call(
         generated_gql = extract_gql(gen_response)
 
         if self.verify_gql:
+            intermediate_steps.append({"raw_generated_gql": generated_gql})
             verify_response = self.gql_verify_chain.invoke(
                 {
                     "question": question,

Original file line number	Diff line number	Diff line change
`@@ -301,6 +301,7 @@ def _call(`
`301`	`301`	`generated_gql = extract_gql(gen_response)`
`302`	`302`
`303`	`303`	`if self.verify_gql:`
	`304`	`+ intermediate_steps.append({"raw_generated_gql": generated_gql})`
`304`	`305`	`verify_response = self.gql_verify_chain.invoke(`
`305`	`306`	`{`
`306`	`307`	`"question": question,`