Unverified Commit 7f6ad965 authored by Jyong's avatar Jyong Committed by GitHub

Fix/grpc gevent compatible (#1314)

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 2851a9f0
......@@ -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
......
......@@ -167,8 +167,6 @@ class Milvus(VectorStore):
self._init()
@property
def embeddings(self) -> Embeddings:
return self.embedding_func
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment