Просмотр исходного кода

Fix/grpc gevent compatible (#1314)

Co-authored-by: jyong <jyong@dify.ai>
Jyong 1 год назад
Родитель
Сommit
7f6ad9653e
2 измененных файлов с 3 добавлено и 2 удалено
  1. 3 0
      api/app.py
  2. 0 2
      api/core/index/vector_index/milvus.py

+ 3 - 0
api/app.py

@@ -6,6 +6,9 @@ from werkzeug.exceptions import Unauthorized
 if not os.environ.get("DEBUG") or os.environ.get("DEBUG").lower() != 'true':
     from gevent import monkey
     monkey.patch_all()
+    if os.environ.get("VECTOR_STORE") == 'milvus':
+        import grpc.experimental.gevent
+        grpc.experimental.gevent.init_gevent()
 
 import logging
 import json

+ 0 - 2
api/core/index/vector_index/milvus.py

@@ -167,8 +167,6 @@ class Milvus(VectorStore):
         self._init()
 
     @property
-
-
     def embeddings(self) -> Embeddings:
         return self.embedding_func