chore: format file

Author: gustavocidornelas

src/openlayer/lib/tracing/tracer.py

@@ -22,7 +22,9 @@
 TRUE_LIST = ["true", "on", "1"]
 
 _publish = utils.get_env_variable("OPENLAYER_DISABLE_PUBLISH") not in TRUE_LIST
-_verify_ssl = (utils.get_env_variable("OPENLAYER_VERIFY_SSL") or "true").lower() in TRUE_LIST
+_verify_ssl = (
+    utils.get_env_variable("OPENLAYER_VERIFY_SSL") or "true"
+).lower() in TRUE_LIST
 _client = None
 
 
@@ -78,7 +80,11 @@ def create_step(
 ) -> Generator[steps.Step, None, None]:
     """Starts a trace and yields a Step object."""
     new_step, is_root_step, token = _create_and_initialize_step(
-        step_name=name, step_type=step_type, inputs=inputs, output=output, metadata=metadata
+        step_name=name,
+        step_type=step_type,
+        inputs=inputs,
+        output=output,
+        metadata=metadata,
     )
     try:
         yield new_step
@@ -90,7 +96,11 @@ def create_step(
             new_step.latency = latency
 
         _current_step.reset(token)
-        _handle_trace_completion(is_root_step=is_root_step, step_name=name, inference_pipeline_id=inference_pipeline_id)
+        _handle_trace_completion(
+            is_root_step=is_root_step,
+            step_name=name,
+            inference_pipeline_id=inference_pipeline_id,
+        )
 
 
 def add_chat_completion_step_to_trace(**kwargs) -> None:
@@ -152,7 +162,9 @@ def wrapper(*func_args, **func_kwargs):
             if step_kwargs.get("name") is None:
                 step_kwargs["name"] = func.__name__
 
-            with create_step(*step_args, inference_pipeline_id=inference_pipeline_id, **step_kwargs) as step:
+            with create_step(
+                *step_args, inference_pipeline_id=inference_pipeline_id, **step_kwargs
+            ) as step:
                 output = exception = None
                 try:
                     output = func(*func_args, **func_kwargs)
@@ -239,8 +251,12 @@ async def __anext__(self):
                             # Initialize tracing on first iteration only
                             if not self._trace_initialized:
                                 self._original_gen = func(*func_args, **func_kwargs)
-                                self._step, self._is_root_step, self._token = _create_step_for_async_generator(
-                                    step_name=step_name, inference_pipeline_id=inference_pipeline_id, **step_kwargs
+                                self._step, self._is_root_step, self._token = (
+                                    _create_step_for_async_generator(
+                                        step_name=step_name,
+                                        inference_pipeline_id=inference_pipeline_id,
+                                        **step_kwargs,
+                                    )
                                 )
                                 self._inputs = _extract_function_inputs(
                                     func_signature=func_signature,
@@ -290,7 +306,11 @@ async def __anext__(self):
                 # Create wrapper for regular async functions
                 @wraps(func)
                 async def async_function_wrapper(*func_args, **func_kwargs):
-                    with create_step(*step_args, inference_pipeline_id=inference_pipeline_id, **step_kwargs) as step:
+                    with create_step(
+                        *step_args,
+                        inference_pipeline_id=inference_pipeline_id,
+                        **step_kwargs,
+                    ) as step:
                         output = exception = None
 
                         try:
@@ -317,7 +337,11 @@ async def async_function_wrapper(*func_args, **func_kwargs):
             # For sync functions, use the existing logic with optimizations
             @wraps(func)
             def sync_wrapper(*func_args, **func_kwargs):
-                with create_step(*step_args, inference_pipeline_id=inference_pipeline_id, **step_kwargs) as step:
+                with create_step(
+                    *step_args,
+                    inference_pipeline_id=inference_pipeline_id,
+                    **step_kwargs,
+                ) as step:
                     output = exception = None
                     try:
                         output = func(*func_args, **func_kwargs)
@@ -382,7 +406,13 @@ def _create_and_initialize_step(
     Returns:
         Tuple of (step, is_root_step, token)
     """
-    new_step = steps.step_factory(step_type=step_type, name=step_name, inputs=inputs, output=output, metadata=metadata)
+    new_step = steps.step_factory(
+        step_type=step_type,
+        name=step_name,
+        inputs=inputs,
+        output=output,
+        metadata=metadata,
+    )
     new_step.start_time = time.time()
 
     parent_step = get_current_step()
@@ -403,7 +433,9 @@ def _create_and_initialize_step(
     return new_step, is_root_step, token
 
 
-def _handle_trace_completion(is_root_step: bool, step_name: str, inference_pipeline_id: Optional[str] = None) -> None:
+def _handle_trace_completion(
+    is_root_step: bool, step_name: str, inference_pipeline_id: Optional[str] = None
+) -> None:
     """Handle trace completion and data streaming."""
     if is_root_step:
         logger.debug("Ending the trace...")
@@ -466,13 +498,21 @@ def _process_wrapper_inputs_and_outputs(
 ) -> None:
     """Extract function inputs and finalize step logging - common pattern across wrappers."""
     inputs = _extract_function_inputs(
-        func_signature=func_signature, func_args=func_args, func_kwargs=func_kwargs, context_kwarg=context_kwarg
+        func_signature=func_signature,
+        func_args=func_args,
+        func_kwargs=func_kwargs,
+        context_kwarg=context_kwarg,
+    )
+    _finalize_step_logging(
+        step=step, inputs=inputs, output=output, start_time=step.start_time
     )
-    _finalize_step_logging(step=step, inputs=inputs, output=output, start_time=step.start_time)
 
 
 def _extract_function_inputs(
-    func_signature: inspect.Signature, func_args: tuple, func_kwargs: dict, context_kwarg: Optional[str] = None
+    func_signature: inspect.Signature,
+    func_args: tuple,
+    func_kwargs: dict,
+    context_kwarg: Optional[str] = None,
 ) -> dict:
     """Extract and clean function inputs for logging."""
     bound = func_signature.bind(*func_args, **func_kwargs)
@@ -522,7 +562,11 @@ def _create_step_for_async_generator(
 ) -> Tuple[steps.Step, bool, Any]:
     """Create and initialize step for async generators - no context manager."""
     return _create_and_initialize_step(
-        step_name=step_name, step_type=enums.StepType.USER_CALL, inputs=None, output=None, metadata=None
+        step_name=step_name,
+        step_type=enums.StepType.USER_CALL,
+        inputs=None,
+        output=None,
+        metadata=None,
     )
 
 
@@ -537,9 +581,13 @@ def _finalize_async_generator_step(
 ) -> None:
     """Finalize async generator step - called when generator is consumed."""
     _current_step.reset(token)
-    _finalize_step_logging(step=step, inputs=inputs, output=output, start_time=step.start_time)
+    _finalize_step_logging(
+        step=step, inputs=inputs, output=output, start_time=step.start_time
+    )
     _handle_trace_completion(
-        is_root_step=is_root_step, step_name=step_name, inference_pipeline_id=inference_pipeline_id
+        is_root_step=is_root_step,
+        step_name=step_name,
+        inference_pipeline_id=inference_pipeline_id,
     )