Add book_search tool

app.py

@@ -33,6 +33,42 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def book_search(title: str, language: str) -> str:
+    """A tool that searches for a book based on title or user query and language.
+    Args:
+        title: A user query or string representing the book title or query.
+        language: An optional 2 char string representing the language code. e.g: en, fr, ta
+    """
+    try:
+        # Construct the URL with the given parameters
+        base_url = "https://openlibrary.org/search.json"
+        params = {
+            "title": title,
+            "fields": "title,author_name",
+            "limit": 10
+        }
+        if language:
+            params["lang"] = language
+        
+        # Make the request
+        response = requests.get(base_url, params=params)
+        response.raise_for_status()
+        
+        # Parse the response
+        data = response.json()
+        
+        # Format results
+        results = []
+        for book in data.get("docs", []):
+            book_title = book.get("title", "Unknown Title")
+            authors = ", ".join(book.get("author_name", ["Unknown Author"]))
+            results.append(f"Title: *{book_title}*, Author(s): _{authors}_")
+            
+        return "\n".join(results) if results else "No books found."
+    except Exception as e:
+        return f"Error fetching books: {str(e)}"
+
 
 final_answer = FinalAnswerTool()
 
@@ -42,7 +78,8 @@ 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',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
@@ -55,14 +92,14 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, book_search, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
+    name="Book Search Agent",
+    description="An agent that searches for books based on user queries and language.",
+    prompt_templates=prompt_templates,
 )