RDBC-934 Remove AiAgentParametersBuilder

Author: poissoncorp

ravendb/__init__.py

@@ -82,8 +82,6 @@
     AiOperations,
     AiConversation,
     AiConversationResult,
-    AiAgentParametersBuilder,
-    IAiAgentParametersBuilder,
 )
 from ravendb.documents.operations.ai.agents import (
     AiAgentConfiguration,

ravendb/documents/ai/__init__.py

@@ -1,15 +1,12 @@
 from .ai_operations import AiOperations
 from .ai_conversation import AiConversation
 from .ai_conversation_result import AiConversationResult
-from .ai_agent_parameters_builder import AiAgentParametersBuilder, IAiAgentParametersBuilder
 from .ai_answer import AiAnswer, AiConversationStatus
 
 __all__ = [
     "AiOperations",
     "AiConversation",
     "AiConversationResult",
-    "AiAgentParametersBuilder",
-    "IAiAgentParametersBuilder",
     "AiAnswer",
     "AiConversationStatus",
 ]

ravendb/documents/ai/ai_agent_parameters_builder.py

@@ -24,7 +24,7 @@ def __str__(self):
 class AiAnswer(Generic[TAnswer]):
     """
     Represents the answer from an AI conversation turn.
-    
+
     This class contains the AI's response, the conversation status,
     token usage statistics, and timing information.
     """
@@ -66,4 +66,3 @@ def __repr__(self) -> str:
             f"usage={self.usage!r}, "
             f"elapsed={self.elapsed!r})"
         )
-

ravendb/documents/ai/ai_answer.py

@@ -143,11 +143,7 @@ def _run_internal(self, answer_type: type = dict) -> AiAnswer:
         import time
 
         # If we already went to the server and have nothing new to tell it, we're done
-        if (
-            self._action_requests is not None
-            and len(self._prompt_parts) == 0
-            and len(self._action_responses) == 0
-        ):
+        if self._action_requests is not None and len(self._prompt_parts) == 0 and len(self._action_responses) == 0:
             return AiAnswer(
                 answer=None,
                 status=AiConversationStatus.DONE,
@@ -188,7 +184,11 @@ def _run_internal(self, answer_type: type = dict) -> AiAnswer:
             # Build AiAnswer
             return AiAnswer(
                 answer=result.response,
-                status=AiConversationStatus.ACTION_REQUIRED if len(self._action_requests) > 0 else AiConversationStatus.DONE,
+                status=(
+                    AiConversationStatus.ACTION_REQUIRED
+                    if len(self._action_requests) > 0
+                    else AiConversationStatus.DONE
+                ),
                 usage=result.usage,
                 elapsed=elapsed,
             )

ravendb/documents/ai/ai_conversation.py

@@ -120,11 +120,7 @@ class AiConversationCreationOptions:
     Options for creating AI agent conversations, including parameters and expiration settings.
     """
 
-    def __init__(
-        self,
-        parameters: Optional[Dict[str, Any]] = None,
-        expiration_in_sec: Optional[int] = None
-    ):
+    def __init__(self, parameters: Optional[Dict[str, Any]] = None, expiration_in_sec: Optional[int] = None):
         self.expiration_in_sec: Optional[int] = expiration_in_sec
         self.parameters: Optional[Dict[str, Any]] = parameters