main
Kye 2 years ago
parent 875dc67eb3
commit fbcc302c75

@ -20,4 +20,8 @@ BING_SEARCH_URL="your_bing_search_url_here"
BING_SUBSCRIPTION_KEY="your_bing_subscription_key_here" BING_SUBSCRIPTION_KEY="your_bing_subscription_key_here"
SERPAPI_API_KEY="your_serpapi_api_key_here" SERPAPI_API_KEY="your_serpapi_api_key_here"
IFTTTKey="" IFTTTKey=""
BRAVE_API_KEY=""
SPOONACULAR_KEY=""
HF_API_KEY=""

@ -1781,14 +1781,12 @@ list_tool.run({})
from langchain.tools import BraveSearch from langchain.tools import BraveSearch
brave_api_key = os.environ["BRAVE_API_KEY"]
api_key = "..." brave_tool = BraveSearch.from_api_key(api_key=brave_api_key, search_kwargs={"count": 3})
brave_tool = BraveSearch.from_api_key(api_key=api_key, search_kwargs={"count": 3})
tool.run("obama middle name")
######################### BRAVE END ######################### BRAVE END
@ -2023,7 +2021,8 @@ mrkl.run(
"I have an end of year party for my Italian class and have to buy some Italian clothes for it" "I have an end of year party for my Italian class and have to buy some Italian clothes for it"
) )
spoonacular_api_key = "" # Copy from the API Console spoonacular_api = os.environ["SPOONACULAR_KEY"]
spoonacular_api_key = spoonacular_api
requests = Requests(headers={"x-api-key": spoonacular_api_key}) requests = Requests(headers={"x-api-key": spoonacular_api_key})
spoonacular_toolkit = NLAToolkit.from_llm_and_url( spoonacular_toolkit = NLAToolkit.from_llm_and_url(
@ -2136,11 +2135,7 @@ ruff_vectorstore_info = VectorStoreInfo(
router_toolkit = VectorStoreRouterToolkit( router_toolkit = VectorStoreRouterToolkit(
vectorstores=[vectorstore_info, ruff_vectorstore_info], llm=llm vectorstores=[vectorstore_info, ruff_vectorstore_info], llm=llm
) )
agent_executor = create_vectorstore_router_agent( #
llm=llm, toolkit=router_toolkit, verbose=True
)
@ -2155,6 +2150,7 @@ import whisperx
from langchain.tools import tool from langchain.tools import tool
hf_api_key = os.environ["HF_API_KEY"]
# define a custom input schema for the youtube url # define a custom input schema for the youtube url
class YouTubeVideoInput(BaseModel): class YouTubeVideoInput(BaseModel):
video_url: str = Field(description="YouTube Video URL to transcribe") video_url: str = Field(description="YouTube Video URL to transcribe")
@ -2195,7 +2191,8 @@ def transcribe_youtube_video(video_url: str) -> str:
result = whisperx.align(result["segments"], model_a, metadata, audio, device, return_char_alignments=False) result = whisperx.align(result["segments"], model_a, metadata, audio, device, return_char_alignments=False)
# 3. Assign speaker labels # 3. Assign speaker labels
diarize_model = whisperx.DiarizationPipeline(use_auth_token='hugging face stable api key', device=device)
diarize_model = whisperx.DiarizationPipeline(use_auth_token=hf_api_key, device=device)
diarize_segments = diarize_model(audio_file) diarize_segments = diarize_model(audio_file)
try: try:
@ -2243,7 +2240,7 @@ class TranscribeAudioTool(BaseTool):
model_a, metadata = whisperx.load_align_model(language_code=result["language"], device=device) model_a, metadata = whisperx.load_align_model(language_code=result["language"], device=device)
result = whisperx.align(result["segments"], model_a, metadata, audio, device, return_char_alignments=False) result = whisperx.align(result["segments"], model_a, metadata, audio, device, return_char_alignments=False)
diarize_model = whisperx.DiarizationPipeline(use_auth_token='hugging face stable api key', device=device) diarize_model = whisperx.DiarizationPipeline(use_auth_token=hf_api_key, device=device)
diarize_segments = diarize_model(audio_file) diarize_segments = diarize_model(audio_file)
try: try:

Loading…
Cancel
Save