diff --git a/literalai/client.py b/literalai/client.py index c3743cd..2156637 100644 --- a/literalai/client.py +++ b/literalai/client.py @@ -3,27 +3,26 @@ from literalai.api import AsyncLiteralAPI, LiteralAPI from literalai.callback.langchain_callback import get_langchain_callback -from literalai.context import active_steps_var, active_thread_var, active_root_run_var +from literalai.context import active_root_run_var, active_steps_var, active_thread_var from literalai.environment import EnvContextManager, env_decorator -from literalai.event_processor import EventProcessor from literalai.evaluation.experiment_item_run import ( ExperimentItemRunContextManager, experiment_item_run_decorator, ) +from literalai.event_processor import EventProcessor from literalai.instrumentation.mistralai import instrument_mistralai from literalai.instrumentation.openai import instrument_openai -from literalai.observability.message import Message from literalai.my_types import Environment +from literalai.observability.message import Message from literalai.observability.step import ( + Attachment, MessageStepType, Step, StepContextManager, TrueStepType, step_decorator, - Attachment, ) from literalai.observability.thread import ThreadContextManager, thread_decorator - from literalai.requirements import check_all_requirements @@ -204,6 +203,7 @@ def run( parent_id: Optional[str] = None, thread_id: Optional[str] = None, root_run_id: Optional[str] = None, + **kwargs, ): return self.step( original_function=original_function, @@ -213,6 +213,7 @@ def run( parent_id=parent_id, thread_id=thread_id, root_run_id=root_run_id, + **kwargs, ) def message( diff --git a/literalai/observability/step.py b/literalai/observability/step.py index b850b67..d582e2f 100644 --- a/literalai/observability/step.py +++ b/literalai/observability/step.py @@ -380,6 +380,7 @@ def __init__( tags: Optional[List[str]] = None, root_run_id: Optional[str] = None, metadata: Optional[Dict] = None, + **kwargs, ): from time import sleep @@ -404,6 +405,10 @@ def __init__( if metadata: self.metadata = metadata + # Set additional attributes from kwargs + for key, value in kwargs.items(): + setattr(self, key, value) + def start(self): active_steps = active_steps_var.get() if len(active_steps) > 0: diff --git a/literalai/version.py b/literalai/version.py index 485f44a..c272cb1 100644 --- a/literalai/version.py +++ b/literalai/version.py @@ -1 +1 @@ -__version__ = "0.1.1" +__version__ = "0.1.101" diff --git a/setup.py b/setup.py index 8b10813..5253249 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="literalai", - version="0.1.1", # update version in literalai/version.py + version="0.1.101", # update version in literalai/version.py description="An SDK for observability in Python applications", long_description=open("README.md").read(), long_description_content_type="text/markdown",