mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-06 17:18:35 +08:00
74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
from flask import Flask, jsonify, request
|
|
from flask_restful import Resource, Api
|
|
from engine import app, db
|
|
from engine.model.GroupTable import GroupTable
|
|
from engine.controller.VectorEngine import VectorEngine
|
|
|
|
# app = Flask(__name__)
|
|
api = Api(app)
|
|
|
|
|
|
from flask_restful import reqparse
|
|
from flask_restful import request
|
|
class Vector(Resource):
|
|
def __init__(self):
|
|
self.__parser = reqparse.RequestParser()
|
|
self.__parser.add_argument('vector', type=float, action='append', location=['json'])
|
|
|
|
def post(self, group_id):
|
|
args = self.__parser.parse_args()
|
|
vector = args['vector']
|
|
return VectorEngine.AddVector(group_id, vector)
|
|
|
|
|
|
class VectorSearch(Resource):
|
|
def __init__(self):
|
|
self.__parser = reqparse.RequestParser()
|
|
self.__parser.add_argument('vector', type=float, action='append', location=['json'])
|
|
|
|
def post(self, group_id):
|
|
args = self.__parser.parse_args()
|
|
print('vector: ', args['vector'])
|
|
# go to search every thing
|
|
return "vectorSearch post"
|
|
|
|
|
|
class Index(Resource):
|
|
def __init__(self):
|
|
self.__parser = reqparse.RequestParser()
|
|
# self.__parser.add_argument('group_id', type=str)
|
|
|
|
def post(self, group_id):
|
|
return VectorEngine.CreateIndex(group_id)
|
|
|
|
|
|
class Group(Resource):
|
|
def __init__(self):
|
|
self.__parser = reqparse.RequestParser()
|
|
self.__parser.add_argument('group_id', type=str)
|
|
|
|
def post(self, group_id):
|
|
return VectorEngine.AddGroup(group_id)
|
|
|
|
def get(self, group_id):
|
|
return VectorEngine.GetGroup(group_id)
|
|
|
|
def delete(self, group_id):
|
|
return VectorEngine.DeleteGroup(group_id)
|
|
|
|
|
|
class GroupList(Resource):
|
|
def get(self):
|
|
return VectorEngine.GetGroupList()
|
|
|
|
|
|
api.add_resource(Vector, '/vector/add/<group_id>')
|
|
api.add_resource(Group, '/vector/group/<group_id>')
|
|
api.add_resource(GroupList, '/vector/group')
|
|
api.add_resource(Index, '/vector/index/<group_id>')
|
|
api.add_resource(VectorSearch, '/vector/search/<group_id>')
|
|
|
|
|
|
# if __name__ == '__main__':
|
|
# app.run()
|