Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-16 06:13:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: use SecretStr for minimax embeddings (#15067)

Browse Source
This commit is contained in:
chyroc 2023-12-23 03:43:23 +08:00 committed by GitHub
parent aa19ca9723
commit 4440ec5ab3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/community/langchain_community/embeddings/minimax.py
View File

@ -5,8 +5,8 @@ from typing import Any, Callable, Dict, List, Optional
import requests
from langchain_core.embeddings import Embeddings
from langchain_core.pydantic_v1 import BaseModel, Extra, root_validator
from langchain_core.utils import get_from_dict_or_env
from langchain_core.pydantic_v1 import BaseModel, Extra, SecretStr, root_validator
from langchain_core.utils import convert_to_secret_str, get_from_dict_or_env
from tenacity import (
before_sleep_log,
retry,
@ -76,7 +76,7 @@ class MiniMaxEmbeddings(BaseModel, Embeddings):
minimax_group_id: Optional[str] = None
"""Group ID for MiniMax API."""
minimax_api_key: Optional[str] = None
minimax_api_key: Optional[SecretStr] = None
"""API Key for MiniMax API."""
class Config:
@ -90,8 +90,8 @@ class MiniMaxEmbeddings(BaseModel, Embeddings):
minimax_group_id = get_from_dict_or_env(
values, "minimax_group_id", "MINIMAX_GROUP_ID"
)
minimax_api_key = get_from_dict_or_env(
values, "minimax_api_key", "MINIMAX_API_KEY"
minimax_api_key = convert_to_secret_str(
get_from_dict_or_env(values, "minimax_api_key", "MINIMAX_API_KEY")
)
values["minimax_group_id"] = minimax_group_id
values["minimax_api_key"] = minimax_api_key
@ -110,7 +110,7 @@ class MiniMaxEmbeddings(BaseModel, Embeddings):
# HTTP headers for authorization
headers = {
"Authorization": f"Bearer {self.minimax_api_key}",
"Authorization": f"Bearer {self.minimax_api_key.get_secret_value()}",
"Content-Type": "application/json",
}