vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

configure logger in the first place in skyvern init (#1965)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-18 11:53:13 -07:00
committed by GitHub
parent c680e33b30
commit e4a986d286
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/__init__.py
View File

@@ -1,9 +1,7 @@
from ddtrace import tracer from ddtrace import tracer
from ddtrace.filters import FilterRequestsOnUrl from ddtrace.filters import FilterRequestsOnUrl
from skyvern.agent import SkyvernAgent, SkyvernClient
from skyvern.forge.sdk.forge_log import setup_logger from skyvern.forge.sdk.forge_log import setup_logger
from skyvern.forge.sdk.workflow.models.workflow import WorkflowRunResponse
tracer.configure( tracer.configure(
settings={ settings={
@@ -14,4 +12,8 @@ tracer.configure(
) )
setup_logger() 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"] __all__ = ["SkyvernAgent", "SkyvernClient", "WorkflowRunResponse"]

2
skyvern/forge/sdk/api/llm/api_handler_factory.py
View File

@@ -166,7 +166,7 @@ class LLMAPIHandlerFactory:
# TODO (suchintan): Properly support reasoning tokens # TODO (suchintan): Properly support reasoning tokens
reasoning_tokens = response.get("usage", {}).get("reasoning_tokens", 0) 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 completion_tokens = response.get("usage", {}).get("completion_tokens", 0) + reasoning_tokens