Add sourceTag to PineconeConfiguration and additionalHeaders to data plane calls

src/control/__tests__/indexOperationsBuilder.test.ts

@@ -0,0 +1,40 @@
+import { indexOperationsBuilder } from '../indexOperationsBuilder';
+import { Configuration } from '../../pinecone-generated-ts-fetch';
+
+jest.mock('../../pinecone-generated-ts-fetch', () => ({
+  ...jest.requireActual('../../pinecone-generated-ts-fetch'),
+  Configuration: jest.fn(),
+}));
+
+describe('indexOperationsBuilder', () => {
+  test('API Configuration basePath is set to api.pinecone.io by default', () => {
+    const config = { apiKey: 'test-api-key' };
+    indexOperationsBuilder(config);
+    expect(Configuration).toHaveBeenCalledWith(
+      expect.objectContaining({ basePath: 'https://api.pinecone.io' })
+    );
+  });
+
+  test('controllerHostUrl overwrites the basePath in API Configuration', () => {
+    const controllerHostUrl = 'https://test-controller-host-url.io';
+    const config = {
+      apiKey: 'test-api-key',
+      controllerHostUrl,
+    };
+    indexOperationsBuilder(config);
+    expect(Configuration).toHaveBeenCalledWith(
+      expect.objectContaining({ basePath: controllerHostUrl })
+    );
+  });
+
+  test('additionalHeaders are passed to the API Configuration', () => {
+    const additionalHeaders = { 'x-test-header': 'test-value' };
+    const config = { apiKey: 'test-api-key', additionalHeaders };
+    indexOperationsBuilder(config);
+    expect(Configuration).toHaveBeenCalledWith(
+      expect.objectContaining({
+        headers: expect.objectContaining(additionalHeaders),
+      })
+    );
+  });
+});

src/control/indexOperationsBuilder.ts

@@ -24,7 +24,7 @@ export const indexOperationsBuilder = (
     apiKey,
     queryParamsStringify,
     headers: {
-      'User-Agent': buildUserAgent(),
+      'User-Agent': buildUserAgent(config),
       ...headers,
     },
     fetchApi: getFetch(config),

src/data/__tests__/dataOperationsProvider.test.ts

@@ -1,8 +1,17 @@
 import { DataOperationsProvider } from '../dataOperationsProvider';
 import { IndexHostSingleton } from '../indexHostSingleton';
+import { Configuration } from '../../pinecone-generated-ts-fetch';
+
+jest.mock('../../pinecone-generated-ts-fetch', () => ({
+  ...jest.requireActual('../../pinecone-generated-ts-fetch'),
+  Configuration: jest.fn(),
+}));
 
 describe('DataOperationsProvider', () => {
   let real;
+  const config = {
+    apiKey: 'test-api-key',
+  };
 
   beforeAll(() => {
     real = IndexHostSingleton.getHostUrl;
@@ -41,9 +50,6 @@ describe('DataOperationsProvider', () => {
   });
 
   test('passing indexHostUrl skips hostUrl resolution', async () => {
-    const config = {
-      apiKey: 'test-api-key',
-    };
     const indexHostUrl = 'http://index-host-url';
     const provider = new DataOperationsProvider(
       config,
@@ -58,4 +64,21 @@ describe('DataOperationsProvider', () => {
     expect(IndexHostSingleton.getHostUrl).not.toHaveBeenCalled();
     expect(provider.buildDataOperationsConfig).toHaveBeenCalled();
   });
+
+  test('passing additionalHeaders applies them to the API Configuration', async () => {
+    const additionalHeaders = { 'x-custom-header': 'custom-value' };
+    const provider = new DataOperationsProvider(
+      config,
+      'index-name',
+      undefined,
+      additionalHeaders
+    );
+
+    await provider.provide();
+    expect(Configuration).toHaveBeenCalledWith(
+      expect.objectContaining({
+        headers: expect.objectContaining(additionalHeaders),
+      })
+    );
+  });
 });