RDBC-1059 Add MissingAiAgentParameterException, IMAGE field, Azure endpoint validation

Author: poissoncorp

ravendb/documents/ai/content_part.py

@@ -3,24 +3,16 @@
 
 
 class AiMessagePromptFields:
-    """Constants for AI message prompt field names."""
-
     TEXT = "text"
     TYPE = "type"
+    IMAGE = "image"
 
 
 class AiMessagePromptTypes:
-    """Constants for AI message prompt types."""
-
     TEXT = "text"
 
 
 class ContentPart:
-    """
-    Base class for content parts in AI prompts.
-    Content parts allow structured prompt content with different types (text, etc.).
-    """
-
     def __init__(self, content_type: str):
         self._type = content_type
 
@@ -29,18 +21,10 @@ def type(self) -> str:
         return self._type
 
     def to_json(self) -> Dict[str, Any]:
-        """
-        Converts the content part to a JSON-serializable dictionary.
-        Subclasses should override this method to include their specific fields.
-        """
         return {AiMessagePromptFields.TYPE: self._type}
 
 
 class TextPart(ContentPart):
-    """
-    Represents a text content part in AI prompts.
-    """
-
     def __init__(self, text: str):
         super().__init__(AiMessagePromptTypes.TEXT)
         self._text = text

ravendb/documents/operations/ai/azure_open_ai_settings.py

@@ -17,6 +17,8 @@ def __init__(
         super().__init__(api_key, endpoint, model, dimensions, temperature, embeddings_max_concurrent_batches)
         if deployment_name is None:
             raise ValueError("deployment_name cannot be None")
+        if endpoint is None or (isinstance(endpoint, str) and endpoint.strip() == ""):
+            raise ValueError("endpoint cannot be None or empty")
         self.deployment_name = deployment_name
 
     @classmethod

ravendb/exceptions/exception_dispatcher.py

@@ -15,6 +15,7 @@
     ConcurrencyException,
     IndexCompactionInProgressException,
     InsufficientQuotaException,
+    MissingAiAgentParameterException,
     PortInUseException,
     RateLimitException,
     RavenException,
@@ -44,6 +45,7 @@
     "RateLimitException": RateLimitException,
     "InsufficientQuotaException": InsufficientQuotaException,
     "TooManyTokensException": TooManyTokensException,
+    "MissingAiAgentParameterException": MissingAiAgentParameterException,
     # documents
     "DocumentConflictException": DocumentConflictException,
     "DocumentDoesNotExistException": DocumentDoesNotExistException,

ravendb/exceptions/raven_exceptions.py

@@ -82,6 +82,10 @@ class TooManyTokensException(TooManyRequestsException):
     pass
 
 
+class MissingAiAgentParameterException(RavenException):
+    pass
+
+
 class SchemaValidationException(RavenException):
     def __init__(self, message: str = None):
         super().__init__(message)