Skip to content

Commit

Permalink
Refactor Weaviate -> VectorStore
Browse files Browse the repository at this point in the history
  • Loading branch information
youngmoneee committed Aug 26, 2024
1 parent 67fffba commit ff862bf
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
8 changes: 4 additions & 4 deletions src/main/java/io/ymon/rag/config/ETLConfig.java
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import io.ymon.rag.document.FileDocumentWriter;
import io.ymon.rag.document.StackOverflowDocumentReader;
import io.ymon.rag.document.Transformer;
import io.ymon.rag.document.WeaviateDocumentWriter;
import io.ymon.rag.document.VectorStoreDocumentWriter;
import io.ymon.rag.annotation.EntryPoint;
import java.io.BufferedWriter;
import java.io.FileWriter;
Expand All @@ -25,7 +25,7 @@
import org.slf4j.LoggerFactory;
import org.springframework.ai.document.Document;
import org.springframework.ai.embedding.EmbeddingModel;
import org.springframework.ai.vectorstore.WeaviateVectorStore;
import org.springframework.ai.vectorstore.VectorStore;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
Expand Down Expand Up @@ -116,8 +116,8 @@ DocumentTransformer transformation(List<DocumentTransformer> transformers) {

//@Bean
//@Primary
DocumentWriter weaviateDocumentWriter(WeaviateVectorStore vectorStore) {
return new WeaviateDocumentWriter(vectorStore);
DocumentWriter vectorstoreDocumentWriter(VectorStore vectorStore) {
return new VectorStoreDocumentWriter(vectorStore);
}

//@Bean
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,16 @@
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.document.Document;
import org.springframework.ai.vectorstore.VectorStore;
import org.springframework.ai.vectorstore.WeaviateVectorStore;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.core.scheduler.Schedulers;

@Slf4j
public class WeaviateDocumentWriter implements DocumentWriter { // Wrapper
public class VectorStoreDocumentWriter implements DocumentWriter { // Wrapper

private final VectorStore vectorStore;

public WeaviateDocumentWriter(WeaviateVectorStore vectorStore) {
public VectorStoreDocumentWriter(VectorStore vectorStore) {
this.vectorStore = vectorStore;
}

Expand Down

0 comments on commit ff862bf

Please sign in to comment.