RDBC-935 black

Author: yh-0

ravendb/documents/operations/ai/abstract_ai_settings.py

@@ -13,4 +13,4 @@ def from_json(cls, json_dict: Dict[str, Any]) -> Any:
 
     @abstractmethod
     def to_json(self) -> Dict[str, Any]:
-        pass
+        pass

ravendb/documents/operations/ai/ai_connection_string.py

@@ -13,11 +13,11 @@
 
 
 class AiModelType(enum.Enum):
-    TEXT_EMBEDDINGS = "TextEmbeddings",
+    TEXT_EMBEDDINGS = ("TextEmbeddings",)
     CHAT = "Chat"
 
 
-class AiConnectionString(ConnectionString): # todo kuba
+class AiConnectionString(ConnectionString):  # todo kuba
     def __init__(
         self,
         name: str,
@@ -29,7 +29,7 @@ def __init__(
         google_settings: Optional[GoogleSettings] = None,
         huggingface_settings: Optional[HuggingFaceSettings] = None,
         mistral_ai_settings: Optional[MistralAiSettings] = None,
-        model_type: AiModelType = None
+        model_type: AiModelType = None,
     ):
         super().__init__(name)
         self.identifier = identifier

ravendb/documents/operations/ai/azure_open_ai_settings.py

@@ -4,7 +4,15 @@
 
 
 class AzureOpenAiSettings(OpenAiBaseSettings):
-    def __init__(self, api_key: str = None, endpoint: str = None, model: str = None, deployment_name: str = None, dimensions: int = None, temperature: float = None):
+    def __init__(
+        self,
+        api_key: str = None,
+        endpoint: str = None,
+        model: str = None,
+        deployment_name: str = None,
+        dimensions: int = None,
+        temperature: float = None,
+    ):
         super().__init__(api_key, endpoint, model, dimensions, temperature)
         self.deployment_name = deployment_name
 
@@ -28,4 +36,4 @@ def to_json(self) -> Dict[str, Any]:
             "Temperature": self.temperature,
             "DeploymentName": self.deployment_name,
             "EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches,
-        }
+        }

ravendb/documents/operations/ai/embedded_settings.py

@@ -12,4 +12,4 @@ def from_json(cls, json_dict: Dict[str, Any]) -> "EmbeddedSettings":
         return cls()
 
     def to_json(self) -> Dict[str, Any]:
-        return {"EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches}
+        return {"EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches}

ravendb/documents/operations/ai/google_settings.py

@@ -5,12 +5,14 @@
 
 
 class GoogleAiVersion(Enum):
-    V1 = "V1",
-    V1_Beta = "V1_Beta",
+    V1 = ("V1",)
+    V1_Beta = ("V1_Beta",)
 
 
 class GoogleSettings(AbstractAiSettings):
-    def __init__(self, model: str = None, api_key: str = None, ai_version: GoogleAiVersion = None, dimensions: int = None):
+    def __init__(
+        self, model: str = None, api_key: str = None, ai_version: GoogleAiVersion = None, dimensions: int = None
+    ):
         super().__init__()
         self.model = model
         self.api_key = api_key

ravendb/documents/operations/ai/hugging_face_settings.py

@@ -24,4 +24,4 @@ def to_json(self) -> Dict[str, Any]:
             "Model": self.model,
             "Endpoint": self.endpoint,
             "EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches,
-        }
+        }

ravendb/documents/operations/ai/mistral_ai_settings.py

@@ -24,4 +24,4 @@ def to_json(self) -> Dict[str, Any]:
             "Model": self.model,
             "Endpoint": self.endpoint,
             "EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches,
-        }
+        }

ravendb/documents/operations/ai/ollama_settings.py

@@ -25,4 +25,4 @@ def to_json(self) -> Dict[str, Any]:
             "Think": self.think,
             "Temperature": self.temperature,
             "EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches,
-        }
+        }

ravendb/documents/operations/ai/open_ai_base_settings.py

@@ -4,7 +4,14 @@
 
 
 class OpenAiBaseSettings(AbstractAiSettings, ABC):
-    def __init__(self, api_key: str = None, endpoint: str = None, model: str = None, dimensions: int = None, temperature: float = None):
+    def __init__(
+        self,
+        api_key: str = None,
+        endpoint: str = None,
+        model: str = None,
+        dimensions: int = None,
+        temperature: float = None,
+    ):
         super().__init__()
         self.api_key = api_key
         self.endpoint = endpoint

ravendb/documents/operations/ai/open_ai_settings.py

@@ -4,7 +4,16 @@
 
 
 class OpenAiSettings(OpenAiBaseSettings):
-    def __init__(self, api_key: str = None, endpoint: str = None, model: str = None, organization_id: str = None, project_id: str = None, dimensions: int = None, temperature: float = None):
+    def __init__(
+        self,
+        api_key: str = None,
+        endpoint: str = None,
+        model: str = None,
+        organization_id: str = None,
+        project_id: str = None,
+        dimensions: int = None,
+        temperature: float = None,
+    ):
         super().__init__(api_key, endpoint, model, dimensions, temperature)
         self.organization_id = organization_id
         self.project_id = project_id
@@ -31,4 +40,4 @@ def to_json(self) -> Dict[str, Any]:
             "OrganizationId": self.organization_id,
             "ProjectId": self.project_id,
             "EmbeddingsMaxConcurrentBatches": self.embeddings_max_concurrent_batches,
-        }
+        }