From e4a986d28671c6b9e8374e756ae3cd992a6c58d7 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Tue, 18 Mar 2025 11:53:13 -0700 Subject: [PATCH] configure logger in the first place in skyvern init (#1965) --- skyvern/__init__.py | 6 ++++-- skyvern/forge/sdk/api/llm/api_handler_factory.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/skyvern/__init__.py b/skyvern/__init__.py index 8d1f410e..c3d97aac 100644 --- a/skyvern/__init__.py +++ b/skyvern/__init__.py @@ -1,9 +1,7 @@ from ddtrace import tracer from ddtrace.filters import FilterRequestsOnUrl -from skyvern.agent import SkyvernAgent, SkyvernClient from skyvern.forge.sdk.forge_log import setup_logger -from skyvern.forge.sdk.workflow.models.workflow import WorkflowRunResponse tracer.configure( settings={ @@ -14,4 +12,8 @@ tracer.configure( ) setup_logger() + +from skyvern.agent import SkyvernAgent, SkyvernClient # noqa: E402 +from skyvern.forge.sdk.workflow.models.workflow import WorkflowRunResponse # noqa: E402 + __all__ = ["SkyvernAgent", "SkyvernClient", "WorkflowRunResponse"] diff --git a/skyvern/forge/sdk/api/llm/api_handler_factory.py b/skyvern/forge/sdk/api/llm/api_handler_factory.py index c7cef93c..4c2af1a3 100644 --- a/skyvern/forge/sdk/api/llm/api_handler_factory.py +++ b/skyvern/forge/sdk/api/llm/api_handler_factory.py @@ -166,7 +166,7 @@ class LLMAPIHandlerFactory: # TODO (suchintan): Properly support reasoning tokens reasoning_tokens = response.get("usage", {}).get("reasoning_tokens", 0) - LOG.info("Reasoning tokens", reasoning_tokens=reasoning_tokens) + LOG.debug("Reasoning tokens", reasoning_tokens=reasoning_tokens) completion_tokens = response.get("usage", {}).get("completion_tokens", 0) + reasoning_tokens