github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-30 10:23:30 +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 import requests
from langchain_core.embeddings import Embeddings from langchain_core.embeddings import Embeddings
from langchain_core.pydantic_v1 import BaseModel, Extra, root_validator from langchain_core.pydantic_v1 import BaseModel, Extra, SecretStr, root_validator
from langchain_core.utils import get_from_dict_or_env from langchain_core.utils import convert_to_secret_str, get_from_dict_or_env
from tenacity import ( from tenacity import (
before_sleep_log, before_sleep_log,
retry, retry,
@ -76,7 +76,7 @@ class MiniMaxEmbeddings(BaseModel, Embeddings):
minimax_group_id: Optional[str] = None minimax_group_id: Optional[str] = None
"""Group ID for MiniMax API.""" """Group ID for MiniMax API."""
minimax_api_key: Optional[str] = None minimax_api_key: Optional[SecretStr] = None
"""API Key for MiniMax API.""" """API Key for MiniMax API."""
class Config: class Config:
@ -90,8 +90,8 @@ class MiniMaxEmbeddings(BaseModel, Embeddings):
minimax_group_id = get_from_dict_or_env( minimax_group_id = get_from_dict_or_env(
values, "minimax_group_id", "MINIMAX_GROUP_ID" values, "minimax_group_id", "MINIMAX_GROUP_ID"
) )
minimax_api_key = get_from_dict_or_env( minimax_api_key = convert_to_secret_str(
values, "minimax_api_key", "MINIMAX_API_KEY" get_from_dict_or_env(values, "minimax_api_key", "MINIMAX_API_KEY")
) )
values["minimax_group_id"] = minimax_group_id values["minimax_group_id"] = minimax_group_id
values["minimax_api_key"] = minimax_api_key values["minimax_api_key"] = minimax_api_key
@ -110,7 +110,7 @@ class MiniMaxEmbeddings(BaseModel, Embeddings):
# HTTP headers for authorization # HTTP headers for authorization
headers = { headers = {
"Authorization": f"Bearer {self.minimax_api_key}", "Authorization": f"Bearer {self.minimax_api_key.get_secret_value()}",
"Content-Type": "application/json", "Content-Type": "application/json",
} }