Skip to content

Commit

Permalink
Fix image version
Browse files Browse the repository at this point in the history
  • Loading branch information
milderhc committed Sep 19, 2024
1 parent 384c7ac commit 09e0388
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 14 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
@TestMethodOrder(MethodOrderer.OrderAnnotation.class)
public class RedisHashSetVectorStoreRecordCollectionTest {

@Container private static final RedisContainer redisContainer = new RedisContainer("redis/redis-stack:latest");
@Container private static final RedisContainer redisContainer = new RedisContainer("redis/redis-stack:7.4.0-v0");

private static final Map<RecordCollectionOptions, RedisHashSetVectorStoreRecordCollectionOptions<Hotel>> optionsMap = new HashMap<>();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
@TestMethodOrder(MethodOrderer.OrderAnnotation.class)
public class RedisJsonVectorStoreRecordCollectionTest {

@Container private static final RedisContainer redisContainer = new RedisContainer("redis/redis-stack:latest");
@Container private static final RedisContainer redisContainer = new RedisContainer("redis/redis-stack:7.4.0-v0");

private static final Map<RecordCollectionOptions, RedisJsonVectorStoreRecordCollectionOptions<Hotel>> optionsMap = new HashMap<>();

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -118,17 +118,19 @@ public static void dataStorageWithAzureAISearch(
OpenAITextEmbeddingGenerationService embeddingGeneration) {

// Create a new Azure AI Search vector store
var azureAISearchVectorStore = com.microsoft.semantickernel.connectors.data.azureaisearch.AzureAISearchVectorStore.builder()
var azureAISearchVectorStore = com.microsoft.semantickernel.connectors.data.azureaisearch.AzureAISearchVectorStore
.builder()
.withSearchIndexAsyncClient(searchClient)
.withOptions(new AzureAISearchVectorStoreOptions())
.build();

String collectionName = "skgithubfiles";
var collection = (AzureAISearchVectorStoreRecordCollection<GitHubFile>) azureAISearchVectorStore.getCollection(
collectionName,
AzureAISearchVectorStoreRecordCollectionOptions.<GitHubFile>builder()
.withRecordClass(GitHubFile.class)
.build());
var collection = (AzureAISearchVectorStoreRecordCollection<GitHubFile>) azureAISearchVectorStore
.getCollection(
collectionName,
AzureAISearchVectorStoreRecordCollectionOptions.<GitHubFile>builder()
.withRecordClass(GitHubFile.class)
.build());

// Create collection if it does not exist and store data
collection
Expand All @@ -146,17 +148,17 @@ public static void dataStorageWithAzureAISearch(
}
var searchResult = results.get(0);
System.out.printf("Search result with score: %f.%n Link: %s, Description: %s%n",
searchResult.getScore(), searchResult.getRecord().link, searchResult.getRecord().description);
searchResult.getScore(), searchResult.getRecord().link,
searchResult.getRecord().description);
}


private static Mono<List<VectorSearchResult<GitHubFile>>> search(
String searchText,
AzureAISearchVectorStoreRecordCollection<GitHubFile> recordCollection,
OpenAITextEmbeddingGenerationService embeddingGeneration) {
String searchText,
AzureAISearchVectorStoreRecordCollection<GitHubFile> recordCollection,
OpenAITextEmbeddingGenerationService embeddingGeneration) {

return embeddingGeneration.generateEmbeddingsAsync(Collections.singletonList(searchText))
.flatMap(r -> recordCollection.searchAsync(r.get(0).getVector(), null));
.flatMap(r -> recordCollection.searchAsync(r.get(0).getVector(), null));
}

private static Mono<List<String>> storeData(
Expand Down

0 comments on commit 09e0388

Please sign in to comment.