agenta 0.13.0__py3-none-any.whl → 0.13.0a0__py3-none-any.whl

agenta/client/backend/types/result.py

@@ -18,19 +18,11 @@ class Result(pydantic.BaseModel):
     error: typing.Optional[Error]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/simple_evaluation_output.py

@@ -20,19 +20,11 @@ class SimpleEvaluationOutput(pydantic.BaseModel):
     evaluation_type: EvaluationType
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/span.py

@@ -30,19 +30,11 @@ class Span(pydantic.BaseModel):
     children: typing.Optional[typing.List[Span]]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/span_detail.py

@@ -31,19 +31,11 @@ class SpanDetail(pydantic.BaseModel):
     config: typing.Optional[typing.Dict[str, typing.Any]]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/span_variant.py

@@ -17,19 +17,11 @@ class SpanVariant(pydantic.BaseModel):
     revision: int
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/template.py

@@ -17,19 +17,11 @@ class Template(pydantic.BaseModel):
     image: TemplateImageInfo
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/template_image_info.py

@@ -22,19 +22,11 @@ class TemplateImageInfo(pydantic.BaseModel):
     template_uri: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/test_set_output_response.py

@@ -17,19 +17,11 @@ class TestSetOutputResponse(pydantic.BaseModel):
     created_at: dt.datetime
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/test_set_simple_response.py

@@ -17,19 +17,11 @@ class TestSetSimpleResponse(pydantic.BaseModel):
     created_at: str
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/trace_detail.py

@@ -31,19 +31,11 @@ class TraceDetail(pydantic.BaseModel):
     config: typing.Dict[str, typing.Any]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/uri.py

@@ -15,19 +15,11 @@ class Uri(pydantic.BaseModel):
     uri: str
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/validation_error.py

@@ -18,19 +18,11 @@ class ValidationError(pydantic.BaseModel):
     type: str
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/variant_action.py

@@ -16,19 +16,11 @@ class VariantAction(pydantic.BaseModel):
     action: VariantActionEnum
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/variant_action_enum.py

@@ -14,9 +14,7 @@ class VariantActionEnum(str, enum.Enum):
     START = "START"
     STOP = "STOP"
 
-    def visit(
-        self, start: typing.Callable[[], T_Result], stop: typing.Callable[[], T_Result]
-    ) -> T_Result:
+    def visit(self, start: typing.Callable[[], T_Result], stop: typing.Callable[[], T_Result]) -> T_Result:
         if self is VariantActionEnum.START:
             return start()
         if self is VariantActionEnum.STOP:

agenta/client/backend/types/with_pagination.py

@@ -18,19 +18,11 @@ class WithPagination(pydantic.BaseModel):
     page_size: int = pydantic.Field(alias="pageSize")
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/workspace_member_response.py

@@ -17,19 +17,11 @@ class WorkspaceMemberResponse(pydantic.BaseModel):
     roles: typing.List[WorkspacePermission]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/workspace_permission.py

@@ -19,19 +19,11 @@ class WorkspacePermission(pydantic.BaseModel):
     permissions: typing.Optional[typing.List[Permission]]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/workspace_response.py

@@ -23,19 +23,11 @@ class WorkspaceResponse(pydantic.BaseModel):
     members: typing.Optional[typing.List[WorkspaceMemberResponse]]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/client/backend/types/workspace_role_response.py

@@ -17,19 +17,11 @@ class WorkspaceRoleResponse(pydantic.BaseModel):
     role_description: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
 
     def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().dict(**kwargs_with_defaults)
 
     class Config:

agenta/sdk/__init__.py

