Spaces:
Sleeping
Sleeping
File size: 2,519 Bytes
9b5b26a c19d193 6aae614 8fe992b 9b5b26a 853cae4 9b5b26a 853cae4 1211f2b 853cae4 9b5b26a 853cae4 8c01ffb 6aae614 ae7a494 e121372 bf6d34c a731eec fe328e0 13d500a 8c01ffb 9b5b26a 8c01ffb 861422e 9b5b26a 8c01ffb 8fe992b 853cae4 8c01ffb 861422e 8fe992b 9b5b26a 8c01ffb |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI
from smolagents.agent_types import AgentText
@tool
def lookup_definition(query: str) -> AgentText:
"""Fetches the definition of a word from the Dictionary API and returns it as AgentText.
Args:
query: The word to look up.
"""
url = f"https://api.dictionaryapi.dev/api/v2/entries/en/{query}"
try:
response = requests.get(url)
response.raise_for_status()
data = response.json()
if not data:
response_text = "No definition found."
else:
word = data[0].get("word", "Unknown word")
origin = data[0].get("origin", "Origin not available")
definitions = [
f"({meaning.get('partOfSpeech', 'Unknown')}) {definition['definition']}"
for meaning in data[0].get("meanings", [])
for definition in meaning.get("definitions", [])
]
if not definitions:
response_text = f"Word: {word}\nOrigin: {origin}\nNo definitions found."
else:
response_text = f"Word: {word}\nOrigin: {origin}\nDefinitions:\n- " + "\n- ".join(definitions)
except requests.RequestException as e:
response_text = f"Error fetching definition: {str(e)}"
return AgentText(response_text)
final_answer = FinalAnswerTool()
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id= 'Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)
# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
agent = CodeAgent(
model=model,
tools=[final_answer, lookup_definition], ## add your tools here (don't remove final answer)
max_steps=6,
verbosity_level=1,
grammar=None,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch() |