Skip to content

decorators

Class info

🛈 DocStrings

Decorators for observability.

track_action 

track_action(msg_template: str | None = None, **kwargs: Any) -> Callable[[F], F]

Register a function for action tracking.

Source code in src/llmling_agent/observability/decorators.py 
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
def track_action(msg_template: str | None = None, **kwargs: Any) -> Callable[[F], F]:
    """Register a function for action tracking."""

    def decorator(func: F) -> F:
        from llmling_agent.observability import registry

        # If we have an active provider, decorate immediately
        logger.info(
            "Decorating function %s.%s with template %s. Current providers: %s",
            func.__module__,
            func.__qualname__,
            msg_template,
            [p.__class__.__name__ for p in registry.providers],
        )
        wrapped = func
        for provider in registry.providers:
            wrapped = provider.wrap_action(wrapped, msg_template=msg_template, **kwargs)  # type: ignore

        # Otherwise queue for later
        action_name = msg_template or func.__name__
        msg = "Queuing action %r with template %r and args %r for later decoration"
        logger.debug(msg, func.__name__, action_name, kwargs)
        registry.register_action(action_name, func, **kwargs)
        return wrapped  # type: ignore

    return decorator

track_agent 

track_agent(name: str | None = None, **kwargs: Any) -> Callable[[T], T]

Register a class for agent tracking.

Source code in src/llmling_agent/observability/decorators.py 
17
18
19
20
21
22
23
24
25
26
27
28
def track_agent(name: str | None = None, **kwargs: Any) -> Callable[[T], T]:
    """Register a class for agent tracking."""

    def decorator(cls: T) -> T:
        from llmling_agent.observability import registry

        agent_name = name or cls.__name__
        logger.debug("Registering agent class %r as %r", cls.__name__, agent_name)
        registry.register_agent(agent_name, cls, **kwargs)
        return cls

    return decorator

track_tool 

track_tool(name: str | None = None, **kwargs: Any) -> Callable[[F], F]

Register a function for tool tracking.

Source code in src/llmling_agent/observability/decorators.py 
31
32
33
34
35
36
37
38
39
40
41
42
def track_tool(name: str | None = None, **kwargs: Any) -> Callable[[F], F]:
    """Register a function for tool tracking."""

    def decorator(func: F) -> F:
        from llmling_agent.observability import registry

        tool_name = name or func.__name__
        logger.debug("Registering tool function %r as %r", func.__name__, tool_name)
        registry.register_tool(tool_name, func, **kwargs)
        return func

    return decorator