OTEL backend for metrics/traces/logs (#4632)

Co-authored-by: Benji Visser <benji@093b.org>
Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Shuchang Zheng
2026-02-04 17:54:07 -08:00
committed by GitHub
parent be00c65071
commit 566a108d5d
19 changed files with 2935 additions and 2807 deletions

View File

@@ -43,7 +43,7 @@ from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
from skyvern.forge.sdk.models import SpeculativeLLMMetadata, Step
from skyvern.forge.sdk.schemas.ai_suggestions import AISuggestion
from skyvern.forge.sdk.schemas.task_v2 import TaskV2, Thought
from skyvern.forge.sdk.trace import TraceManager
from skyvern.forge.sdk.trace import traced
from skyvern.utils.image_resizer import Resolution, get_resize_target_dimension, resize_screenshots
LOG = structlog.get_logger()
@@ -410,7 +410,7 @@ class LLMAPIHandlerFactory:
)
main_model_group = llm_config.main_model_group
@TraceManager.traced_async(tags=[llm_key], ignore_inputs=["prompt", "screenshots", "parameters"])
@traced(tags=[llm_key])
async def llm_api_handler_with_router_and_fallback(
prompt: str,
prompt_name: str,
@@ -843,7 +843,7 @@ class LLMAPIHandlerFactory:
assert isinstance(llm_config, LLMConfig)
@TraceManager.traced_async(tags=[llm_key], ignore_inputs=["prompt", "screenshots", "parameters"])
@traced(tags=[llm_key])
async def llm_api_handler(
prompt: str,
prompt_name: str,
@@ -1569,7 +1569,7 @@ class LLMCaller:
return get_resize_target_dimension(window_dimension)
return self.screenshot_resize_target_dimension
@TraceManager.traced_async(ignore_input=True)
@traced()
async def _dispatch_llm_call(
self,
messages: list[dict[str, Any]],