lint

Author: PROFeNoM

ddtrace/llmobs/_integrations/vllm.py

@@ -153,7 +153,11 @@ def _llmobs_set_tags(
             return
 
         stats = kwargs.get("stats")
-        ctx = self._build_embedding_context(data, stats) if operation == "embedding" else self._build_completion_context(data, stats)
+        ctx = (
+            self._build_embedding_context(data, stats)
+            if operation == "embedding"
+            else self._build_completion_context(data, stats)
+        )
         ctx[MODEL_NAME] = span.get_tag("vllm.request.model") or ""
         ctx[MODEL_PROVIDER] = span.get_tag("vllm.request.provider") or ""
         span._set_ctx_items(ctx)