Refactor: use SecretStr for minimax embeddings (#15067)

pull/15080/head
chyroc 9 months ago committed by GitHub
parent aa19ca9723
commit 4440ec5ab3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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",
}

Loading…
Cancel
Save