Skip to content

Commit 7c47812

Browse files
author
Milder Hernandez Cagua
committed
Fix JDBCVectorStoreTest
1 parent 5306263 commit 7c47812

File tree

1 file changed

+10
-0
lines changed
  • api-test/integration-tests/src/test/java/com/microsoft/semantickernel/tests/connectors/memory/jdbc

1 file changed

+10
-0
lines changed

api-test/integration-tests/src/test/java/com/microsoft/semantickernel/tests/connectors/memory/jdbc/JDBCVectorStoreTest.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,12 @@
66
import com.microsoft.semantickernel.connectors.data.jdbc.JDBCVectorStoreRecordCollectionOptions;
77
import com.microsoft.semantickernel.connectors.data.mysql.MySQLVectorStoreQueryProvider;
88
import com.microsoft.semantickernel.connectors.data.postgres.PostgreSQLVectorStoreQueryProvider;
9+
import com.microsoft.semantickernel.connectors.data.sqlite.SQLiteVectorStoreQueryProvider;
910
import com.mysql.cj.jdbc.MysqlDataSource;
1011
import org.junit.jupiter.params.ParameterizedTest;
1112
import org.junit.jupiter.params.provider.EnumSource;
1213
import org.postgresql.ds.PGSimpleDataSource;
14+
import org.sqlite.SQLiteDataSource;
1315
import org.testcontainers.containers.MySQLContainer;
1416
import org.testcontainers.containers.PostgreSQLContainer;
1517
import org.testcontainers.junit.jupiter.Container;
@@ -59,6 +61,14 @@ private JDBCVectorStore buildVectorStore(QueryProvider provider) {
5961
.withDataSource(dataSource)
6062
.build();
6163
break;
64+
case SQLite:
65+
SQLiteDataSource sqliteDataSource = new SQLiteDataSource();
66+
sqliteDataSource.setUrl("jdbc:sqlite:file:test");
67+
dataSource = sqliteDataSource;
68+
queryProvider = SQLiteVectorStoreQueryProvider.builder()
69+
.withDataSource(sqliteDataSource)
70+
.build();
71+
break;
6272
default:
6373
throw new IllegalArgumentException("Unknown query provider: " + provider);
6474
}

0 commit comments

Comments
 (0)