ag2 0.9.1a1__py3-none-any.whl → 0.9.1.post0__py3-none-any.whl

autogen/agentchat/contrib/captainagent/tools/math/find_continuity_point.py

@@ -0,0 +1,34 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+def find_continuity_point(f_leq, f_gt, x_value):
+    """Find the value 'a' that ensures the continuity of a piecewise function at a given point.
+
+    Args:
+        f_leq (str): The function expression for f(x) when x is less than or equal to the continuity point, in the form of a string.
+        f_gt (str): The function expression for f(x) when x is greater than the continuity point, in the form of a string.
+        x_value (float): The x-value at which continuity is to be ensured.
+
+    Returns:
+        float or None: The value of 'a' that satisfies the continuity condition,
+        or None if no such value exists.
+    """
+    from sympy import Eq, solve, symbols, sympify
+
+    x, a = symbols("x a")
+
+    # Convert string to sympy expression
+    f_leq_expr = sympify(f_leq)
+    f_gt_expr = sympify(f_gt)
+
+    # Evaluate the expressions at the given x_value
+    f_leq_value = f_leq_expr.subs(x, x_value)
+    f_gt_value = f_gt_expr.subs(x, x_value)
+
+    # Set up the equation for a
+    equation = Eq(f_leq_value, f_gt_value)
+
+    # Solve the equation
+    a_value = solve(equation, a)
+
+    return a_value[0] if a_value else None

autogen/agentchat/contrib/captainagent/tools/math/fraction_to_mixed_numbers.py

@@ -0,0 +1,39 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+def fraction_to_mixed_numbers(numerator, denominator):
+    """Simplifies a fraction to its lowest terms and returns it as a mixed number.
+
+    Args:
+        numerator (int): The numerator of the fraction.
+        denominator (int): The denominator of the fraction.
+
+    Returns:
+        str: The simplified fraction as a string. If the fraction is already an integer, it returns the integer as a string.
+             If the fraction is a proper fraction, it returns the mixed number representation as a string.
+             If the numerator or denominator is not an integer, it returns an error message.
+             If the denominator is zero, it returns an error message.
+    """
+    from sympy import Rational
+
+    # Ensure that numerator and denominator are integers
+    if not isinstance(numerator, int) or not isinstance(denominator, int):
+        return "Error: Numerator and denominator must be integers."
+
+    # Handle the case where the denominator is zero
+    if denominator == 0:
+        return "Error: Denominator cannot be zero."
+
+    # Simplify the fraction to its lowest terms
+    result = Rational(numerator, denominator)
+    # Return the result as a mixed number if needed
+    if result.is_integer:
+        return str(int(result))
+    else:
+        # Result as a mixed number
+        integer_part = int(result)
+        fractional_part = result - integer_part
+        if fractional_part != 0:
+            return f"{integer_part} {fractional_part}"
+        else:
+            return str(integer_part)

autogen/agentchat/contrib/captainagent/tools/math/modular_inverse_sum.py

@@ -0,0 +1,23 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+from contextlib import suppress
+
+
+def modular_inverse_sum(expressions, modulus):
+    """Calculates the sum of modular inverses of the given expressions modulo the specified modulus.
+
+    Args:
+        expressions (list): A list of numbers for which the modular inverses need to be calculated.
+        modulus (int): The modulus value.
+
+    Returns:
+        int: The sum of modular inverses modulo the specified modulus.
+    """
+    from sympy import mod_inverse
+
+    mod_sum = 0
+    for number in expressions:
+        with suppress(ValueError):
+            mod_sum += mod_inverse(number, modulus)
+    return mod_sum % modulus

autogen/agentchat/contrib/captainagent/tools/math/simplify_mixed_numbers.py

@@ -0,0 +1,36 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+def simplify_mixed_numbers(numerator1, denominator1, numerator2, denominator2, whole_number1, whole_number2):
+    """Simplifies the sum of two mixed numbers and returns the result as a string in the format 'a b/c'.
+
+    Args:
+        numerator1 (int): The numerator of the first fraction.
+        denominator1 (int): The denominator of the first fraction.
+        numerator2 (int): The numerator of the second fraction.
+        denominator2 (int): The denominator of the second fraction.
+        whole_number1 (int): The whole number part of the first mixed number.
+        whole_number2 (int): The whole number part of the second mixed number.
+
+    Returns:
+        str: The simplified sum of the two mixed numbers as a string in the format 'a b/c'.
+    """
+    from fractions import Fraction
+
+    # Convert mixed numbers to improper fractions
+    fraction1 = whole_number1 * denominator1 + numerator1
+    fraction2 = whole_number2 * denominator2 + numerator2
+    # Create Fraction objects
+    frac1 = Fraction(fraction1, denominator1)
+    frac2 = Fraction(fraction2, denominator2)
+    # Calculate the sum
+    result = frac1 + frac2
+    # Convert to mixed number
+    mixed_number = result.numerator // result.denominator
+    mixed_fraction_numerator = result.numerator % result.denominator
+    mixed_fraction = Fraction(mixed_fraction_numerator, result.denominator)
+    # Return as a string in the format 'a b/c'
+    if mixed_fraction_numerator > 0:
+        return f"{mixed_number} {mixed_fraction}"
+    else:
+        return str(mixed_number)

