diff --git a/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs b/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs index 760ba86..99932c3 100644 --- a/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs +++ b/src/Senparc.AI.Kernel/KernelConfigExtensions/KernelConfigExtensions.cs @@ -243,9 +243,9 @@ public static VectorStoreCollection GetVectorCollection collection = null; + var kernel = iWwantToRun.Kernel; //TODO: If the logic becomes overly complex in the future, different combinations can be considered to be separated into different libraries @@ -253,7 +253,8 @@ public static VectorStoreCollection GetVectorCollection(); collection = vectorStore.GetCollection(name, vectorStoreRecordDefinition); break; } @@ -263,12 +264,9 @@ public static VectorStoreCollection GetVectorCollection(name, vectorStoreRecordDefinition); - } + // Retrieve from DI container (registered via ConfigVectorStore) + vectorStore = kernel.Services.GetRequiredService(); + collection = vectorStore.GetCollection(name, vectorStoreRecordDefinition); break; } case VectorDBType.Milvus: @@ -293,12 +291,14 @@ public static VectorStoreCollection GetVectorCollection(); collection = vectorStore.GetCollection(name, vectorStoreRecordDefinition); break; } default: - vectorStore = new InMemoryVectorStore(); + // For unsupported types, try to get from DI or throw an error + vectorStore = kernel.Services.GetRequiredService(); collection = vectorStore.GetCollection(name, vectorStoreRecordDefinition); break; }