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