fix search

Author: clun

astra-db-java/src/main/java/com/datastax/astra/client/DataAPIClients.java

@@ -78,6 +78,7 @@ public static DataAPIClient createForLocal() {
                 new UsernamePasswordTokenProvider().getToken(),
                 DataAPIOptions.builder()
                         .withDestination(DataAPIOptions.DataAPIDestination.CASSANDRA)
+                        .logRequests()
                         .withObserver(new LoggingCommandObserver(DataAPIClient.class))
                         .build());
     }

langchain4j-astradb/src/main/java/com/datastax/astra/langchain4j/store/embedding/AstraDbEmbeddingStore.java

@@ -32,6 +32,7 @@
 import dev.langchain4j.data.embedding.Embedding;
 import dev.langchain4j.data.segment.TextSegment;
 import dev.langchain4j.store.embedding.EmbeddingMatch;
+import dev.langchain4j.store.embedding.EmbeddingSearchRequest;
 import dev.langchain4j.store.embedding.EmbeddingSearchResult;
 import dev.langchain4j.store.embedding.EmbeddingStore;
 import lombok.Getter;
@@ -215,6 +216,18 @@ public List<EmbeddingMatch<TextSegment>> findRelevant(Embedding referenceEmbeddi
         return findRelevant(referenceEmbedding, (Filter) null, maxResults, minScore);
     }
 
+    public EmbeddingSearchResult<TextSegment> search(EmbeddingSearchRequest request) {
+        dev.langchain4j.store.embedding.filter.Filter lc4jFilter = request.filter();
+        if (lc4jFilter != null) {
+            // Map Filter
+            Filter astraFilter = AstraDbFilterMapper.map(lc4jFilter);
+            List<EmbeddingMatch<TextSegment>> matches = this.findRelevant(request.queryEmbedding(), astraFilter, request.maxResults(), request.minScore());
+            return new EmbeddingSearchResult(matches);
+        }
+        List<EmbeddingMatch<TextSegment>> matches2 = this.findRelevant(request.queryEmbedding(), request.maxResults(), request.minScore());
+        return new EmbeddingSearchResult(matches2);
+    }
+
     /**
      * Implementation of the Search to add the metadata Filtering.
      *