add upsert and search records

Author: rohanshah18

src/main/java/io/pinecone/clients/Pinecone.java

@@ -982,6 +982,12 @@ public AsyncIndex getAsyncIndexConnection(String indexName) throws PineconeValid
         return new AsyncIndex(config, connection, indexName);
     }
 
+    public RestIndex getRestIndexConnection(String indexName) throws PineconeValidationException {
+        PineconeConfig perConnectionConfig = new PineconeConfig(config.getApiKey(), config.getSourceTag());
+        perConnectionConfig.setHost(getIndexHost(indexName));
+        return new RestIndex(perConnectionConfig);
+    }
+
     /**
      * A method to create and return a new instance of the {@link Inference} client.
      * <p>

src/main/java/io/pinecone/clients/RestAsyncIndex.java

@@ -0,0 +1,4 @@
+package io.pinecone.clients;
+
+public class RestAsyncIndex {
+}

src/main/java/io/pinecone/clients/RestIndex.java

@@ -0,0 +1,44 @@
+package io.pinecone.clients;
+
+import io.pinecone.configs.PineconeConfig;
+import okhttp3.OkHttpClient;
+import org.openapitools.db_data.client.ApiClient;
+import org.openapitools.db_data.client.ApiException;
+import org.openapitools.db_data.client.Configuration;
+import org.openapitools.db_data.client.api.VectorOperationsApi;
+import org.openapitools.db_data.client.model.*;
+
+import java.util.List;
+
+import static io.pinecone.clients.Pinecone.buildOkHttpClient;
+
+public class RestIndex {
+    private final VectorOperationsApi vectorOperations;
+
+    public RestIndex(PineconeConfig config) {
+        OkHttpClient customOkHttpClient = config.getCustomOkHttpClient();
+        ApiClient apiClient = (customOkHttpClient != null) ? new ApiClient(customOkHttpClient) : new ApiClient(buildOkHttpClient(config.getProxyConfig()));
+        apiClient.setApiKey(config.getApiKey());
+        apiClient.setUserAgent(config.getUserAgent());
+        apiClient.addDefaultHeader("X-Pinecone-Api-Version", Configuration.VERSION);
+
+        this.vectorOperations = new VectorOperationsApi(apiClient);
+        String protocol = config.isTLSEnabled() ? "https://" : "http://";
+        vectorOperations.setCustomBaseUrl(protocol + config.getHost());
+    }
+
+    public void upsertRecords(String namespace, List<UpsertRecord> upsertRecord) throws ApiException {
+        vectorOperations.upsertRecordsNamespace(namespace, upsertRecord);
+    }
+
+    public SearchRecordsResponse searchRecords(String id, String namespace, Object filter, int topK, SearchRecordsVector vector) throws ApiException {
+        SearchRecordsRequestQuery searchRecordsRequestquery = new SearchRecordsRequestQuery()
+                .id(id)
+                .filter(filter)
+                .topK(topK)
+                .vector(vector);
+        SearchRecordsRequest request = new SearchRecordsRequest().query(searchRecordsRequestquery);
+
+        return vectorOperations.searchRecordsNamespace(namespace, request);
+    }
+}

src/main/java/org/openapitools/db_data/client/ApiClient.java

@@ -931,6 +931,18 @@ public RequestBody serialize(Object obj, String contentType) throws ApiException
             return RequestBody.create((File) obj, MediaType.parse(contentType));
         } else if ("text/plain".equals(contentType) && obj instanceof String) {
             return RequestBody.create((String) obj, MediaType.parse(contentType));
+        } else if ("application/x-ndjson".equals(contentType)) {
+            // Handle NDJSON (Newline Delimited JSON)
+            if (obj instanceof Iterable) { // If obj is a collection of objects
+                StringBuilder ndjsonContent = new StringBuilder();
+                for (Object item : (Iterable<?>) obj) {
+                    String json = JSON.serialize(item);
+                    ndjsonContent.append(json).append("\n");
+                }
+                return RequestBody.create(ndjsonContent.toString(), MediaType.parse(contentType));
+            } else {
+                throw new ApiException("NDJSON content requires a collection of objects.");
+            }
         } else if (isJsonMime(contentType)) {
             String content;
             if (obj != null) {