Create codeforces.py

tools/codeforces.py

@@ -0,0 +1,128 @@
+import requests
+from urllib.parse import urlencode
+from typing import List, Dict, Set
+
+BASE_URL = "https://codeforces.com/api/"
+
+
+import requests
+from urllib.parse import urlencode
+from typing import List, Dict, Set
+from smolagents.tools import tool  # Ensure this is the correct import
+
+BASE_URL = "https://codeforces.com/api/"
+
+
+@tool
+def get_contest_data(cid:int, show_unofficial:bool=False)-> Dict: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Fetches the contest standings data for a given contest ID
+    Args:
+        cid: The contest ID
+        show_unofficial: Whether to include unofficial participants
+    """
+    params = {
+        "contestId": str(cid),
+        "showUnofficial": str(show_unofficial).lower(),
+    }
+    response = requests.get(f"{BASE_URL}contest.standings?{urlencode(params)}").json()
+    return response.get("result", {})
+
+
+@tool
+def get_rating_data(handle:str)-> List[Dict]: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Fetches rating change history for a given user handle
+    Args:
+        handle: The Codeforces user handle
+    """
+    response = requests.get(f"{BASE_URL}user.rating?handle={handle}").json()
+    return response.get("result", [])
+
+
+@tool
+def count_of_rated_matches(handle:str)-> int: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Counts the number of rated contests a user has participated in
+    Args:
+        handle: The Codeforces user handle
+    """
+    return len(get_rating_data(handle))
+
+
+@tool
+def get_user_info(handle:str)-> Dict: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Fetches user information from Codeforces
+    Args:
+        handle: The Codeforces user handle
+    """
+    response = requests.get(f"{BASE_URL}user.info?handles={handle}").json()
+    return response.get("result", [{}])[0]
+
+
+@tool
+def get_handle(standings_obj:Dict)-> str: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Extracts the handle from a standings object
+    Args:
+        standings_obj: A dictionary representing a user's contest standings
+    """
+    return standings_obj["party"]["members"][0]["handle"]
+
+
+@tool
+def solve_cutoff(cid:int, show_unofficial:bool, L:int, R:int)-> List[str]: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Returns all handles that solved between L and R number of problems in a contest
+    Args:
+        cid: The contest ID
+        show_unofficial: Whether to include unofficial participants
+        L: The minimum number of problems solved
+        R: The maximum number of problems solved
+    """
+    standings = get_contest_data(cid, show_unofficial)
+    return [
+        get_handle(entry)
+        for entry in standings.get("rows", [])
+        if L <= entry.get("points", 0) <= R
+    ]
+
+
+@tool
+def get_all_rated_users()-> List[List]: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Fetches all rated users with their highest rating
+    """
+    response = requests.get(
+        f"{BASE_URL}user.ratedList?activeOnly=true&includeRetired=false"
+    ).json()
+    return [[user["handle"], user["maxRating"]] for user in response.get("result", [])]
+
+
+@tool
+def filter_rated_users(L:int, R:int)-> List[str]: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Filters users who have a max rating within a given range
+    Args:
+        L: The minimum rating
+        R: The maximum rating
+    """
+    all_handles = get_all_rated_users()
+    return [handle for handle, max_rating in all_handles if L <= max_rating <= R]
+
+
+@tool
+def intersect(sets:List[List[str]])-> List[str]: #it's import to specify the return type
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    """Finds the intersection of multiple lists of handles
+    Args:
+        sets: A list of lists containing handles
+    """
+    sorted_lists = sorted(sets, key=len)
+    result = set(sorted_lists[0]) if sorted_lists else set()
+
+    for s in sorted_lists:
+        result &= set(s)
+
+    return list(result)