fix: Allow cohere_api_key to be SecretStr in CohereEmbedding (#102)

Resolves issue #101
langchain-ai · Oct 14, 2024 · 27262d8 · 27262d8
1 parent 9a9681d
commit 27262d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/libs/cohere/langchain_cohere/embeddings.py b/libs/cohere/langchain_cohere/embeddings.py
@@ -78,6 +78,9 @@ def validate_environment(cls, values: Dict) -> Any:
         cohere_api_key = get_from_dict_or_env(
             values, "cohere_api_key", "COHERE_API_KEY"
         )
+        if isinstance(cohere_api_key, SecretStr):
+            cohere_api_key = cohere_api_key.get_secret_value()
+
         request_timeout = values.get("request_timeout")
 
         client_name = values.get("user_agent", "langchain:partner")

diff --git a/libs/cohere/pyproject.toml b/libs/cohere/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-cohere"
-version = "0.3.0"
+version = "0.3.1"
 description = "An integration package connecting Cohere and LangChain"
 authors = []
 readme = "README.md"