From e9abb6ea8180da3e2af3b3d1a75e7246858a4717 Mon Sep 17 00:00:00 2001 From: mohammadrezapourreza Date: Wed, 11 Oct 2023 16:11:52 -0400 Subject: [PATCH] DH-4627/reformat with black --- dataherald/model/chat_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dataherald/model/chat_model.py b/dataherald/model/chat_model.py index d2333334..5d637f68 100644 --- 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: