ruslanmv commited on
Commit
f496b94
·
1 Parent(s): 623fdec

Update milvus_singleton.py

Browse files
Files changed (1) hide show
  1. milvus_singleton.py +8 -2
milvus_singleton.py CHANGED
@@ -2,13 +2,19 @@ from pymilvus import connections
2
  from pymilvus.exceptions import ConnectionConfigException
3
 
4
  class MilvusClientSingleton:
5
- # ... (rest of the class code)
 
 
 
 
 
 
6
 
7
  def __init__(self, uri):
8
  if MilvusClientSingleton._instance is not None:
9
  raise Exception("This class is a singleton!")
10
  try:
11
- # Use connections.connect()
12
  connections.connect(uri=uri)
13
  self._instance = connections # Store the connections object
14
  print(f"Successfully connected to Milvus at {uri}")
 
2
  from pymilvus.exceptions import ConnectionConfigException
3
 
4
  class MilvusClientSingleton:
5
+ _instance = None
6
+
7
+ @staticmethod
8
+ def get_instance(uri):
9
+ if MilvusClientSingleton._instance is None:
10
+ MilvusClientSingleton(uri)
11
+ return MilvusClientSingleton._instance
12
 
13
  def __init__(self, uri):
14
  if MilvusClientSingleton._instance is not None:
15
  raise Exception("This class is a singleton!")
16
  try:
17
+ # Use connections.connect() to establish the connection
18
  connections.connect(uri=uri)
19
  self._instance = connections # Store the connections object
20
  print(f"Successfully connected to Milvus at {uri}")