Browse Source

第一版初始化

luojunhui 17 hours ago
parent
commit
1f01521dc9
2 changed files with 5 additions and 3 deletions
  1. 2 2
      applications/vector_database/field.py
  2. 3 1
      requirements.txt

+ 2 - 2
applications/vector_database/field.py

@@ -1,10 +1,10 @@
-from applications.config import MODEL_CONFIG
+from applications.config import LOCAL_MODEL_CONFIG
 
 from pymilvus import Collection, CollectionSchema, FieldSchema, DataType
 
 
 collections = {}
-for model_name, cfg in MODEL_CONFIG.items():
+for model_name, cfg in LOCAL_MODEL_CONFIG.items():
     col_name = model_name.replace("/", "_").replace("-", "_").lower()
     fields = [
         FieldSchema(name="id", dtype=DataType.INT64, is_primary=True, auto_id=True),

+ 3 - 1
requirements.txt

@@ -1,3 +1,5 @@
 hypercorn
 quart_cors
-quart
+quart
+aiohttp
+pymilvus