Update tools/book_tool.py

tools/book_tool.py

@@ -6,9 +6,9 @@ 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 = List[Dict]
+    output_type = "string"
 
-    def forward(self, query: str) -> List[Dict]:
+    def forward(self, query: str) -> str:
         try:
             url = f"https://openlibrary.org/search.json?q={query}&limit=5"
             response = requests.get(url)
@@ -25,32 +25,52 @@ class BookSearchTool(Tool):
                     'key': book.get('key', '')
                 }
                 results.append(result)
-            return results
+
+            # Format results as a string
+            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 [{"error": f"Error searching books: {str(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 = Dict
+    output_type = "string"
 
-    def forward(self, book_key: str) -> Dict:
+    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()
             
-            return {
-                'title': data.get('title', 'Unknown Title'),
-                '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')
-            }
+            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')
+            
+            # Format as a string
+            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 {"error": f"Error fetching book details: {str(e)}"}
+            return f"Error fetching book details: {str(e)}"