New examples (ListVectorIds, QueryVectorsByFilter, DescribeIndex) + Version 0.1.2

Author: peterbanda

build.sbt

@@ -2,7 +2,7 @@ organization := "io.cequence"
 
 name := "pinecone-scala-demo"
 
-version := "0.1.0"
+version := "0.1.2"
 
 // Supported Scala versions
 val scala212 = "2.12.15"

src/main/scala/io/cequence/pineconescala/demo/DescribeIndex.scala

@@ -0,0 +1,9 @@
+package io.cequence.pineconescala.demo
+
+// run me - env. variables PINECONE_SCALA_CLIENT_API_KEY and PINECONE_SCALA_CLIENT_ENV must be set
+object DescribeIndex extends PineconeDemoApp {
+  override protected def exec =
+    pineconeIndexService.describeIndex("auto-gpt-test").map(
+      _.foreach(println)
+    )
+}

src/main/scala/io/cequence/pineconescala/demo/ListIndexes.scala

@@ -1,6 +1,6 @@
 package io.cequence.pineconescala.demo
 
-// run me - env. variables PINECONE_SCALA_CLIENT_API_KEY and PINECONE_SCALA_CLIENT_ENV must be set
+// run me - env. variables PINECONE_SCALA_CLIENT_API_KEY and PINECONE_SCALA_CLIENT_ENV (optional) must be set
 object ListIndexes extends PineconeDemoApp {
   override protected def exec =
     pineconeIndexService.listIndexes.map(

src/main/scala/io/cequence/pineconescala/demo/ListVectorIds.scala

@@ -0,0 +1,17 @@
+package io.cequence.pineconescala.demo
+
+// run me - env. variables PINECONE_SCALA_CLIENT_API_KEY and PINECONE_SCALA_CLIENT_ENV must be set
+object ListVectorIds extends PineconeDemoApp {
+  override protected def exec =
+    createPineconeVectorService("auto-gpt-test").flatMap(
+      _.listVectorIDs(
+        namespace = "wild-test",
+        limit = Some(10),
+      )
+    ).map { queryResponse =>
+      val ids = queryResponse.vectors.map(_.id).mkString(", ")
+      println(s"Vector Ids: ${ids}")
+      println(s"Pagination: ${queryResponse.pagination}")
+      println(s"Namespace : ${queryResponse.namespace}")
+    }
+}

src/main/scala/io/cequence/pineconescala/demo/QueryVectors.scala

@@ -1,15 +1,20 @@
 package io.cequence.pineconescala.demo
 
+import io.cequence.pineconescala.domain.settings.QuerySettings
+
 import scala.util.Random
 
 // run me - env. variables PINECONE_SCALA_CLIENT_API_KEY and PINECONE_SCALA_CLIENT_ENV must be set
 object QueryVectors extends PineconeDemoApp {
 
+  private val indexName = "auto-gpt-test"
+  private val namespace = "my_namespace"
+
   override protected def exec =
-    createPineconeVectorService("auto-gpt-test").flatMap(
+    createPineconeVectorService(indexName).flatMap(
       _.query(
         vector = Seq.fill(1536)(Random.nextDouble), // some values/embeddings
-        namespace = "my_namespace"
+        namespace
       )
     ).map { queryResponse =>
       queryResponse.matches.foreach { matchInfo =>

src/main/scala/io/cequence/pineconescala/demo/QueryVectorsByFilter.scala

@@ -0,0 +1,36 @@
+package io.cequence.pineconescala.demo
+
+import io.cequence.pineconescala.domain.settings.QuerySettings
+
+import scala.util.Random
+
+// run me - env. variables PINECONE_SCALA_CLIENT_API_KEY and PINECONE_SCALA_CLIENT_ENV must be set
+object QueryVectorsByFilter extends PineconeDemoApp {
+
+  private val indexName = "auto-gpt-test"
+  private val namespace = "my_namespace"
+
+  override protected def exec =
+    createPineconeVectorService(indexName).flatMap(
+      _.query(
+        vector = Seq.fill(1536)(Random.nextDouble), // some values/embeddings
+        namespace,
+        settings = QuerySettings(
+          topK = 10,
+          includeValues = false,
+          includeMetadata = true,
+          filter = Map(
+            "subjectId" -> Seq("1234", "5678")
+          )
+        )
+      )
+    ).map { queryResponse =>
+      println("Matches #: " + queryResponse.matches.size)
+      queryResponse.matches.foreach { matchInfo =>
+        println(s"Matched vector id: ${matchInfo.id}")
+        println(s"Matched vector values: ${matchInfo.values.take(20).mkString(", ")}..")  // by default values are not included
+        println(s"Matched vector score: ${matchInfo.score}")
+        println(s"Matched vector metadata: ${matchInfo.metadataUnwrapped.filter(_._1 != "text")}")               // by default metadata is included
+      }
+    }
+}