update flag

pinecone-io · Sep 4, 2024 · da7ebc2 · da7ebc2
1 parent 2cc2ce5
commit da7ebc2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 10 deletions.
diff --git a/src/main/java/io/pinecone/configs/PineconeConfig.java b/src/main/java/io/pinecone/configs/PineconeConfig.java
@@ -52,7 +52,7 @@ public class PineconeConfig {
     private String sourceTag;
     private ProxyConfig proxyConfig;
     private ManagedChannel customManagedChannel;
-    private boolean usePineconeLocal = false;
+    private boolean enableTLS = true;
 
     /**
      * Constructs a {@link PineconeConfig} instance with the specified API key.
@@ -208,12 +208,22 @@ public String getUserAgent() {
         return buildUserAgent();
     }
 
-    public boolean getUsePineconeLocal() {
-        return usePineconeLocal;
+    /**
+     * Returns true if TLS is enabled or false otherwise.
+     *
+     * @return enableTls
+     */
+    public boolean getEnableTLS() {
+        return enableTLS;
     }
 
-    public void setPineconeLocal(boolean pineconeLocal) {
-        this.usePineconeLocal = pineconeLocal;
+    /**
+     * Sets whether TLS is enabled.
+     *
+     * @param enableTLS true to enable TLS, false to disable it.
+     */
+    public void setEnableTLS(boolean enableTLS) {
+        this.enableTLS = enableTLS;
     }
 
     private String buildUserAgent() {

diff --git a/src/main/java/io/pinecone/configs/PineconeConnection.java b/src/main/java/io/pinecone/configs/PineconeConnection.java
@@ -139,17 +139,17 @@ private ManagedChannel buildChannel() {
         NettyChannelBuilder builder = NettyChannelBuilder.forTarget(endpoint);
 
         try {
-            if(config.getUsePineconeLocal()) {
-                builder = builder
-                        .negotiationType(NegotiationType.PLAINTEXT);
-            }
-            else {
+            if(config.getEnableTLS()) {
                 builder = builder
                         .overrideAuthority(endpoint)
                         .negotiationType(NegotiationType.TLS)
                         .sslContext(GrpcSslContexts.forClient().build())
                         .userAgent(config.getUserAgent());
             }
+            else {
+                builder = builder
+                        .negotiationType(NegotiationType.PLAINTEXT);
+            }
 
             if(config.getProxyConfig() != null) {
                 ProxyDetector proxyDetector = getProxyDetector();