Updated support for knowledge param

Author: Jinash Rouniyar

src/collections/generate/config.ts

@@ -319,6 +319,7 @@ export const generativeParameters = {
             topP: config.topP,
             systemPrompt: config.systemPrompt,
             avoidCommentary: config.avoidCommentary,
+            knowledge: config.knowledge,
           }
         : undefined,
     };

src/collections/generate/integration.test.ts

@@ -676,6 +676,29 @@ maybeContextualAI('Testing of the collection.generate methods with Contextual AI
       }
     );
 
+    expect(response.objects).toHaveLength(1);
+    expect(response.objects[0].generative).toBeDefined();
+    expect(response.objects[0].generative?.text).toBeDefined();
+    expect(typeof response.objects[0].generative?.text).toBe('string');
+  });
+
+  it('should handle knowledge parameter override in runtime configuration', async () => {
+    const response = await collection.generate.nearText(
+      'machine learning',
+      {
+        singlePrompt: 'What is the custom knowledge?',
+        config: generativeParameters.contextualai({
+          model: 'v2',
+          maxTokens: 100,
+          temperature: 0.7,
+          knowledge: ['Custom knowledge override', 'Additional context for testing'],
+        }),
+      },
+      {
+        limit: 1,
+      }
+    );
+
     expect(response.objects).toHaveLength(1);
     expect(response.objects[0].generative).toBeDefined();
     expect(response.objects[0].generative?.text).toBeDefined();

src/collections/generate/unit.test.ts

@@ -332,6 +332,7 @@ describe('Unit testing of the generativeParameters factory methods', () => {
         topP: 0.9,
         systemPrompt: 'sys',
         avoidCommentary: false,
+        knowledge: ['knowledge1', 'knowledge2'],
       });
       expect(config).toEqual<
         ModuleConfig<
@@ -347,6 +348,7 @@ describe('Unit testing of the generativeParameters factory methods', () => {
           topP: 0.9,
           systemPrompt: 'sys',
           avoidCommentary: false,
+          knowledge: ['knowledge1', 'knowledge2'],
         },
       });
     });

src/collections/types/generate.ts

@@ -182,6 +182,7 @@ export type GenerativeConfigRuntimeType<G> = G extends 'generative-anthropic'
       topP?: number;
       systemPrompt?: string;
       avoidCommentary?: boolean;
+      knowledge?: string[];
     }
   : G extends 'none'
   ? undefined
@@ -347,4 +348,5 @@ export type GenerativeContextualAIConfigRuntime = {
   topP?: number | undefined;
   systemPrompt?: string | undefined;
   avoidCommentary?: boolean | undefined;
+  knowledge?: string[] | undefined;
 };