[metadata] Use before and after cursor parameters

cloudera · Feb 10, 2021 · e6ad166 · e6ad166
1 parent 01d1615
commit e6ad166
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/desktop/libs/metadata/src/metadata/optimizer/optimizer_client.py b/desktop/libs/metadata/src/metadata/optimizer/optimizer_client.py
@@ -261,13 +261,13 @@ def query_risk(self, query, source_platform, db_name, page_size=100, startingTok
     }
 
 
-  def predict(self, query, source_platform, connector):
+  def predict(self, before_cursor, after_cursor, connector):
     response = self._call(
       'predict', {
         'tenant': self._tenant_id,
         'connector': connector,
-        'query': _clean_query(query),
-        'sourcePlatform': source_platform,
+        'before_cursor': before_cursor,
+        'after_cursor': after_cursor,
       }
     )
 

diff --git a/desktop/libs/metadata/src/metadata/optimizer_api.py b/desktop/libs/metadata/src/metadata/optimizer_api.py
@@ -216,12 +216,12 @@ def predict(request):
 
   interface = request.POST.get('interface', OPTIMIZER.INTERFACE.get())
   connector = json.loads(request.POST.get('connector', '{}'))
-  query = json.loads(request.POST.get('query', '""'))
-  source_platform = request.POST.get('sourcePlatform')
+  before_cursor = request.POST.get('beforeCursor', '')
+  after_cursor = request.POST.get('afterCursor', '')
 
   api = get_api(request.user, interface)
 
-  data = api.predict(query=query, source_platform=source_platform, connector=connector)
+  data = api.predict(before_cursor=before_cursor, after_cursor=after_cursor, connector=connector)
 
   if data:
     response['status'] = 0