(name, **kwargs)
| 37 | |
| 38 | @staticmethod |
| 39 | def get(name, **kwargs): |
| 40 | top_k = kwargs.get("top_k", TOP_K) |
| 41 | if name == "milvus": |
| 42 | from modelcache.manager.vector_data.milvus import Milvus |
| 43 | milvus_config = kwargs.get("milvus_config") |
| 44 | dimension = kwargs.get("dimension", DIMENSION) |
| 45 | VectorBase.check_dimension(dimension) |
| 46 | host = milvus_config.get('milvus', 'host') |
| 47 | port = milvus_config.get('milvus', 'port') |
| 48 | user = milvus_config.get('milvus', 'user') |
| 49 | password = milvus_config.get('milvus', 'password') |
| 50 | |
| 51 | secure = kwargs.get("secure", MILVUS_SECURE) |
| 52 | collection_name = kwargs.get("collection_name", COLLECTION_NAME) |
| 53 | index_params = kwargs.get("index_params", MILVUS_INDEX_PARAMS) |
| 54 | search_params = kwargs.get("search_params", None) |
| 55 | local_mode = kwargs.get("local_mode", False) |
| 56 | local_data = kwargs.get("local_data", "./milvus_data") |
| 57 | vector_base = Milvus( |
| 58 | host=host, |
| 59 | port=port, |
| 60 | user=user, |
| 61 | password=password, |
| 62 | secure=secure, |
| 63 | collection_name=collection_name, |
| 64 | dimension=dimension, |
| 65 | top_k=top_k, |
| 66 | index_params=index_params, |
| 67 | search_params=search_params, |
| 68 | local_mode=local_mode, |
| 69 | local_data=local_data |
| 70 | ) |
| 71 | elif name == "redis": |
| 72 | from modelcache_mm.manager.vector_data.redis import RedisVectorStore |
| 73 | redis_config = kwargs.get("redis_config") |
| 74 | |
| 75 | mm_dimension = kwargs.get("mm_dimension", DIMENSION) |
| 76 | print('mm_dimension: {}'.format(mm_dimension)) |
| 77 | i_dimension = kwargs.get("i_dimension", DIMENSION) |
| 78 | t_dimension = kwargs.get("t_dimension", DIMENSION) |
| 79 | VectorBase.check_dimension(mm_dimension) |
| 80 | VectorBase.check_dimension(i_dimension) |
| 81 | VectorBase.check_dimension(t_dimension) |
| 82 | |
| 83 | host = redis_config.get('redis', 'host') |
| 84 | port = redis_config.get('redis', 'port') |
| 85 | user = redis_config.get('redis', 'user') |
| 86 | password = redis_config.get('redis', 'password') |
| 87 | namespace = kwargs.get("namespace", "") |
| 88 | # collection_name = kwargs.get("collection_name", COLLECTION_NAME) |
| 89 | |
| 90 | vector_base = RedisVectorStore( |
| 91 | host=host, |
| 92 | port=port, |
| 93 | username=user, |
| 94 | password=password, |
| 95 | namespace=namespace, |
| 96 | top_k=top_k, |
no test coverage detected