Run formatter

autogenhub · Sep 30, 2024 · 279bfdb · 279bfdb
1 parent 5d7c06a
commit 279bfdb
Show file tree

Hide file tree

Showing 5 changed files with 948 additions and 943 deletions.
diff --git a/autogen/logger/sqlite_logger.py b/autogen/logger/sqlite_logger.py
@@ -25,8 +25,8 @@
 if TYPE_CHECKING:
     from autogen import Agent, ConversableAgent, OpenAIWrapper
     from autogen.oai.anthropic import AnthropicClient
-    from autogen.oai.cerebras import CerebrasClient
     from autogen.oai.bedrock import BedrockClient
+    from autogen.oai.cerebras import CerebrasClient
     from autogen.oai.cohere import CohereClient
     from autogen.oai.gemini import GeminiClient
     from autogen.oai.groq import GroqClient

diff --git a/autogen/oai/cerebras.py b/autogen/oai/cerebras.py
@@ -42,7 +42,7 @@
 class CerebrasClient:
     """Client for Cerebras's API."""
 
-    def __init__(self, api_key = None, **kwargs):
+    def __init__(self, api_key=None, **kwargs):
         """Requires api_key or environment variable to be set
 
         Args:
@@ -98,7 +98,9 @@ def parse_params(self, params: Dict[str, Any]) -> Dict[str, Any]:
         cerebras_params["max_tokens"] = validate_parameter(params, "max_tokens", int, True, None, (0, None), None)
         cerebras_params["seed"] = validate_parameter(params, "seed", int, True, None, None, None)
         cerebras_params["stream"] = validate_parameter(params, "stream", bool, True, False, None, None)
-        cerebras_params["temperature"] = validate_parameter(params, "temperature", (int, float), True, 1, (0, 1.5), None)
+        cerebras_params["temperature"] = validate_parameter(
+            params, "temperature", (int, float), True, 1, (0, 1.5), None
+        )
         cerebras_params["top_p"] = validate_parameter(params, "top_p", (int, float), True, None, None, None)
 
         return cerebras_params