Spaces:
Sleeping
Sleeping
File size: 2,739 Bytes
910ae58 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
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)
|