DH-4627/reformat with black

Dataherald · Oct 11, 2023 · e9abb6e · e9abb6e
1 parent 8540da8
commit e9abb6e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dataherald/model/chat_model.py b/dataherald/model/chat_model.py
@@ -27,13 +27,13 @@ def get_model(
                 database_connection.llm_credentials.api_key
             )
             if model_family == "openai":
-                os.environ['OPENAI_API_KEY'] = api_key
+                os.environ["OPENAI_API_KEY"] = api_key
             elif model_family == "anthropic":
-                os.environ['ANTHROPIC_API_KEY'] = api_key
+                os.environ["ANTHROPIC_API_KEY"] = api_key
             elif model_family == "google":
-                os.environ['GOOGLE_API_KEY'] = api_key
+                os.environ["GOOGLE_API_KEY"] = api_key
             elif model_family == "cohere":
-                os.environ['COHERE_API_KEY'] = api_key
+                os.environ["COHERE_API_KEY"] = api_key
         try:
             return ChatLiteLLM(model_name=self.model_name, **kwargs)
         except Exception as e: