fix(graph): Fix Issue 156 (#167)

amullick-git · web-flow · commit 9940b86bb2c8 · 2025-04-15T14:34:37.000-07:00
* Check if verified_gql is generated

* Fixed typo

---------

Co-authored-by: Amarnath Mullick &lt;amullick@google.com&gt;
diff --git a/src/langchain_google_spanner/graph_qa.py b/src/langchain_google_spanner/graph_qa.py
@@ -308,8 +308,11 @@ def _call(
                     "graph_schema": self.graph.get_schema,
                 }
             )
-            verified_gql = fix_gql_syntax(verify_response["verified_gql"])
-            intermediate_steps.append({"verified_gql": verified_gql})
+            if "verified_gql" in verify_response:
+                verified_gql = fix_gql_syntax(verify_response["verified_gql"])
+                intermediate_steps.append({"verified_gql": verified_gql})
+            else:
+                verified_gql = generated_gql
         else:
             verified_gql = generated_gql
 

Original file line number	Diff line number	Diff line change
`@@ -308,8 +308,11 @@ def _call(`
`308`	`308`	`"graph_schema": self.graph.get_schema,`
`309`	`309`	`}`
`310`	`310`	`)`
`311`		`- verified_gql = fix_gql_syntax(verify_response["verified_gql"])`
`312`		`- intermediate_steps.append({"verified_gql": verified_gql})`
	`311`	`+ if "verified_gql" in verify_response:`
	`312`	`+ verified_gql = fix_gql_syntax(verify_response["verified_gql"])`
	`313`	`+ intermediate_steps.append({"verified_gql": verified_gql})`
	`314`	`+ else:`
	`315`	`+ verified_gql = generated_gql`
`313`	`316`	`else:`
`314`	`317`	`verified_gql = generated_gql`
`315`	`318`