Преглед на файлове

Fix/grpc gevent compatible (#1314)

Co-authored-by: jyong <jyong@dify.ai>
Jyong преди 1 година
родител
ревизия
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