From fbcc302c7552566e2ae2946cdd13b87e59c7d05a Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 30 Jun 2023 09:14:30 -0400 Subject: [PATCH] clean up --- .env.example | 6 +++++- swarms/tools/main.py | 21 +++++++++------------ 2 files changed, 14 insertions(+), 13 deletions(-) diff --git a/.env.example b/.env.example index 10cbf1ba..4661378b 100644 --- a/.env.example +++ b/.env.example @@ -20,4 +20,8 @@ BING_SEARCH_URL="your_bing_search_url_here" BING_SUBSCRIPTION_KEY="your_bing_subscription_key_here" SERPAPI_API_KEY="your_serpapi_api_key_here" -IFTTTKey="" \ No newline at end of file +IFTTTKey="" + +BRAVE_API_KEY="" +SPOONACULAR_KEY="" +HF_API_KEY="" \ No newline at end of file diff --git a/swarms/tools/main.py b/swarms/tools/main.py index 77d10013..a244b691 100644 --- a/swarms/tools/main.py +++ b/swarms/tools/main.py @@ -1781,14 +1781,12 @@ list_tool.run({}) 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 @@ -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" ) -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}) spoonacular_toolkit = NLAToolkit.from_llm_and_url( @@ -2136,11 +2135,7 @@ ruff_vectorstore_info = VectorStoreInfo( router_toolkit = VectorStoreRouterToolkit( 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 +hf_api_key = os.environ["HF_API_KEY"] # define a custom input schema for the youtube url class YouTubeVideoInput(BaseModel): 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) # 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) try: @@ -2243,7 +2240,7 @@ class TranscribeAudioTool(BaseTool): 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) - 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) try: