RDBC-1041: add Version field to GenAiConfiguration for hash version backward compatibility

Author: poissoncorp

ravendb/documents/operations/ai/gen_ai_configuration.py

@@ -33,6 +33,7 @@ def __init__(
         queries: List[AiAgentToolQuery] = None,
         enable_tracing: bool = False,
         expiration_in_sec: int = None,
+        version: int = None,
         disabled: bool = False,
         mentor_node: str = None,
         pin_to_mentor_node: bool = False,
@@ -60,6 +61,7 @@ def __init__(
         self.queries: List[AiAgentToolQuery] = queries or []
         self.enable_tracing = enable_tracing
         self.expiration_in_sec: Optional[int] = expiration_in_sec
+        self.version: Optional[int] = version
 
         self._transforms: Optional[List[Transformation]] = None
 
@@ -179,6 +181,8 @@ def to_json(self) -> Dict[str, Any]:
                 "ExpirationInSec": self.expiration_in_sec,
             }
         )
+        if self.version is not None:
+            result["Version"] = self.version
         return result
 
     @classmethod
@@ -201,6 +205,7 @@ def from_json(cls, json_dict: Dict[str, Any]) -> "GenAiConfiguration":
             queries=[AiAgentToolQuery.from_json(q) for q in queries_data] if queries_data else None,
             enable_tracing=json_dict.get("EnableTracing", False),
             expiration_in_sec=json_dict.get("ExpirationInSec"),
+            version=json_dict.get("Version"),
             disabled=json_dict.get("Disabled", False),
             mentor_node=json_dict.get("MentorNode"),
             pin_to_mentor_node=json_dict.get("PinToMentorNode", False),

ravendb/tests/gen_ai_tests/test_gen_ai_operations.py

@@ -253,6 +253,70 @@ def test_gen_ai_configuration_round_trip(self):
         self.assertEqual(original.max_concurrency, restored.max_concurrency)
         self.assertEqual(original.enable_tracing, restored.enable_tracing)
 
+    def test_version_not_in_json_when_none(self):
+        """Tests that Version key is omitted from JSON when version is None."""
+        config = GenAiConfiguration(
+            name="TestGenAi",
+            identifier="test-gen-ai-1",
+            collection="Documents",
+            connection_string_name="my-connection",
+            prompt="Test prompt",
+            gen_ai_transformation=GenAiTransformation(script="ai.genContext(ctx);"),
+            update_script="this.Summary = $result;",
+            sample_object='{"summary": "test"}',
+        )
+        json_data = config.to_json()
+        self.assertNotIn("Version", json_data)
+
+    def test_version_in_json_when_set(self):
+        """Tests that Version key is included in JSON when version has a value."""
+        config = GenAiConfiguration(
+            name="TestGenAi",
+            identifier="test-gen-ai-1",
+            collection="Documents",
+            connection_string_name="my-connection",
+            prompt="Test prompt",
+            gen_ai_transformation=GenAiTransformation(script="ai.genContext(ctx);"),
+            update_script="this.Summary = $result;",
+            sample_object='{"summary": "test"}',
+            version=1,
+        )
+        json_data = config.to_json()
+        self.assertEqual(1, json_data["Version"])
+
+    def test_version_round_trip(self):
+        """Tests that version survives serialization -> deserialization."""
+        config = GenAiConfiguration(
+            name="TestGenAi",
+            identifier="test-gen-ai-1",
+            collection="Documents",
+            connection_string_name="my-connection",
+            prompt="Test prompt",
+            gen_ai_transformation=GenAiTransformation(script="ai.genContext(ctx);"),
+            update_script="this.Summary = $result;",
+            sample_object='{"summary": "test"}',
+            version=1,
+        )
+        json_data = config.to_json()
+        restored = GenAiConfiguration.from_json(json_data)
+        self.assertEqual(1, restored.version)
+
+    def test_version_none_round_trip(self):
+        """Tests that None version survives serialization -> deserialization."""
+        config = GenAiConfiguration(
+            name="TestGenAi",
+            identifier="test-gen-ai-1",
+            collection="Documents",
+            connection_string_name="my-connection",
+            prompt="Test prompt",
+            gen_ai_transformation=GenAiTransformation(script="ai.genContext(ctx);"),
+            update_script="this.Summary = $result;",
+            sample_object='{"summary": "test"}',
+        )
+        json_data = config.to_json()
+        restored = GenAiConfiguration.from_json(json_data)
+        self.assertIsNone(restored.version)
+
 
 class TestGenAiTransformation(unittest.TestCase):
     """Tests for GenAiTransformation. No server required."""