RDBC-925 Run black linter

Author: poissoncorp

ravendb/documents/ai/ai_agent_parameters_builder.py

@@ -17,11 +17,11 @@ class IAiAgentParametersBuilder(ABC, Generic[TResponse]):
     def with_parameter(self, name: str, value: Any) -> IAiAgentParametersBuilder[TResponse]:
         """
         Adds a parameter to the conversation.
-        
+
         Args:
             name: The parameter name
             value: The parameter value
-            
+
         Returns:
             The builder instance for method chaining
         """
@@ -31,7 +31,7 @@ def with_parameter(self, name: str, value: Any) -> IAiAgentParametersBuilder[TRe
     def build(self) -> IAiConversationOperations[TResponse]:
         """
         Builds and returns the conversation operations instance.
-        
+
         Returns:
             The conversation operations interface
         """
@@ -46,7 +46,7 @@ class AiAgentParametersBuilder(IAiAgentParametersBuilder[TResponse]):
     def __init__(self, conversation_factory):
         """
         Initializes the parameters builder.
-        
+
         Args:
             conversation_factory: A callable that creates the conversation with the built parameters
         """
@@ -56,11 +56,11 @@ def __init__(self, conversation_factory):
     def with_parameter(self, name: str, value: Any) -> IAiAgentParametersBuilder[TResponse]:
         """
         Adds a parameter to the conversation.
-        
+
         Args:
             name: The parameter name
             value: The parameter value
-            
+
         Returns:
             The builder instance for method chaining
         """
@@ -70,7 +70,7 @@ def with_parameter(self, name: str, value: Any) -> IAiAgentParametersBuilder[TRe
     def build(self) -> IAiConversationOperations[TResponse]:
         """
         Builds and returns the conversation operations instance.
-        
+
         Returns:
             The conversation operations interface
         """

ravendb/documents/ai/ai_conversation.py

@@ -49,12 +49,12 @@ def with_conversation_id(
     ) -> AiConversation[TResponse]:
         """
         Creates a conversation instance for continuing an existing conversation.
-        
+
         Args:
             store: The document store
             conversation_id: The ID of the existing conversation
             change_vector: Optional change vector for optimistic concurrency
-            
+
         Returns:
             A new conversation instance
         """
@@ -92,9 +92,7 @@ def add_action_response(self, action_id: str, action_response: Union[str, TRespo
             # More robust JSON serialization
             try:
                 response.content = json.dumps(
-                    action_response.__dict__ if hasattr(action_response, '__dict__')
-                    else action_response,
-                    default=str
+                    action_response.__dict__ if hasattr(action_response, "__dict__") else action_response, default=str
                 )
             except (TypeError, ValueError) as e:
                 response.content = str(action_response)
@@ -144,7 +142,7 @@ def run(self) -> AiConversationResult[TResponse]:
             self._change_vector = result.change_vector
 
         # Preserve agent ID for future conversation turns
-        if not self._agent_id and hasattr(operation, '_agent_id'):
+        if not self._agent_id and hasattr(operation, "_agent_id"):
             self._agent_id = operation._agent_id
 
         # Clear processed data for next turn

ravendb/documents/ai/ai_conversation_operations.py

@@ -32,7 +32,7 @@ def required_actions(self) -> List[AiAgentActionRequest]:
     def add_action_response(self, action_id: str, action_response: Union[str, TResponse]) -> None:
         """
         Adds a response for a given action request.
-        
+
         Args:
             action_id: The ID of the action to respond to
             action_response: The response content (string or typed response object)
@@ -45,7 +45,7 @@ def run(self) -> AiConversationResult[TResponse]:
         Executes one "turn" of the conversation:
         sends the current prompt, processes any required actions,
         and awaits the agent's reply.
-        
+
         Returns:
             The result of the conversation turn
         """
@@ -55,7 +55,7 @@ def run(self) -> AiConversationResult[TResponse]:
     def set_user_prompt(self, user_prompt: str) -> None:
         """
         Sets the next user prompt to send to the AI agent.
-        
+
         Args:
             user_prompt: The prompt text to send to the agent
         """

ravendb/documents/ai/ai_conversation_result.py

@@ -55,7 +55,4 @@ def get_action_request_by_id(self, action_id: str) -> Optional[AiAgentActionRequ
         Returns:
             The action request if found, None otherwise
         """
-        return next(
-            (request for request in self.action_requests if request.tool_id == action_id),
-            None
-        )
+        return next((request for request in self.action_requests if request.tool_id == action_id), None)

ravendb/documents/ai/ai_operations.py

@@ -20,89 +20,79 @@ def __init__(self, store: DocumentStore):
         self._store = store
 
     def add_or_update_agent(
-        self, 
-        configuration: AiAgentConfiguration, 
-        schema_type: Type = None
+        self, configuration: AiAgentConfiguration, schema_type: Type = None
     ) -> AiAgentConfigurationResult:
         """
         Adds or updates an AI agent configuration.
-        
+
         Args:
             configuration: The AI agent configuration to add or update
             schema_type: Optional type to use for generating sample schema
-            
+
         Returns:
             Result containing the agent identifier and raft command index
         """
         from ravendb.documents.operations.ai.agents import AddOrUpdateAiAgentOperation
-        
+
         operation = AddOrUpdateAiAgentOperation(configuration, schema_type)
         return self._store.maintenance.send(operation)
 
     def delete_agent(self, identifier: str) -> AiAgentConfigurationResult:
         """
         Deletes an AI agent configuration.
-        
+
         Args:
             identifier: The identifier of the agent to delete
-            
+
         Returns:
             Result containing the raft command index
         """
         from ravendb.documents.operations.ai.agents import DeleteAiAgentOperation
-        
+
         operation = DeleteAiAgentOperation(identifier)
         return self._store.maintenance.send(operation)
 
     def get_agents(self, agent_id: str = None) -> GetAiAgentsResponse:
         """
         Gets AI agent configurations.
-        
+
         Args:
             agent_id: Optional specific agent ID to retrieve. If None, returns all agents.
-            
+
         Returns:
             Response containing the list of AI agent configurations
         """
         from ravendb.documents.operations.ai.agents import GetAiAgentOperation
-        
+
         operation = GetAiAgentOperation(agent_id)
         return self._store.maintenance.send(operation)
 
-    def conversation(
-        self, 
-        agent_id: str, 
-        parameters: Dict[str, Any] = None
-    ) -> IAiConversationOperations:
+    def conversation(self, agent_id: str, parameters: Dict[str, Any] = None) -> IAiConversationOperations:
         """
         Creates a new conversation with the specified AI agent.
-        
+
         Args:
             agent_id: The identifier of the AI agent to start a conversation with
             parameters: Optional parameters to pass to the agent
-            
+
         Returns:
             Conversation operations interface for managing the conversation
         """
         from ravendb.documents.ai.ai_conversation import AiConversation
-        
+
         return AiConversation(self._store, agent_id, parameters)
 
-    def conversation_with_id(
-        self, 
-        conversation_id: str, 
-        change_vector: str = None
-    ) -> IAiConversationOperations:
+    def conversation_with_id(self, conversation_id: str, change_vector: str = None) -> IAiConversationOperations:
         """
         Continues an existing conversation by its ID.
-        
+
         Args:
             conversation_id: The ID of the existing conversation
             change_vector: Optional change vector for optimistic concurrency
-            
+
         Returns:
             Conversation operations interface for managing the conversation
         """
         from ravendb.documents.ai.ai_conversation import AiConversation
-        
+
         return AiConversation.with_conversation_id(self._store, conversation_id, change_vector)

ravendb/documents/operations/ai/agents/add_or_update_ai_agent_operation.py

@@ -31,11 +31,7 @@ def __init__(self, configuration: AiAgentConfiguration, schema_type: Type = None
         if configuration is None:
             raise ValueError("configuration cannot be None")
 
-        if (
-            not configuration.output_schema
-            and not configuration.sample_object
-            and schema_type is None
-        ):
+        if not configuration.output_schema and not configuration.sample_object and schema_type is None:
             raise ValueError(
                 "Please provide a non-empty value for either output_schema or sample_object or schema_type"
             )
@@ -70,7 +66,7 @@ def create_request(self, node: ServerNode) -> requests.Request:
 
         # Set sample object if not provided but we have a schema type
         if not config_to_send.sample_object and self._sample_schema:
-            if hasattr(self._sample_schema, '__dict__'):
+            if hasattr(self._sample_schema, "__dict__"):
                 config_to_send.sample_object = json.dumps(self._sample_schema.__dict__)
             else:
                 config_to_send.sample_object = json.dumps(self._sample_schema)
@@ -96,4 +92,5 @@ def set_response(self, response: str, from_cache: bool) -> None:
     def get_raft_unique_request_id(self) -> str:
         # Generate a unique ID for Raft operations
         import uuid
+
         return str(uuid.uuid4())

ravendb/documents/operations/ai/agents/ai_agent_configuration.py

@@ -32,7 +32,9 @@ def from_json(cls, json_dict: Dict[str, Any]) -> AiAgentToolQuery:
         instance.name = json_dict.get("name") or json_dict.get("Name")
         instance.description = json_dict.get("description") or json_dict.get("Description")
         instance.query = json_dict.get("query") or json_dict.get("Query")
-        instance.parameters_sample_object = json_dict.get("parametersSampleObject") or json_dict.get("ParametersSampleObject")
+        instance.parameters_sample_object = json_dict.get("parametersSampleObject") or json_dict.get(
+            "ParametersSampleObject"
+        )
         instance.parameters_schema = json_dict.get("parametersSchema") or json_dict.get("ParametersSchema")
         return instance
 
@@ -63,7 +65,9 @@ def from_json(cls, json_dict: Dict[str, Any]) -> AiAgentToolAction:
         instance = cls()
         instance.name = json_dict.get("name") or json_dict.get("Name")
         instance.description = json_dict.get("description") or json_dict.get("Description")
-        instance.parameters_sample_object = json_dict.get("parametersSampleObject") or json_dict.get("ParametersSampleObject")
+        instance.parameters_sample_object = json_dict.get("parametersSampleObject") or json_dict.get(
+            "ParametersSampleObject"
+        )
         instance.parameters_schema = json_dict.get("parametersSchema") or json_dict.get("ParametersSchema")
         return instance
 
@@ -88,7 +92,9 @@ def to_json(self) -> Dict[str, Any]:
     def from_json(cls, json_dict: Dict[str, Any]) -> AiAgentPersistenceConfiguration:
         instance = cls()
         instance.conversation_id_prefix = json_dict.get("conversationIdPrefix") or json_dict.get("ConversationIdPrefix")
-        instance.conversation_expiration_in_sec = json_dict.get("conversationExpirationInSec") or json_dict.get("ConversationExpirationInSec")
+        instance.conversation_expiration_in_sec = json_dict.get("conversationExpirationInSec") or json_dict.get(
+            "ConversationExpirationInSec"
+        )
         return instance
 
 
@@ -121,7 +127,9 @@ def from_json(cls, json_dict: Dict[str, Any]) -> AiAgentSummarizationByTokens:
         instance.summarization_task_beginning_prompt = json_dict.get("SummarizationTaskBeginningPrompt")
         instance.summarization_task_end_prompt = json_dict.get("SummarizationTaskEndPrompt")
         instance.result_prefix = json_dict.get("ResultPrefix")
-        instance.max_tokens_before_summarization = json_dict.get("MaxTokensBeforeSummarization", cls.DEFAULT_MAX_TOKENS_BEFORE_SUMMARIZATION)
+        instance.max_tokens_before_summarization = json_dict.get(
+            "MaxTokensBeforeSummarization", cls.DEFAULT_MAX_TOKENS_BEFORE_SUMMARIZATION
+        )
         instance.max_tokens_after_summarization = json_dict.get("MaxTokensAfterSummarization", 1024)
         return instance
 
@@ -146,8 +154,12 @@ def to_json(self) -> Dict[str, Any]:
     @classmethod
     def from_json(cls, json_dict: Dict[str, Any]) -> AiAgentTruncateChat:
         instance = cls()
-        instance.messages_length_before_truncate = json_dict.get("MessagesLengthBeforeTruncate", cls.DEFAULT_MESSAGES_LENGTH_BEFORE_TRUNCATE)
-        instance.messages_length_after_truncate = json_dict.get("MessagesLengthAfterTruncate", cls.DEFAULT_MESSAGES_LENGTH_BEFORE_TRUNCATE // 2)
+        instance.messages_length_before_truncate = json_dict.get(
+            "MessagesLengthBeforeTruncate", cls.DEFAULT_MESSAGES_LENGTH_BEFORE_TRUNCATE
+        )
+        instance.messages_length_after_truncate = json_dict.get(
+            "MessagesLengthAfterTruncate", cls.DEFAULT_MESSAGES_LENGTH_BEFORE_TRUNCATE // 2
+        )
         return instance
 
 
@@ -227,10 +239,7 @@ def __init__(self, name: str = None, connection_string_name: str = None, system_
 
     def to_json(self) -> Dict[str, Any]:
         # Convert parameters set to list of parameter objects using list comprehension
-        parameters_list = [
-            {"Name": param_name, "Description": None}
-            for param_name in self.parameters
-        ]
+        parameters_list = [{"Name": param_name, "Description": None} for param_name in self.parameters]
 
         return {
             "Identifier": self.identifier,
@@ -284,5 +293,7 @@ def from_json(cls, json_dict: Dict[str, Any]) -> AiAgentConfiguration:
         if trimming_data:
             instance.chat_trimming = AiAgentChatTrimmingConfiguration.from_json(trimming_data)
 
-        instance.max_model_iterations_per_call = json_dict.get("maxModelIterationsPerCall") or json_dict.get("MaxModelIterationsPerCall")
+        instance.max_model_iterations_per_call = json_dict.get("maxModelIterationsPerCall") or json_dict.get(
+            "MaxModelIterationsPerCall"
+        )
         return instance

ravendb/documents/operations/ai/agents/delete_ai_agent_operation.py

@@ -46,4 +46,5 @@ def set_response(self, response: str, from_cache: bool) -> None:
     def get_raft_unique_request_id(self) -> str:
         # Generate a unique ID for Raft operations
         import uuid
+
         return str(uuid.uuid4())

ravendb/documents/operations/ai/agents/get_ai_agent_operation.py

@@ -20,13 +20,10 @@ def __init__(self):
     @classmethod
     def from_json(cls, json_dict: Dict[str, Any]) -> GetAiAgentsResponse:
         from ravendb.documents.operations.ai.agents.ai_agent_configuration import AiAgentConfiguration
-        
+
         response = cls()
         if json_dict.get("AiAgents"):
-            response.ai_agents = [
-                AiAgentConfiguration.from_json(agent_json) 
-                for agent_json in json_dict["AiAgents"]
-            ]
+            response.ai_agents = [AiAgentConfiguration.from_json(agent_json) for agent_json in json_dict["AiAgents"]]
         return response