laminar fix attempt (#3742)
This commit is contained in:
@@ -1,9 +1,11 @@
|
||||
from functools import wraps
|
||||
from typing import Any, Awaitable, Callable, ParamSpec, TypeVar
|
||||
|
||||
from skyvern.forge import app
|
||||
from skyvern.forge.sdk.core import skyvern_context
|
||||
from skyvern.forge.sdk.settings_manager import SettingsManager
|
||||
from skyvern.forge.sdk.trace.base import BaseTrace, NoOpTrace
|
||||
from skyvern.forge.sdk.trace.experiment_utils import collect_experiment_metadata_safely
|
||||
|
||||
P = ParamSpec("P")
|
||||
R = TypeVar("R")
|
||||
@@ -37,6 +39,11 @@ class TraceManager:
|
||||
new_metadata["organization_name"] = context.organization_name
|
||||
user_id = context.run_id
|
||||
|
||||
# Collect experiment metadata and include it in the span metadata
|
||||
experiment_metadata = await collect_experiment_metadata_safely(app.EXPERIMENTATION_PROVIDER)
|
||||
if experiment_metadata:
|
||||
new_metadata.update(experiment_metadata)
|
||||
|
||||
new_tags: list[str] = tags or []
|
||||
new_tags.append(SettingsManager.get_settings().ENV)
|
||||
|
||||
|
||||
@@ -42,7 +42,7 @@ class LaminarTrace(BaseTrace):
|
||||
|
||||
# Map status to appropriate tag
|
||||
status_tag_map = {
|
||||
"completed": "COMPLETION",
|
||||
"completed": "COMPLETED",
|
||||
"failed": "FAILURE",
|
||||
"timed_out": "TIMEOUT",
|
||||
"canceled": "CANCELED",
|
||||
|
||||
Reference in New Issue
Block a user