gen python api

Former-commit-id: 1c9b4a1fea7c450b55e8d683085d8779460ea3c4

cpp/src/thrift/gen-py/zilliz/VecService-remote

@@ -0,0 +1,187 @@
+#!/usr/bin/env python
+#
+# Autogenerated by Thrift Compiler (0.12.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+#  options string: py
+#
+
+import sys
+import pprint
+if sys.version_info[0] > 2:
+    from urllib.parse import urlparse
+else:
+    from urlparse import urlparse
+from thrift.transport import TTransport, TSocket, TSSLSocket, THttpClient
+from thrift.protocol.TBinaryProtocol import TBinaryProtocol
+
+from zilliz import VecService
+from zilliz.ttypes import *
+
+if len(sys.argv) <= 1 or sys.argv[1] == '--help':
+    print('')
+    print('Usage: ' + sys.argv[0] + ' [-h host[:port]] [-u url] [-f[ramed]] [-s[sl]] [-novalidate] [-ca_certs certs] [-keyfile keyfile] [-certfile certfile] function [arg1 [arg2...]]')
+    print('')
+    print('Functions:')
+    print('  void add_group(VecGroup group)')
+    print('  VecGroup get_group(string group_id)')
+    print('  void del_group(string group_id)')
+    print('  void add_vector(string group_id, VecTensor tensor)')
+    print('  void add_vector_batch(string group_id, VecTensorList tensor_list)')
+    print('  void add_binary_vector(string group_id, VecBinaryTensor tensor)')
+    print('  void add_binary_vector_batch(string group_id, VecBinaryTensorList tensor_list)')
+    print('  VecSearchResult search_vector(string group_id, i64 top_k, VecTensor tensor, VecSearchFilter filter)')
+    print('  VecSearchResultList search_vector_batch(string group_id, i64 top_k, VecTensorList tensor_list, VecSearchFilter filter)')
+    print('  VecSearchResult search_binary_vector(string group_id, i64 top_k, VecBinaryTensor tensor, VecSearchFilter filter)')
+    print('  VecSearchResultList search_binary_vector_batch(string group_id, i64 top_k, VecBinaryTensorList tensor_list, VecSearchFilter filter)')
+    print('')
+    sys.exit(0)
+
+pp = pprint.PrettyPrinter(indent=2)
+host = 'localhost'
+port = 9090
+uri = ''
+framed = False
+ssl = False
+validate = True
+ca_certs = None
+keyfile = None
+certfile = None
+http = False
+argi = 1
+
+if sys.argv[argi] == '-h':
+    parts = sys.argv[argi + 1].split(':')
+    host = parts[0]
+    if len(parts) > 1:
+        port = int(parts[1])
+    argi += 2
+
+if sys.argv[argi] == '-u':
+    url = urlparse(sys.argv[argi + 1])
+    parts = url[1].split(':')
+    host = parts[0]
+    if len(parts) > 1:
+        port = int(parts[1])
+    else:
+        port = 80
+    uri = url[2]
+    if url[4]:
+        uri += '?%s' % url[4]
+    http = True
+    argi += 2
+
+if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':
+    framed = True
+    argi += 1
+
+if sys.argv[argi] == '-s' or sys.argv[argi] == '-ssl':
+    ssl = True
+    argi += 1
+
+if sys.argv[argi] == '-novalidate':
+    validate = False
+    argi += 1
+
+if sys.argv[argi] == '-ca_certs':
+    ca_certs = sys.argv[argi+1]
+    argi += 2
+
+if sys.argv[argi] == '-keyfile':
+    keyfile = sys.argv[argi+1]
+    argi += 2
+
+if sys.argv[argi] == '-certfile':
+    certfile = sys.argv[argi+1]
+    argi += 2
+
+cmd = sys.argv[argi]
+args = sys.argv[argi + 1:]
+
+if http:
+    transport = THttpClient.THttpClient(host, port, uri)
+else:
+    if ssl:
+        socket = TSSLSocket.TSSLSocket(host, port, validate=validate, ca_certs=ca_certs, keyfile=keyfile, certfile=certfile)
+    else:
+        socket = TSocket.TSocket(host, port)
+    if framed:
+        transport = TTransport.TFramedTransport(socket)
+    else:
+        transport = TTransport.TBufferedTransport(socket)
+protocol = TBinaryProtocol(transport)
+client = VecService.Client(protocol)
+transport.open()
+
+if cmd == 'add_group':
+    if len(args) != 1:
+        print('add_group requires 1 args')
+        sys.exit(1)
+    pp.pprint(client.add_group(eval(args[0]),))
+
+elif cmd == 'get_group':
+    if len(args) != 1:
+        print('get_group requires 1 args')
+        sys.exit(1)
+    pp.pprint(client.get_group(args[0],))
+
+elif cmd == 'del_group':
+    if len(args) != 1:
+        print('del_group requires 1 args')
+        sys.exit(1)
+    pp.pprint(client.del_group(args[0],))
+
+elif cmd == 'add_vector':
+    if len(args) != 2:
+        print('add_vector requires 2 args')
+        sys.exit(1)
+    pp.pprint(client.add_vector(args[0], eval(args[1]),))
+
+elif cmd == 'add_vector_batch':
+    if len(args) != 2:
+        print('add_vector_batch requires 2 args')
+        sys.exit(1)
+    pp.pprint(client.add_vector_batch(args[0], eval(args[1]),))
+
+elif cmd == 'add_binary_vector':
+    if len(args) != 2:
+        print('add_binary_vector requires 2 args')
+        sys.exit(1)
+    pp.pprint(client.add_binary_vector(args[0], eval(args[1]),))
+
+elif cmd == 'add_binary_vector_batch':
+    if len(args) != 2:
+        print('add_binary_vector_batch requires 2 args')
+        sys.exit(1)
+    pp.pprint(client.add_binary_vector_batch(args[0], eval(args[1]),))
+
+elif cmd == 'search_vector':
+    if len(args) != 4:
+        print('search_vector requires 4 args')
+        sys.exit(1)
+    pp.pprint(client.search_vector(args[0], eval(args[1]), eval(args[2]), eval(args[3]),))
+
+elif cmd == 'search_vector_batch':
+    if len(args) != 4:
+        print('search_vector_batch requires 4 args')
+        sys.exit(1)
+    pp.pprint(client.search_vector_batch(args[0], eval(args[1]), eval(args[2]), eval(args[3]),))
+
+elif cmd == 'search_binary_vector':
+    if len(args) != 4:
+        print('search_binary_vector requires 4 args')
+        sys.exit(1)
+    pp.pprint(client.search_binary_vector(args[0], eval(args[1]), eval(args[2]), eval(args[3]),))
+
+elif cmd == 'search_binary_vector_batch':
+    if len(args) != 4:
+        print('search_binary_vector_batch requires 4 args')
+        sys.exit(1)
+    pp.pprint(client.search_binary_vector_batch(args[0], eval(args[1]), eval(args[2]), eval(args[3]),))
+
+else:
+    print('Unrecognized method %s' % cmd)
+    sys.exit(1)
+
+transport.close()

cpp/src/thrift/gen-py/zilliz/__init__.py

@@ -0,0 +1 @@
+__all__ = ['ttypes', 'constants', 'VecService']

cpp/src/thrift/gen-py/zilliz/constants.py

@@ -0,0 +1,14 @@
+#
+# Autogenerated by Thrift Compiler (0.12.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+#  options string: py
+#
+
+from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
+from thrift.protocol.TProtocol import TProtocolException
+from thrift.TRecursive import fix_spec
+
+import sys
+from .ttypes import *

cpp/src/thrift/py_gen.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+../../third_party/build/bin/thrift -r --gen py ./VectorService.thrift
+