Added support for knowledge param

Author: Jinash Rouniyar

integration/test_collection_openai.py

@@ -787,6 +787,51 @@ def test_contextualai_generative_search_single(
     assert res.generated is None
 
 
+def test_contextualai_generative_with_knowledge_parameter(
+    collection_factory: CollectionFactory,
+) -> None:
+    """Test Contextual AI generative search with knowledge parameter override."""
+    api_key = os.environ.get("CONTEXTUAL_API_KEY")
+    if api_key is None:
+        pytest.skip("No Contextual AI API key found.")
+
+    collection = collection_factory(
+        name="TestContextualAIGenerativeKnowledge",
+        generative_config=Configure.Generative.contextualai(
+            model="v2",
+            max_tokens=100,
+            temperature=0.1,
+            system_prompt="You are a helpful assistant.",
+            avoid_commentary=False,
+        ),
+        vectorizer_config=Configure.Vectorizer.none(),
+        properties=[
+            Property(name="text", data_type=DataType.TEXT),
+        ],
+        headers={"X-Contextual-Api-Key": api_key},
+        ports=(8086, 50057),
+    )
+    if collection._connection._weaviate_version.is_lower_than(1, 23, 1):
+        pytest.skip("Generative search requires Weaviate 1.23.1 or higher")
+
+    collection.data.insert_many(
+        [
+            DataObject(properties={"text": "base knowledge"}),
+        ]
+    )
+
+    # Test with knowledge parameter override
+    res = collection.generate.fetch_objects(
+        single_prompt="What is the custom knowledge?",
+        config=GenerativeConfig.contextualai(
+            knowledge=["Custom knowledge override", "Additional context"],
+        ),
+    )
+    for obj in res.objects:
+        assert obj.generated is not None
+        assert isinstance(obj.generated, str)
+
+
 def test_contextualai_generative_and_rerank_combined(collection_factory: CollectionFactory) -> None:
     """Test Contextual AI generative search combined with reranking."""
     contextual_api_key = os.environ.get("CONTEXTUAL_API_KEY")

test/collection/test_classes_generative.py

@@ -422,6 +422,7 @@ def test_generative_parameters_images_parsing(
                 top_p=0.9,
                 system_prompt="You are a helpful assistant that provides accurate and informative responses based on the given context.",
                 avoid_commentary=False,
+                knowledge=["knowledge1", "knowledge2"],
             )._to_grpc(_GenerativeConfigRuntimeOptions(return_metadata=True)),
             generative_pb2.GenerativeProvider(
                 return_metadata=True,
@@ -432,6 +433,7 @@ def test_generative_parameters_images_parsing(
                     top_p=0.9,
                     system_prompt="You are a helpful assistant that provides accurate and informative responses based on the given context.",
                     avoid_commentary=False,
+                    knowledge=base_pb2.TextArray(values=["knowledge1", "knowledge2"]),
                 ),
             ),
         ),

weaviate/collections/classes/generative.py

@@ -455,6 +455,7 @@ class _GenerativeContextualAI(_GenerativeConfigRuntime):
     top_p: Optional[float]
     system_prompt: Optional[str]
     avoid_commentary: Optional[bool]
+    knowledge: Optional[List[str]]
 
     def _to_grpc(self, opts: _GenerativeConfigRuntimeOptions) -> generative_pb2.GenerativeProvider:
         self._validate_multi_modal(opts)
@@ -467,6 +468,7 @@ def _to_grpc(self, opts: _GenerativeConfigRuntimeOptions) -> generative_pb2.Gene
                 top_p=self.top_p,
                 system_prompt=self.system_prompt,
                 avoid_commentary=self.avoid_commentary or False,
+                knowledge=_to_text_array(self.knowledge),
             ),
         )
 
@@ -615,6 +617,7 @@ def contextualai(
         top_p: Optional[float] = None,
         system_prompt: Optional[str] = None,
         avoid_commentary: Optional[bool] = None,
+        knowledge: Optional[List[str]] = None,
     ) -> _GenerativeConfigRuntime:
         """Create a `_GenerativeContextualAI` object for use with the `generative-contextualai` module.
 
@@ -625,6 +628,7 @@ def contextualai(
             top_p: The top P to use. Defaults to `None`, which uses the server-defined default
             system_prompt: The system prompt to prepend to the conversation
             avoid_commentary: Whether to avoid model commentary in responses
+            knowledge: Optional knowledge array to override the default knowledge from retrieved objects
         """
         return _GenerativeContextualAI(
             model=model,
@@ -633,6 +637,7 @@ def contextualai(
             top_p=top_p,
             system_prompt=system_prompt,
             avoid_commentary=avoid_commentary,
+            knowledge=knowledge,
         )
 
     @staticmethod