File size: 1,714 Bytes
9b5b26a c19d193 6aae614 f31224c 8fe992b 9b5b26a f31224c 9b5b26a f31224c 9b5b26a f31224c 9b5b26a f31224c 9b5b26a f31224c 8c01ffb 6aae614 ae7a494 e121372 bf6d34c 29ec968 fe328e0 13d500a 8c01ffb 9b5b26a 8c01ffb 8fe992b f31224c 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 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
import wikipedia
from Gradio_UI import GradioUI
# Below is aa tool that searches wikipedia
@tool
def wikipedia_search(query: str, sentences: int = 2) -> str:
"""Search Wikipedia and return a short summary.
Args:
query: The search term for Wikipedia.
sentences: The number of sentences to return from the summary.
"""
try:
summary = wikipedia.summary(query, sentences=sentences)
return summary
except wikipedia.exceptions.DisambiguationError as e:
return f"Multiple results found: {', '.join(e.options[:5])}..."
except wikipedia.exceptions.PageError:
return "No Wikipedia page found for that query."
except Exception as e:
return f"An error occurred: {str(e)}"
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,
)
agent = CodeAgent(
model=model,
tools=[final_answer, wikipedia_search], ## 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() |