|
@@ -1,12 +1,12 @@
|
|
from quart import Blueprint, jsonify, request
|
|
from quart import Blueprint, jsonify, request
|
|
|
|
|
|
from applications.config import DEFAULT_MODEL, MODEL_CONFIG
|
|
from applications.config import DEFAULT_MODEL, MODEL_CONFIG
|
|
-from applications.embedding import get_basic_embedding
|
|
|
|
|
|
+from applications.embedding import get_basic_embedding, get_local_embedding
|
|
|
|
|
|
|
|
|
|
server_bp = Blueprint('api', __name__, url_prefix='/api')
|
|
server_bp = Blueprint('api', __name__, url_prefix='/api')
|
|
|
|
|
|
-def server_routes(vector_db):
|
|
|
|
|
|
+def server_routes(llm, vector_db):
|
|
|
|
|
|
@server_bp.route('/embed', methods=['POST'])
|
|
@server_bp.route('/embed', methods=['POST'])
|
|
async def embed():
|
|
async def embed():
|
|
@@ -23,6 +23,21 @@ def server_routes(vector_db):
|
|
"embedding": embedding
|
|
"embedding": embedding
|
|
})
|
|
})
|
|
|
|
|
|
|
|
+ @server_bp.route('/embed_v1', methods=['POST'])
|
|
|
|
+ async def embed_v1():
|
|
|
|
+ body = await request.get_json()
|
|
|
|
+ text = body.get('text')
|
|
|
|
+ model_name = body.get('model', DEFAULT_MODEL)
|
|
|
|
+ if not MODEL_CONFIG.get(model_name):
|
|
|
|
+ return jsonify(
|
|
|
|
+ {"error": "error model"}
|
|
|
|
+ )
|
|
|
|
+
|
|
|
|
+ embedding = await get_local_embedding(text, llm)
|
|
|
|
+ return jsonify({
|
|
|
|
+ "embedding": embedding
|
|
|
|
+ })
|
|
|
|
+
|
|
|
|
|
|
@server_bp.route('/search', methods=['POST'])
|
|
@server_bp.route('/search', methods=['POST'])
|
|
async def search():
|
|
async def search():
|