github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-22 02:50:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Community[patch]use secret str in Tavily and HuggingFaceInferenceEmbeddings (#16109)

Browse Source 
So the api keys don't show up in repr's 

Still need to do tests
This commit is contained in:
William FH
2024-01-17 00:30:07 -08:00
committed by GitHub
parent f3601b0aaf
commit e5cf1e2414
5 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/community/langchain_community/utilities/tavily_search.py
View File

@@ -7,7 +7,7 @@ from typing import Dict, List, Optional
import aiohttp
import requests
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
TAVILY_API_URL = "https://api.tavily.com"
@@ -16,7 +16,7 @@ TAVILY_API_URL = "https://api.tavily.com"
class TavilySearchAPIWrapper(BaseModel):
"""Wrapper for Tavily Search API."""
tavily_api_key: str
tavily_api_key: SecretStr
class Config:
"""Configuration for this pydantic object."""
@@ -45,7 +45,7 @@ class TavilySearchAPIWrapper(BaseModel):
include_images: Optional[bool] = False,
) -> Dict:
params = {
"api_key": self.tavily_api_key,
"api_key": self.tavily_api_key.get_secret_value(),
"query": query,
"max_results": max_results,
"search_depth": search_depth,
@@ -126,7 +126,7 @@ class TavilySearchAPIWrapper(BaseModel):
# Function to perform the API call
async def fetch() -> str:
params = {
"api_key": self.tavily_api_key,
"api_key": self.tavily_api_key.get_secret_value(),
"query": query,
"max_results": max_results,
"search_depth": search_depth,