initialize_some

tools.py

@@ -13,7 +13,9 @@ from langchain_core.tools import tool
 from langchain_core.messages import SystemMessage, HumanMessage, AIMessage
 import os
 from typing import Optional, Tuple
-
+from huggingface_hub import list_models
+from pathlib import Path
+import json 
 
 
 def extract_image_text(img_path: str) -> str:
@@ -69,24 +71,87 @@ def extract_image_text(img_path: str) -> str:
         error_msg = f"Error extracting text: {str(e)}"
         print(error_msg)
         return ""
+    
+extract_image_text = Tool(
+    name="extract_image_text",
+    func=extract_image_text,
+    description="Extracts text that might be visible on an image by reading the image. Works with path/url to image."
+)
 
 
-def divide(a: int, b: int) -> float:
-    """Divide a and b."""
-    return a / b
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers.
+    Args:
+        a: first int
+        b: second int
+    """
+    return a * b
 
-def add(a: int, b: int) -> float:
-    """Add a and b."""
-    return a + b
+multiply = Tool(
+    name="multiply",
+    func=multiply,
+    description="Multiplies two integer numbers to get the product."
+)
 
-def subtract(a: int, b: int) -> float:
-    """Subtract a and b."""
+def add(a: int, b: int) -> int:
+    """Add two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    return a + b
+    
+add = Tool(
+    name="add",
+    func=add,
+    description="Adds two integer numbers to get the result."
+)
+
+def subtract(a: int, b: int) -> int:
+    """Subtract two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
     return a - b
-
-def multiply(a: int, b: int) -> float:
-    """Multiply a and b."""
-    return a*b
-
+subtract = Tool(
+    name="subtract",
+    func=subtract,
+    description="Subtracts two integer numbers to get the result."
+)
+
+def divide(a: int, b: int) -> int:
+    """Divide two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    if b == 0:
+        raise ValueError("Cannot divide by zero.")
+    return a / b
+divide = Tool(
+    name="divide",
+    func=divide,
+    description="Divides two integer numbers to get the result."
+)
+
+def modulus(a: int, b: int) -> int:
+    """Get the modulus of two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    return a % b
+    
+modulus = Tool(
+    name="modulus",
+    func=modulus,
+    description="Calculates the modulus of two integer numbers, a mod b, to get the result."
+)
 
 def calculate_cargo_travel_time(
     origin_coords: Tuple[float, float],
@@ -138,4 +203,33 @@ def calculate_cargo_travel_time(
     flight_time = (actual_distance / cruising_speed_kmh) + 1.0
 
     # Format the results
+
     return round(flight_time, 2)
+
+calculate_cargo_travel_time = Tool(
+    name="calculate_cargo_travel_time",
+    func=calculate_cargo_travel_time,
+    description="Calculates the travel time between two points on the earth based on an object with optionally set speed, usually a cargo plane."
+)
+
+
+def get_hub_stats(author: str) -> str:
+    """Fetches the most downloaded model from a specific author on the Hugging Face Hub."""
+    try:
+        # List models from the specified author, sorted by downloads
+        models = list(list_models(author=author, sort="downloads", direction=-1, limit=1))
+
+        if models:
+            model = models[0]
+            return f"The most downloaded model by {author} is {model.id} with {model.downloads:,} downloads."
+        else:
+            return f"No models found for author {author}."
+    except Exception as e:
+        return f"Error fetching models for {author}: {str(e)}"
+
+# Initialize the tool
+hub_stats_tool = Tool(
+    name="get_hub_stats",
+    func=get_hub_stats,
+    description="Fetches the most downloaded model from a specific author on the Hugging Face Hub."
+)