Move raw_generated_gql just before verified_gql

jordimas · jordimas · commit fc2e72ff77e4 · 2025-05-08T07:32:09.000+02:00
diff --git a/src/langchain_google_spanner/graph_qa.py b/src/langchain_google_spanner/graph_qa.py
@@ -301,7 +301,6 @@ 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,
@@ -310,6 +309,7 @@ def _call(
                 }
             )
             if "verified_gql" in verify_response:
+                intermediate_steps.append({"raw_generated_gql": generated_gql})
                 verified_gql = fix_gql_syntax(verify_response["verified_gql"])
                 intermediate_steps.append({"verified_gql": verified_gql})
             else:

Original file line number	Diff line number	Diff line change
`@@ -301,7 +301,6 @@ 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})`
`305`	`304`	`verify_response = self.gql_verify_chain.invoke(`
`306`	`305`	`{`
`307`	`306`	`"question": question,`
`@@ -310,6 +309,7 @@ def _call(`
`310`	`309`	`}`
`311`	`310`	`)`
`312`	`311`	`if "verified_gql" in verify_response:`
	`312`	`+ intermediate_steps.append({"raw_generated_gql": generated_gql})`
`313`	`313`	`verified_gql = fix_gql_syntax(verify_response["verified_gql"])`
`314`	`314`	`intermediate_steps.append({"verified_gql": verified_gql})`
`315`	`315`	`else:`