wanjia9506
/
langchain4j
mirror of https://github.com/langchain4j/langchain4j.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use Testcontainers for Weaviate IT (#332) 

Run `semitechnologies/weaviate` image using `GenericContainer` and
allow rapid feeback when using Weaviate. Relax `apiKey` constraint
to allow empty value for testing purposes.
This commit is contained in:
Eddú Meléndez Gonzales 2023-12-08 03:33:56 -06:00 committed by GitHub
parent b2f358c926
commit a886eee15a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 25 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain4j-weaviate/pom.xml
View File

@ -70,6 +70,12 @@
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.testcontainers</groupId>
<artifactId>junit-jupiter</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
<build>

3
langchain4j-weaviate/src/main/java/dev/langchain4j/store/embedding/weaviate/WeaviateEmbeddingStore.java
View File

@ -2,6 +2,7 @@ package dev.langchain4j.store.embedding.weaviate;
import static dev.langchain4j.internal.Utils.*;
import static dev.langchain4j.internal.ValidationUtils.ensureNotBlank;
import static dev.langchain4j.internal.ValidationUtils.ensureNotNull;
import static io.weaviate.client.v1.data.replication.model.ConsistencyLevel.QUORUM;
import static java.util.Arrays.stream;
import static java.util.Collections.emptyList;
@ -66,7 +67,7 @@ public class WeaviateEmbeddingStore implements EmbeddingStore<TextSegment> {
) {
try {
Config config = new Config(ensureNotBlank(scheme, "scheme"), ensureNotBlank(host, "host"));
this.client = WeaviateAuthClient.apiKey(config, ensureNotBlank(apiKey, "apiKey"));
this.client = WeaviateAuthClient.apiKey(config, ensureNotNull(apiKey, "apiKey"));
} catch (AuthException e) {
throw new IllegalArgumentException(e);
}

22
langchain4j-weaviate/src/test/java/dev/langchain4j/store/embedding/weaviate/WeaviateEmbeddingStoreIT.java
View File

@ -5,17 +5,29 @@ import dev.langchain4j.model.embedding.AllMiniLmL6V2QuantizedEmbeddingModel;
import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.EmbeddingStoreWithoutMetadataIT;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.testcontainers.containers.GenericContainer;
import org.testcontainers.containers.wait.strategy.Wait;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;
import static dev.langchain4j.internal.Utils.randomUUID;
@EnabledIfEnvironmentVariable(named = "WEAVIATE_API_KEY", matches = ".+")
@Testcontainers
class WeaviateEmbeddingStoreIT extends EmbeddingStoreWithoutMetadataIT {
@Container
static GenericContainer<?> weaviate = new GenericContainer<>("semitechnologies/weaviate:1.22.4")
.withEnv("AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED", "true")
.withEnv("PERSISTENCE_DATA_PATH", "/var/lib/weaviate")
.withEnv("QUERY_DEFAULTS_LIMIT", "25")
.withEnv("DEFAULT_VECTORIZER_MODULE", "none")
.withEnv("CLUSTER_HOSTNAME", "node1")
.withExposedPorts(8080);
EmbeddingStore<TextSegment> embeddingStore = WeaviateEmbeddingStore.builder()
.apiKey(System.getenv("WEAVIATE_API_KEY"))
.scheme("https")
.host("test-am8ocede.weaviate.network")
.apiKey("")
.scheme("http")
.host(String.format("%s:%d", weaviate.getHost(), weaviate.getMappedPort(8080)))
.objectClass("Test" + randomUUID().replace("-", ""))
.build();