useragents/src/registry.py


1
2
3
4
5
6
7
8
9
10
11

class AgentRegistry:
    def __init__(self):
        self.registry = {}

    def __call__(self, func):
        self.registry[func.__name__] = func
        print(f"Registered agent: {func.__name__}")
        return func


agentRegistry = AgentRegistry()