fix(updateCustomQuery): update json payload

tigergraph · Apr 25, 2024 · f85ce2d · f85ce2d
1 parent 779a642
commit f85ce2d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pyTigerGraph/ai/ai.py b/pyTigerGraph/ai/ai.py
@@ -79,7 +79,8 @@ def registerCustomQuery(self, query_name: str, description: str = None, docstrin
                 "function_header": query_name,
                 "description": description,
                 "docstring": docstring,
-                "param_types": param_types
+                "param_types": param_types,
+                "graphname": self.conn.graphname
             }
             url = self.nlqs_host+"/"+self.conn.graphname+"/register_docs"
             return self.conn._req("POST", url, authMode="pwd", data = data, jsonData=True, resKey=None)
@@ -123,10 +124,11 @@ def updateCustomQuery(self, query_name: str, description: str = None, docstring:
                 "function_header": query_name,
                 "description": description,
                 "docstring": docstring,
-                "param_types": param_types
+                "param_types": param_types,
+                "graphname": self.conn.graphname
             }
 
-            json_payload = {"query_info": data}
+            json_payload = {"id": "", "query_info": data}
             url = self.nlqs_host+"/"+self.conn.graphname+"/upsert_docs"
             return self.conn._req("POST", url, authMode="pwd", data = json_payload, jsonData=True, resKey=None)