move setup_logger to skyvern/__init__.py (#658)

This commit is contained in:
Kerem Yilmaz 2024-07-29 14:20:58 -07:00 committed by GitHub
parent dd188b5f4d
commit 3d2e27ebd1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 3 additions and 4 deletions

View file

@ -0,0 +1,3 @@
from skyvern.forge.sdk.forge_log import setup_logger
setup_logger()

View file

@ -2,7 +2,6 @@ import structlog
import uvicorn import uvicorn
from dotenv import load_dotenv from dotenv import load_dotenv
import skyvern.forge.sdk.forge_log as forge_log
from skyvern import analytics from skyvern import analytics
from skyvern.forge.sdk.settings_manager import SettingsManager from skyvern.forge.sdk.settings_manager import SettingsManager
@ -11,7 +10,6 @@ LOG = structlog.stdlib.get_logger()
if __name__ == "__main__": if __name__ == "__main__":
analytics.capture("skyvern-oss-run-server") analytics.capture("skyvern-oss-run-server")
forge_log.setup_logger()
port = SettingsManager.get_settings().PORT port = SettingsManager.get_settings().PORT
LOG.info("Agent server starting.", host="0.0.0.0", port=port) LOG.info("Agent server starting.", host="0.0.0.0", port=port)
load_dotenv() load_dotenv()

View file

@ -12,7 +12,6 @@ from skyvern.forge.sdk.artifact.manager import ArtifactManager
from skyvern.forge.sdk.artifact.storage.factory import StorageFactory from skyvern.forge.sdk.artifact.storage.factory import StorageFactory
from skyvern.forge.sdk.db.client import AgentDB from skyvern.forge.sdk.db.client import AgentDB
from skyvern.forge.sdk.experimentation.providers import BaseExperimentationProvider, NoOpExperimentationProvider from skyvern.forge.sdk.experimentation.providers import BaseExperimentationProvider, NoOpExperimentationProvider
from skyvern.forge.sdk.forge_log import setup_logger
from skyvern.forge.sdk.models import Organization from skyvern.forge.sdk.models import Organization
from skyvern.forge.sdk.settings_manager import SettingsManager from skyvern.forge.sdk.settings_manager import SettingsManager
from skyvern.forge.sdk.workflow.context_manager import WorkflowContextManager from skyvern.forge.sdk.workflow.context_manager import WorkflowContextManager
@ -27,7 +26,6 @@ tracer.configure(
}, },
) )
setup_logger()
SETTINGS_MANAGER = SettingsManager.get_settings() SETTINGS_MANAGER = SettingsManager.get_settings()
DATABASE = AgentDB( DATABASE = AgentDB(
SettingsManager.get_settings().DATABASE_STRING, SettingsManager.get_settings().DATABASE_STRING,