Skip to content

Commit

Permalink
feat(RAGConfiguration): 테스트용 ChromaVectorStore를 prod 환경에서도 bean으로 등록하…
Browse files Browse the repository at this point in the history
…도록 변경
  • Loading branch information
zbqmgldjfh committed Jul 22, 2024
1 parent 645cb77 commit a508dd5
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -30,14 +30,14 @@ public class RAGConfiguration {
@Value("vectorstore.json")
private String vectorStoreName;

@Profile("test")
@Profile("dev | test")
@Bean
public ChromaApi chromaApi(RestTemplate restTemplate) {
String chromaUrl = "http://127.0.0.1:8000";
return new ChromaApi(chromaUrl, restTemplate);
}

@Profile("test")
@Profile("dev | test")
@Bean
public ChromaVectorStore chromaVectorStore(EmbeddingModel embeddingModel, ChromaApi chromaApi) {
return new ChromaVectorStore(embeddingModel, chromaApi, false);
Expand Down

0 comments on commit a508dd5

Please sign in to comment.