Refactor cohere api key

weni-ai · Apr 7, 2024 · 3cb128a · 3cb128a
1 parent 113e807
commit 3cb128a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/config.py b/app/config.py
@@ -22,7 +22,7 @@ def __init__(self):
             "model": os.environ.get(
                 "COHERE_MODEL", "embed-multilingual-v3.0"
             ),
-            "cohere_api_token": os.environ.get(
+            "cohere_api_key": os.environ.get(
                 "COHERE_API_KEY", "f9BfpzFleAcsa4HueXrCSWyMxJEOuMcpbesB2Rxo"
             ),
         }

diff --git a/app/main.py b/app/main.py
@@ -40,7 +40,7 @@ def __init__(self, config: AppConfig):
         elif config.embedding_type == "cohere":
             self.embeddings = CohereEmbeddings(
                 model=config.cohere["model"],
-                cohere_api_key=config.cohere["cohere_api_token"]
+                cohere_api_key=config.cohere["cohere_api_key"]
             )
         else:  # sagemaker by default
             content_handler = ContentHandler()