diff --git a/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs b/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs index 760ba86..082ea95 100644 --- a/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs +++ b/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs @@ -244,7 +244,6 @@ public static VectorStoreCollection GetVectorCollection collection = null; //TODO: If the logic becomes overly complex in the future, different combinations can be considered to be separated into different libraries @@ -253,8 +252,10 @@ public static VectorStoreCollection GetVectorCollection(name, vectorStoreRecordDefinition); + using (var inMemoryVectorStore = new InMemoryVectorStore()) + { + collection = inMemoryVectorStore.GetCollection(name, vectorStoreRecordDefinition); + } break; } case VectorDBType.HardDisk: @@ -293,13 +294,17 @@ public static VectorStoreCollection GetVectorCollection(name, vectorStoreRecordDefinition); + using (var qdrantVectorStore = new QdrantVectorStore(new QdrantClient(vectorDb.ConnectionString), ownsClient: true)) + { + collection = qdrantVectorStore.GetCollection(name, vectorStoreRecordDefinition); + } break; } default: - vectorStore = new InMemoryVectorStore(); - collection = vectorStore.GetCollection(name, vectorStoreRecordDefinition); + using (var inMemoryVectorStore = new InMemoryVectorStore()) + { + collection = inMemoryVectorStore.GetCollection(name, vectorStoreRecordDefinition); + } break; }