test an API lookup tool

app.py

@@ -1,4 +1,5 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from pydantic import BaseModel
 import datetime
 import requests
 import pytz
@@ -30,15 +31,39 @@ trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
 
 SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
 
+GAME_API_KEY = os.getenv("GAME_API_KEY")
+GAME_API_BASE_URL = os.getenv("GAME_API_BASE_URL")
+
+class Game_Response(BaseModel):
+    name: str
+    description: str
+    year_published: int
+    min_players: int
+    max_players: int
+    users_rated: int
+    average_rating: float
+    bayes_adjusted_average: float
+    complexity: float
+
 @tool
-def multiplication_tool(arg1:int, arg2:int)-> int:
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that multiplies two integers together
+def boardgame_lookup_tool(q:str)-> list[Game_Response]:
+    """A tool that fetches information about board games
     Args:
-        arg1: the first integers
-        arg2: the second integers
+        q: a search term representing part or all of a board game's name or description, prefixed with either "name:" or "description:"
     """
-    return arg1*arg2
+    url = f'{GAME_API_BASE_URL}/games'
+    headers = {'x-api-key': f'{GAME_API_KEY}'}
+    response = requests.get(url, headers=headers, params={'q': q, 'search_type': search_type})
+    api_data_list = response.json()
+    return [Game_Response(name = api_data['name'],
+                         description = api_data['description'],
+                         year_published = api_data['year_published'],
+                         min_players = api_data['min_players'],
+                         max_players = api_data['max_players'],
+                         users_rated = api_data['users_rated'],
+                         average_rating = api_data['average'],
+                         bayes_adjusted_average = api_data['bayes_average'],
+                         complexity = api_data['weight']) for api_data in api_data_list]
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -64,22 +89,23 @@ final_answer = FinalAnswerTool()
 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
+#model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
 custom_role_conversions=None,
 )
 
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+#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,multiplication_tool,image_generation_tool],
+    tools=[final_answer,boardgame_lookup_tool],
     max_steps=6,
-#    verbosity_level=1,
+    verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,

requirements.txt

@@ -3,6 +3,7 @@ smolagents
 requests
 duckduckgo_search
 pandas
+pydantic
 
 opentelemetry-sdk
 opentelemetry-exporter-otlp