第一版初始化(black)

applications/api/__init__.py

@@ -1,5 +1,3 @@
 from .embedding import get_basic_embedding
 
-__all__ = [
-    "get_basic_embedding"
-]
+__all__ = ["get_basic_embedding"]

applications/api/embedding.py

@@ -16,9 +16,7 @@ async def get_basic_embedding(text: str, model: str):
             json={"input": text, "model": model_path},
             headers={"Content-Type": "application/json"},
         )
-        return response['data'][0]["embedding"]
+        return response["data"][0]["embedding"]
 
 
-__all__ = [
-    "get_basic_embedding"
-]
+__all__ = ["get_basic_embedding"]

applications/config/__init__.py

@@ -1,7 +1,3 @@
-from .model_config import  DEFAULT_MODEL, LOCAL_MODEL_CONFIG, VLLM_SERVER_URL
+from .model_config import DEFAULT_MODEL, LOCAL_MODEL_CONFIG, VLLM_SERVER_URL
 
-__all__ = [
-    "DEFAULT_MODEL",
-    "LOCAL_MODEL_CONFIG",
-    "VLLM_SERVER_URL"
-]
+__all__ = ["DEFAULT_MODEL", "LOCAL_MODEL_CONFIG", "VLLM_SERVER_URL"]

applications/config/model_config.py

@@ -6,4 +6,4 @@ LOCAL_MODEL_CONFIG = {
 
 DEFAULT_MODEL = "Qwen3-Embedding-4B"
 
-VLLM_SERVER_URL = "http://vllm-qwen:8000/v1/embeddings"
+VLLM_SERVER_URL = "http://vllm-qwen:8000/v1/embeddings"

applications/utils/__init__.py

@@ -1 +1 @@
-from .async_http_client import AsyncHttpClient
+from .async_http_client import AsyncHttpClient

applications/utils/response.py

@@ -2,4 +2,4 @@ class BaseResponse:
 
     @staticmethod
     def negative_response():
-        pass
+        pass

applications/vector_database/field.py

@@ -9,10 +9,9 @@ for model_name, cfg in LOCAL_MODEL_CONFIG.items():
     fields = [
         FieldSchema(name="id", dtype=DataType.INT64, is_primary=True, auto_id=True),
         FieldSchema(name="text", dtype=DataType.VARCHAR, max_length=1024),
-        FieldSchema(name="embedding", dtype=DataType.FLOAT_VECTOR, dim=cfg["dim"])
+        FieldSchema(name="embedding", dtype=DataType.FLOAT_VECTOR, dim=cfg["dim"]),
     ]
     schema = CollectionSchema(fields, description=f"{model_name} embeddings")
     collection = Collection(col_name, schema=schema)
     collection.load()
     collections[model_name] = collection
-

routes/__init__.py

@@ -1,5 +1,3 @@
 from .buleprint import server_routes
 
-__all__ = [
-    "server_routes"
-]
+__all__ = ["server_routes"]

routes/buleprint.py

@@ -4,28 +4,24 @@ from applications.config import DEFAULT_MODEL, LOCAL_MODEL_CONFIG
 from applications.api import get_basic_embedding
 
 
-server_bp = Blueprint('api', __name__, url_prefix='/api')
+server_bp = Blueprint("api", __name__, url_prefix="/api")
+
 
 def server_routes(vector_db):
 
-    @server_bp.route('/embed', methods=['POST'])
+    @server_bp.route("/embed", methods=["POST"])
     async def embed():
         body = await request.get_json()
-        text = body.get('text')
-        model_name = body.get('model', DEFAULT_MODEL)
+        text = body.get("text")
+        model_name = body.get("model", DEFAULT_MODEL)
         if not LOCAL_MODEL_CONFIG.get(model_name):
-            return jsonify(
-                {"error": "error  model"}
-            )
+            return jsonify({"error": "error  model"})
 
         embedding = await get_basic_embedding(text, model_name)
-        return jsonify({
-            "embedding": embedding
-        })
+        return jsonify({"embedding": embedding})
 
-    @server_bp.route('/search', methods=['POST'])
+    @server_bp.route("/search", methods=["POST"])
     async def search():
         pass
 
-
-    return server_bp
+    return server_bp

vector_app.py

@@ -1,5 +1,6 @@
 from quart import Quart
 from quart_cors import cors
+
 # from pymilvus import connections
 
 from applications.config import LOCAL_MODEL_CONFIG, DEFAULT_MODEL
@@ -18,4 +19,3 @@ connections = None
 # 注册路由
 app_route = server_routes(connections)
 app.register_blueprint(app_route)
-