Remove tags from Redis

microsoft · Oct 31, 2024 · 5b791ad · 5b791ad
1 parent 8ba80a8
commit 5b791ad
Showing 1 changed file with 50 additions and 1 deletion.
diff --git a/.../microsoft/semantickernel/samples/documentationexamples/data/vectorstores/redis/Main.java b/.../microsoft/semantickernel/samples/documentationexamples/data/vectorstores/redis/Main.java
@@ -5,9 +5,16 @@
 import com.microsoft.semantickernel.data.redis.RedisStorageType;
 import com.microsoft.semantickernel.data.redis.RedisVectorStore;
 import com.microsoft.semantickernel.data.redis.RedisVectorStoreOptions;
-import com.microsoft.semantickernel.samples.documentationexamples.data.index.Hotel;
+import com.microsoft.semantickernel.data.vectorstorage.annotations.VectorStoreRecordData;
+import com.microsoft.semantickernel.data.vectorstorage.annotations.VectorStoreRecordKey;
+import com.microsoft.semantickernel.data.vectorstorage.annotations.VectorStoreRecordVector;
+import com.microsoft.semantickernel.data.vectorstorage.definition.DistanceFunction;
+import com.microsoft.semantickernel.data.vectorstorage.definition.IndexKind;
 import redis.clients.jedis.JedisPooled;
 
+import java.util.Collections;
+import java.util.List;
+
 public class Main {
     public static void main(String[] args) {
         JedisPooled jedis = new JedisPooled("<your-redis-url>");
@@ -34,4 +41,46 @@ public static void main(String[] args) {
 
         collection.getAsync("myprefix_h1", null).block();
     }
+
+    public static class Hotel {
+        @VectorStoreRecordKey
+        private String hotelId;
+
+        @VectorStoreRecordData(isFilterable = true)
+        private String name;
+
+        @VectorStoreRecordData(isFullTextSearchable = true)
+        private String description;
+
+        @VectorStoreRecordVector(dimensions = 4, indexKind = IndexKind.HNSW, distanceFunction = DistanceFunction.COSINE_DISTANCE)
+        private List<Float> descriptionEmbedding;
+
+        public Hotel() {
+        }
+
+        public Hotel(String hotelId, String name, String description, List<Float> descriptionEmbedding) {
+            this.hotelId = hotelId;
+            this.name = name;
+            this.description = description;
+            this.descriptionEmbedding = Collections.unmodifiableList(descriptionEmbedding);
+        }
+
+        public String getHotelId() {
+            return hotelId;
+        }
+
+        public String getName() {
+            return name;
+        }
+
+        public String getDescription() {
+            return description;
+        }
+
+        public List<Float> getDescriptionEmbedding() {
+            return descriptionEmbedding;
+        }
+    }
 }
+
+