@@ -9,7 +9,6 @@ from .types import (
     InFile,
     IntParam,
     MultipleChoiceParam,
-    GroupedMultipleChoiceParam,
     TextParam,
     MessagesInput,
     FileInputURL,

agenta/sdk/agenta_decorator.py

@@ -25,7 +25,6 @@ from .types import (
     InFile,
     IntParam,
     MultipleChoiceParam,
-    GroupedMultipleChoiceParam,
     TextParam,
     MessagesInput,
     FileInputURL,
@@ -84,8 +83,7 @@ def entrypoint(func: Callable[..., Any]) -> Callable[..., Any]:
         tracing.start_parent_span(
             name=func.__name__,
             inputs=func_params,
-            config=config_params,
-            environment="playground",  # type: ignore #NOTE: wrapper is only called in playground
+            config=api_config_params,
         )
 
         # Ingest files, prepare configurations and run llm app
@@ -99,6 +97,7 @@ def entrypoint(func: Callable[..., Any]) -> Callable[..., Any]:
         tracing.end_recording(
             outputs=llm_result.dict(),
             span=tracing.active_trace,
+            environment="playground",  # type: ignore #NOTE: wrapper is only called in playground
         )
         return llm_result
 
@@ -121,7 +120,6 @@ def entrypoint(func: Callable[..., Any]) -> Callable[..., Any]:
             name=func.__name__,
             inputs=func_params,
             config=config,
-            environment=kwargs["environment"],  # type: ignore #NOTE: wrapper is only called in playground
         )
 
         llm_result = await execute_function(
@@ -132,6 +130,7 @@ def entrypoint(func: Callable[..., Any]) -> Callable[..., Any]:
         tracing.end_recording(
             outputs=llm_result.dict(),
             span=tracing.active_trace,
+            environment="playground",  # type: ignore #NOTE: wrapper is only called in playground
         )
         return llm_result
 
@@ -447,13 +446,6 @@ def override_schema(openapi_schema: dict, func_name: str, endpoint: str, params:
         f"Body_{func_name}_{endpoint}_post"
     ]["properties"]
     for param_name, param_val in params.items():
-        if isinstance(param_val, GroupedMultipleChoiceParam):
-            subschema = find_in_schema(schema_to_override, param_name, "grouped_choice")
-            assert (
-                subschema
-            ), f"GroupedMultipleChoiceParam '{param_name}' is in the parameters but could not be found in the openapi.json"
-            subschema["choices"] = param_val.choices
-            subschema["default"] = param_val.default
         if isinstance(param_val, MultipleChoiceParam):
             subschema = find_in_schema(schema_to_override, param_name, "choice")
             default = str(param_val)

agenta/sdk/agenta_init.py

@@ -90,7 +90,6 @@ class AgentaSingleton:
         self.host = host
         self.app_id = os.environ.get("AGENTA_APP_ID") if app_id is None else app_id
         self.variant_id = os.environ.get("AGENTA_VARIANT_ID")
-        self.variant_name = os.environ.get("AGENTA_VARIANT_NAME")
         self.api_key = api_key
         self.config = Config(base_id=base_id, host=host)
 
@@ -194,7 +193,7 @@ class Config:
                     + str(ex)
                 )
         try:
-            self.set(**{"current_version": config.current_version, **config.parameters})
+            self.set(**config.parameters)
         except Exception as ex:
             logger.warning("Failed to set the configuration with error: " + str(ex))
 
@@ -238,7 +237,6 @@ def llm_tracing(max_workers: Optional[int] = None) -> Tracing:
         base_url=singleton.host,
         app_id=singleton.app_id,  # type: ignore
         variant_id=singleton.variant_id,  # type: ignore
-        variant_name=singleton.variant_name,
         api_key=singleton.api_key,
         max_workers=max_workers,
     )

agenta/sdk/tracing/llm_tracing.py

@@ -36,7 +36,6 @@ class Tracing(object):
         base_url: str,
         app_id: str,
         variant_id: str,
-        variant_name: Optional[str] = None,
         api_key: Optional[str] = None,
         max_workers: Optional[int] = None,
     ):
@@ -45,7 +44,6 @@ class Tracing(object):
         self.llm_logger = llm_logger
         self.app_id = app_id
         self.variant_id = variant_id
-        self.variant_name = variant_name
         self.tasks_manager = TaskQueue(
             max_workers if max_workers else 4, logger=llm_logger
         )
@@ -94,7 +92,7 @@ class Tracing(object):
         self.tags.extend(tags)
 
     def start_parent_span(
-        self, name: str, inputs: Dict[str, Any], config: Dict[str, Any], **kwargs
+        self, name: str, inputs: Dict[str, Any], config: Dict[str, Any]
     ):
         trace_id = self._create_trace_id()
         span_id = self._create_span_id()
