瀏覽代碼

[metadata] Use before and after cursor parameters

Romain Rigaux 4 年之前
父節點
當前提交
e6ad166212

+ 3 - 3
desktop/libs/metadata/src/metadata/optimizer/optimizer_client.py

@@ -261,13 +261,13 @@ class OptimizerClient(object):
     }
 
 
-  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,
       }
     )
 

+ 3 - 3
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