INNER CODE UNIT · Python
get_agent_class
nottelabs/notte · packages/notte-agent/src/notte_agent/main.py:23
def get_agent_class(self) -> type[FalcoAgent]:
match self:
case AgentType.FALCO:
return FalcoAgent
class Agent(BaseAgent):
def __init__(
self,
session: NotteSession,
vault: BaseVault | None = None,
persona: BasePersona | None = None,
notifier: BaseNotifier | None = None,
agent_type: AgentType = AgentType.FALCO,
trajectory: Trajectory | None = None,
**data: Unpack[AgentCreateRequestDict],
):
super().__init__(session=session)