Merge pull request #214 from tigergraph/GML-1602-fix-the-register-end…

…point-inconsistence GML-1602 fix the endpoint to match copilot
tigergraph · Apr 5, 2024 · 9fca80b · 9fca80b
2 parents 061e105 + 8bcbfe8
commit 9fca80b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyTigerGraph/ai/ai.py b/pyTigerGraph/ai/ai.py
@@ -41,7 +41,7 @@ def registerCustomQuery(self, function_header: str, description: str, docstring:
             "docstring": docstring,
             "param_types": param_types
         }
-        url = self.nlqs_host+"/"+self.conn.graphname+"/registercustomquery"
+        url = self.nlqs_host+"/"+self.conn.graphname+"/register_docs"
         return self.conn._req("POST", url, authMode="pwd", data = data, jsonData=True, resKey=None)
 
     def retrieveDocs(self, query:str, top_k:int = 3):
@@ -58,7 +58,7 @@ def retrieveDocs(self, query:str, top_k:int = 3):
             "query": query
         }
 
-        url = self.nlqs_host+"/"+self.conn.graphname+"/retrievedocs?top_k="+str(top_k)
+        url = self.nlqs_host+"/"+self.conn.graphname+"/retrieve_docs?top_k="+str(top_k)
         return self.conn._req("POST", url, authMode="pwd", data = data, jsonData=True, resKey=None, skipCheck=True) 
 
     def query(self, query):