work on nyt tool

nyt_tool.ipynb

@@ -0,0 +1,16 @@
+{
+ "cells": [],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "agents_env",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "name": "python",
+   "version": "3.10.12"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

tools/book_search.py

@@ -0,0 +1,25 @@
+import os
+from typing import Any, Optional
+from smolagents.tools import Tool
+import duckduckgo_search
+
+class NYTSearchTool(Tool):
+    name = "best_sellers_tool"
+    description = (
+        "Performs a search of the New York Times best seller list " 
+        "for a specific genre and returns the best selling books"
+        )
+    
+    inputs = {'genre': {'type': 'string', 'description': 'The genre to search'}}
+    output_type = "string"
+
+    def __init__(self, max_results=10, **kwargs):
+        super().__init__()
+        self.api_prefix = "https://api.nytimes.com/svc/books/v3/lists/current"
+
+    def forward(self, query: str) -> str:
+        results = self.ddgs.text(query, max_results=self.max_results)
+        if len(results) == 0:
+            raise Exception("No results found! Try a less restrictive/shorter query.")
+        postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
+        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)