File size: 657 Bytes
e222391
 
02c9b56
 
e222391
 
 
 
 
02c9b56
e222391
 
 
02c9b56
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""Base Agent"""

from agents.manager_agent import manager_agent_factory


class BasicAgent:  # pylint: disable=too-few-public-methods
    """Base Agent for Evaluation"""

    def __init__(self):
        self.manager_agent = manager_agent_factory()
        print("BasicAgent initialized.")

    def __call__(self, question: str) -> str:
        if not self.manager_agent:
            print(f"Agent received question (first 50 chars): {question[:50]}...")
            fixed_answer = "This is a default answer."
            print(f"Agent returning fixed answer: {fixed_answer}")
            return fixed_answer
        return self.manager_agent.run(question)