File size: 2,427 Bytes
1aa16a4 9e58d03 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from smolagents import tool, Tool
from tavily import TavilyClient
from smolagents import DuckDuckGoSearchTool, GoogleSearchTool, VisitWebpageTool, WikipediaSearchTool, PythonInterpreterTool, FinalAnswerTool
from dotenv import load_dotenv
import os
load_dotenv() # take environment variables
# init tools
duck_search = DuckDuckGoSearchTool()
google_search = GoogleSearchTool()
visit_page = VisitWebpageTool()
wiki_search = WikipediaSearchTool()
do_python = PythonInterpreterTool()
final_answer = FinalAnswerTool()
tavily_search = TavilyClient()
speech_to_text_tool = Tool.from_space("hf-audio/whisper-large-v3-turbo",
name="speech_to_text_tool",
description="""Converts audio to text by providing a file or url.
Use the with the command speed_to_text_tool(filename).
Example: 'speech_to_text_tool("cache\\audio.mp3'""",
api_name="/predict")
visual_qa_tool = Tool.from_space("sitammeur/PicQ",
name="visual_qa_tool",
description="""Can answer question about a provided image.
Use it with visual_qa_tool(question=<question>, image=<image filename>).
Example visual_qa_tool(question='How many items are in the image?', image='cache\\image.png').""",
api_name="/predict")
# image_generation_tool = Tool.from_space(
# "black-forest-labs/FLUX.1-schnell",
# name="image_generator",
# description="Generate an image from a prompt"
# )
# image = image_generation_tool("A sunny beach")
# print(image)
## Testing the tools
# print(duck_search.forward("smolagents")) # connect error, does not take proxy
# print(google_search.forward("smolagents"))
# print(visit_page.forward("https://www.wikipedia.de"))
# print(wiki_search.forward("Python_(programming_language)"))
# print(do_python('print("hello world")'))
# print(tavily_search.search("smolagents"))
# print(speech_to_text_tool("cache\\1f975693-876d-457b-a649-393859e79bf3.mp3"))
# print(visual_qa_tool(question="Name all black and white figures with each coordinates, e. g. black king on g8", image="cache\\cca530fc-4052-43b2-b130-b30968d8aa44.png")) |