agents.py

from args import Args
from itf_agent import IAgent


class Manager(IAgent):
    """
    Orchestrates the workflow by delegating tasks to specialized nodes and integrating their outputs
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "01_manager.txt", Args.primary_llm_interface)


class Auditor(IAgent):
    """
    Reviews manager's outputs for accuracy, safety, and quality
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "02_auditor.txt", Args.primary_llm_interface)


class Summarizer(IAgent):
    """
    Generates concise summaries of conversations or passages.
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "04_summarizer.txt", Args.primary_llm_interface)


class Solver(IAgent):
    """
    Central problem-solving node that coordinates with specialized experts based on task requirements
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "03_solver.txt", Args.primary_llm_interface)


class Researcher(IAgent):
    """
    Retrieves and synthesizes information from various sources to answer knowledge-based questions
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "05_researcher.txt", Args.primary_llm_interface)


class EncryptionExpert(IAgent):
    """
    Handles encryption/decryption tasks and encoding/decoding operations
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "06_encryption_expert.txt", Args.primary_llm_interface)


class MathExpert(IAgent):
    """
    Performs mathematical calculations and solves numerical problems
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "07_math_expert.txt", Args.primary_llm_interface)


class Reasoner(IAgent):
    """
    Performs logical reasoning, inference, and step-by-step problem-solving
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "08_reasoner.txt", Args.primary_llm_interface)


class ImageHandler(IAgent):
    """
    Processes, analyzes, and generates information related to images
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "09_image_handler.txt", Args.vlm_interface)


class VideoHandler(IAgent):
    """
    Processes, analyzes, and generates information related to videos
    """
    def __init__(self, temperature, max_tokens):
        super().__init__(temperature, max_tokens, "10_video_handler.txt", Args.vlm_interface)