@@ -103,11 +101,9 @@ class Tracing(object):
             id=span_id,
             app_id=self.app_id,
             variant_id=self.variant_id,
-            variant_name=self.variant_name,
             inputs=inputs,
             name=name,
             config=config,
-            environment=kwargs.get("environment"),
             spankind=SpanKind.WORKFLOW.value,
             status=SpanStatusCode.UNSET.value,
             start_time=datetime.now(timezone.utc),
@@ -115,9 +111,7 @@ class Tracing(object):
         self.active_trace = span
         self.recording_trace_id = trace_id
         self.parent_span_id = span.id
-        self.llm_logger.info(
-            f"Recorded active_trace and setting parent_span_id: {span.id}"
-        )
+        self.llm_logger.info(f"Recorded active_trace and parent_span_id: {span.id}")
 
     def start_span(
         self,
@@ -132,11 +126,7 @@ class Tracing(object):
             id=span_id,
             inputs=input,
             name=name,
-            app_id=self.app_id,
-            variant_id=self.variant_id,
-            variant_name=self.variant_name,
             config=config,
-            environment=self.active_trace.environment,
             parent_span_id=self.parent_span_id,
             spankind=spankind.upper(),
             attributes={},
@@ -147,9 +137,7 @@ class Tracing(object):
         self.active_span = span
         self.span_dict[span.id] = span
         self.parent_span_id = span.id
-        self.llm_logger.info(
-            f"Recorded active_span and setting parent_span_id: {span.id}"
-        )
+        self.llm_logger.info(f"Recorded active_span and parent_span_id: {span.id}")
         return span
 
     def update_span_status(self, span: CreateSpan, value: str):
@@ -162,34 +150,39 @@ class Tracing(object):
             end_time=datetime.now(timezone.utc),
             outputs=[outputs["message"]],
             cost=outputs.get("cost", None),
+            environment=kwargs.get("environment"),
             tokens=outputs.get("usage"),
         )
 
         # Push span to list of recorded spans
         self.recorded_spans.append(updated_span)
-        self.llm_logger.info(
-            f"Pushed {updated_span.spankind} span {updated_span.id} to recorded spans."
-        )
+        self.llm_logger.info(f"Pushed {updated_span.spankind} span {updated_span.id} to recorded spans.")
 
     def end_recording(self, outputs: Dict[str, Any], span: CreateSpan, **kwargs):
-        self.end_span(outputs=outputs, span=span, **kwargs)
-        if self.api_key == "":
-            return
-
+        updated_span = CreateSpan(
+            **span.dict(),
+            end_time=datetime.now(timezone.utc),
+            outputs=[outputs["message"]],
+            cost=outputs.get("cost", None),
+            environment=kwargs.get("environment"),
+            tokens=outputs.get("usage"),
+        )
+        self.recorded_spans.append(updated_span)
+        self.llm_logger.info(
+            f"Pushed workflow span {updated_span.id} to recorded spans."
+        )
         self.llm_logger.info(f"Preparing to send recorded spans for processing.")
-        self.llm_logger.info(f"Recorded spans => {len(self.recorded_spans)}")
         self.tasks_manager.add_task(
             self.active_trace.id,
             "trace",
             self.client.create_traces(
-                trace=self.recording_trace_id, spans=self.recorded_spans  # type: ignore
+                trace=self.recording_trace_id, spans=self.recorded_spans
             ),
             self.client,
         )
         self.llm_logger.info(
             f"Tracing for {span.id} recorded successfully and sent for processing."
         )
-        self._clear_recorded_spans()
 
     def _create_trace_id(self) -> str:
         """Creates a unique mongo id for the trace object.
@@ -208,13 +201,3 @@ class Tracing(object):
         """
 
         return str(ObjectId())
-
-    def _clear_recorded_spans(self) -> None:
-        """
-        Clear the list of recorded spans to prepare for next batch processing.
-        """
-
-        self.recorded_spans = []
-        self.llm_logger.info(
-            f"Cleared all recorded spans from batch: {self.recorded_spans}"
-        )