Fix environment variable setting in embed_text component (#722)

Set the environment key before calling the embedding model to avoid "no api key found" error.
ml6team · Dec 12, 2023 · bc7bc98 · bc7bc98
1 parent 6780012
commit bc7bc98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/components/embed_text/src/main.py b/components/embed_text/src/main.py
@@ -32,14 +32,14 @@ def __init__(
         auth_kwargs: dict,
         **kwargs,
     ):
+        to_env_vars(api_keys)
+
         self.embedding_model = self.get_embedding_model(
             model_provider,
             model,
             auth_kwargs,
         )
 
-        to_env_vars(api_keys)
-
     @staticmethod
     def get_embedding_model(
         model_provider,