Spaces:
Sleeping
Sleeping
import smolagents, numpy, math, xlrd, os | |
from smolagents import ( | |
CodeAgent, | |
HfApiModel, | |
InferenceClientModel, | |
WebSearchTool, | |
PythonInterpreterTool, | |
FinalAnswerTool, | |
DuckDuckGoSearchTool, | |
GoogleSearchTool, | |
) | |
#model_id = "Qwen/Qwen2.5-Coder-32B-Instruct" | |
#model_id = 'meta-llama/Llama-3.3-70B-Instruct' | |
#model = HfApiModel(model_id=model_id, token="HUGGINGFACEHUB_API_TOKEN") | |
#agent = CodeAgent(tools=[], model=model, add_base_tools=True) | |
#max_steps=5 | |
#import os | |
#from smolagents import CodeAgent, HfApiModel, InferenceClientModel, WebSearchTool | |
class newAgent: | |
"""Adapts smolagents.CodeAgent to the HF course template API.""" | |
def __init__(self): | |
model_id = "meta-llama/Meta-Llama-3-70B-Instruct" # correct repo name | |
hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN") # read real secret | |
if not hf_token: | |
raise RuntimeError("HUGGINGFACEHUB_API_TOKEN not set in Space secrets") | |
model = HfApiModel(model_id=model_id, token=hf_token) | |
# add_base_tools=True already gives you search, python, etc. | |
self.agent = CodeAgent(tools=[], model=model, add_base_tools=True) | |
#* | |
# include FinalAnswerTool in tools so agent knows when to stop | |
tools = [FinalAnswerTool()] | |
self.agent = CodeAgent( | |
tools=tools, | |
model=model, | |
add_base_tools=True | |
) | |
#* | |
def __call__(self, question: str) -> str: | |
"""ONE question in β ONE pure-text answer out.""" | |
#β Replace .run with whatever method actually returns the answer string. | |
result = self.agent.run(question) | |
return answer | |
#answer = self.run | |
#agent.run( | |
# "At what temperature and for how long should I bake French baguettes made with type 65 flour?", | |
#) |