def main(): try: opts, args = getopt.getopt( sys.argv[1:], "t:q:a:ls", ["table=", "question=", "answer=", "load", "sentence=", "search"], ) except: print("Usage: test.py -t <table> -l -s") sys.exit(2) table_name = config.DEFAULT_TABLE for opt_name, opt_value in opts: if opt_name in ("-t", "--table"): table_name = opt_value elif opt_name in ("-q", "--question"): question_dir = opt_value elif opt_name in ("-a", "--answer"): answer_dir = opt_value elif opt_name in ("-l", "--load"): milvus_bert.import_data(table_name, question_dir, answer_dir) elif opt_name in("--sentence"): query_sentence = opt_value print(query_sentence) elif opt_name in ("-s","--search"): print("begin search") out_put = milvus_bert.search_in_milvus(table_name, query_sentence) print(out_put)
def do_search_api(): args = reqparse.RequestParser(). \ add_argument("Table", type=str). \ add_argument("query_text", type=str). \ parse_args() table_name = args['Table'] if not table_name: table_name = config.DEFAULT_TABLE query_sentence = args['query_text'] if not query_sentence: return "no text" if query_sentence: try: output = search_in_milvus(table_name, query_sentence) if output: return output else: return "未收录该问题" except Exception as e: return "Error with {}".format(e) return "not found", 400