autogen/agentchat/contrib/captainagent/tools/math/sum_of_digit_factorials.py

@@ -0,0 +1,15 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+def sum_of_digit_factorials(number):
+    """Calculates the sum of the factorial of each digit in a number, often used in problems involving curious numbers like 145.
+
+    Args:
+        number (int): The number for which to calculate the sum of digit factorials.
+
+    Returns:
+        int: The sum of the factorials of the digits in the given number.
+    """
+    from math import factorial
+
+    return sum(factorial(int(digit)) for digit in str(number))

autogen/agentchat/contrib/captainagent/tools/math/sum_of_primes_below.py

@@ -0,0 +1,15 @@
+# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
+#
+# SPDX-License-Identifier: Apache-2.0
+def sum_of_primes_below(threshold):
+    """Calculates the sum of all prime numbers below a given threshold.
+
+    Args:
+        threshold (int): The maximum number (exclusive) up to which primes are summed.
+
+    Returns:
+        int: The sum of all prime numbers below the threshold.
+    """
+    from sympy import primerange
+
+    return sum(primerange(2, threshold))

autogen/agentchat/contrib/captainagent/tools/requirements.txt

@@ -0,0 +1,10 @@
+markdownify
+arxiv
+pymupdf
+wikipedia-api
+easyocr
+python-pptx
+openai-whisper
+pandas
+scipy
+sentence-transformers

autogen/agentchat/contrib/captainagent/tools/tool_description.tsv

@@ -0,0 +1,34 @@
+docid	document_content
+1	math complex_numbers_product Calculates the product of a list of complex numbers.
+2	math calculate_matrix_power Calculate the power of a given matrix.
+3	math calculate_day_of_the_week Calculates the day of the week after a given number of days starting from a specified day.
+4	math modular_inverse_sum Calculates the sum of modular inverses of the given expressions modulo the specified modulus.
+5	math sum_of_digit_factorials Calculates the sum of the factorial of each digit in a number, often used in problems involving curious numbers like 145.
+6	math sum_of_primes_below Calculates the sum of all prime numbers below a given threshold.
+7	math evaluate_expression Evaluates a mathematical expression with support for floor function notation and power notation.
+8	math compute_currency_conversion Compute the currency conversion of the given amount using the provided exchange rate.
+9	math find_continuity_point Find the value 'a' that ensures the continuity of a piecewise function at a given point.
+10	math simplify_mixed_numbers Simplifies the sum of two mixed numbers and returns the result as a string in the format 'a b/c'.
+11	math fraction_to_mixed_numbers Simplifies a fraction to its lowest terms and returns it as a mixed number.
+12	math calculate_fraction_sum Calculates the sum of two fractions and returns the result as a mixed number.
+13	math count_distinct_permutations Counts the number of distinct permutations of a sequence where items may be indistinguishable.
+14	math calculate_circle_area_from_diameter Calculate the area of a circle given its diameter.
+15	math calculate_reflected_point Calculates the reflection point of a given point about the line y=x.
+16	data_analysis explore_csv Reads a CSV file and prints the column names, shape, data types, and the first few lines of data.
+17	data_analysis calculate_correlation Calculate the correlation between two columns in a CSV file.
+18	data_analysis detect_outlier_zscore Detect outliers in a CSV file based on a specified column. The outliers are determined by calculating the z-score of the data points in the column.
+19	data_analysis detect_outlier_iqr Detect outliers in a specified column of a CSV file using the IQR method.
+20	data_analysis shapiro_wilk_test Perform the Shapiro-Wilk test on a specified column of a CSV file.
+21	data_analysis calculate_skewness_and_kurtosis Calculate the skewness and kurtosis of a specified column in a CSV file. The kurtosis is calculated using the Fisher definition. The two metrics are computed using scipy.stats functions.
+22	information_retrieval perform_web_search Perform a web search using Bing API.
+23	information_retrieval transcribe_audio_file Transcribes the audio file located at the given file path.
+24	information_retrieval arxiv_search Search for articles on arXiv based on the given query.
+25	information_retrieval arxiv_download Downloads PDF files from ArXiv based on a list of arxiv paper IDs.
+26	information_retrieval scrape_wikipedia_tables Scrapes Wikipedia tables based on a given URL and header keyword.
+27	information_retrieval extract_pdf_text Extracts text from a specified page or the entire PDF file.
+28	information_retrieval extract_pdf_image Extracts images from a PDF file and saves them to the specified output directory.
+29	information_retrieval image_qa Perform question answering on an image using a pre-trained VQA model.
+30	information_retrieval optical_character_recognition Perform optical character recognition (OCR) on the given image.
+31	information_retrieval get_youtube_caption Retrieves the captions for a YouTube video.
+32	information_retrieval youtube_download Downloads a YouTube video and returns the download link.
+33	information_retrieval get_wikipedia_text Retrieves the text content of a Wikipedia page. It does not support tables and other complex formatting.