Source code for vein.sources.memory

from typing import Callable

from vein.flow.items import LiveFlowItem
from vein.sources.memory.callable import CallableTracker
from vein.sources.memory.environment import EnvironmentLiveTracker




def from_env(key: str, **kwargs) -> LiveFlowItem:
    return LiveFlowItem(EnvironmentLiveTracker(key, **kwargs))





def from_callable(callable: Callable, *args, **kwargs) -> LiveFlowItem:
    return LiveFlowItem(CallableTracker(callable, *args, **kwargs))