|
|
from typing import List, Dict |
|
|
import requests |
|
|
from smolagents.tools import Tool |
|
|
|
|
|
class BookSearchTool(Tool): |
|
|
name = "search_books" |
|
|
description = "Search for books using the Open Library API" |
|
|
inputs = {'query': {'type': 'string', 'description': 'The search query to find books.'}} |
|
|
output_type = "string" |
|
|
|
|
|
def forward(self, query: str) -> str: |
|
|
try: |
|
|
url = f"https://openlibrary.org/search.json?q={query}&limit=5" |
|
|
response = requests.get(url) |
|
|
response.raise_for_status() |
|
|
data = response.json() |
|
|
|
|
|
results = [] |
|
|
for book in data.get('docs', [])[:5]: |
|
|
result = { |
|
|
'title': book.get('title', 'Unknown Title'), |
|
|
'author': book.get('author_name', ['Unknown Author'])[0] if book.get('author_name') else 'Unknown Author', |
|
|
'first_publish_year': book.get('first_publish_year', 'Unknown Year'), |
|
|
'subject': book.get('subject', [])[:3] if book.get('subject') else [], |
|
|
'key': book.get('key', '') |
|
|
} |
|
|
results.append(result) |
|
|
|
|
|
|
|
|
formatted_results = "## Book Search Results\n\n" |
|
|
for book in results: |
|
|
formatted_results += f"### {book['title']}\n" |
|
|
formatted_results += f"- Author: {book['author']}\n" |
|
|
formatted_results += f"- Year: {book['first_publish_year']}\n" |
|
|
formatted_results += f"- Subjects: {', '.join(book['subject'])}\n\n" |
|
|
|
|
|
return formatted_results |
|
|
|
|
|
except Exception as e: |
|
|
return f"Error searching books: {str(e)}" |
|
|
|
|
|
class BookDetailsTool(Tool): |
|
|
name = "get_book_details" |
|
|
description = "Get detailed information about a specific book from Open Library" |
|
|
inputs = {'book_key': {'type': 'string', 'description': 'The Open Library key for the book.'}} |
|
|
output_type = "string" |
|
|
|
|
|
def forward(self, book_key: str) -> str: |
|
|
try: |
|
|
url = f"https://openlibrary.org{book_key}.json" |
|
|
response = requests.get(url) |
|
|
response.raise_for_status() |
|
|
data = response.json() |
|
|
|
|
|
description = data.get('description', {}).get('value', 'No description available') \ |
|
|
if isinstance(data.get('description'), dict) \ |
|
|
else data.get('description', 'No description available') |
|
|
|
|
|
subjects = data.get('subjects', [])[:5] |
|
|
first_sentence = data.get('first_sentence', {}).get('value', 'Not available') \ |
|
|
if isinstance(data.get('first_sentence'), dict) \ |
|
|
else data.get('first_sentence', 'Not available') |
|
|
|
|
|
|
|
|
details = f"## {data.get('title', 'Unknown Title')}\n\n" |
|
|
details += "### Description\n" |
|
|
details += f"{description}\n\n" |
|
|
details += "### Subjects\n" |
|
|
details += ', '.join(subjects) + "\n\n" |
|
|
details += "### First Sentence\n" |
|
|
details += f"{first_sentence}\n" |
|
|
|
|
|
return details |
|
|
|
|
|
except Exception as e: |
|
|
return f"Error fetching book details: {str(e)}" |
