fix: import cyclical loop

Author: justinmerrell

runpod/__init__.py

@@ -2,7 +2,6 @@
 
 import os
 import logging
-import platform
 
 from . import serverless
 from .serverless.modules.rp_logger import RunPodLogger
@@ -36,13 +35,6 @@
 endpoint_url_base = "https://api.runpod.ai/v2"  # pylint: disable=invalid-name
 
 
-# ----------------------------- Config User Agent ---------------------------- #
-os_info = f"{platform.system()} {platform.release()}; {platform.machine()}"
-USER_AGENT = f"RunPod-Python-SDK/{__version__} ({os_info}) Language/Python{platform.python_version()}"  # pylint: disable=line-too-long
-
-if os.environ.get('RUNPOD_UA_INTEGRATION') is not None:
-    USER_AGENT += f" {os.environ.get('RUNPOD_UA_INTEGRATION')}"
-
 # --------------------------- Force Logging Levels --------------------------- #
 logging.getLogger("urllib3").setLevel(logging.WARNING)
 logging.getLogger("paramiko").setLevel(logging.WARNING)

runpod/api/graphql.py

@@ -7,7 +7,8 @@
 
 import requests
 
-from runpod import error, USER_AGENT
+from runpod import error
+from runpod.user_agent import USER_AGENT
 
 HTTP_STATUS_UNAUTHORIZED = 401
 

runpod/serverless/modules/rp_ping.py

@@ -9,12 +9,13 @@
 import requests
 from urllib3.util.retry import Retry
 
+from runpod.version import __version__ as runpod_version
 from runpod.serverless.modules.rp_logger import RunPodLogger
-from .worker_state import Jobs, WORKER_ID
-from ...version import __version__ as runpod_version
+from runpod.serverless.modules.worker_state import Jobs, WORKER_ID
+
 
 log = RunPodLogger()
-jobs = Jobs() # Contains the list of jobs that are currently running.
+jobs = Jobs()  # Contains the list of jobs that are currently running.
 
 
 class Heartbeat:

runpod/serverless/worker.py

@@ -8,7 +8,7 @@
 
 import aiohttp
 
-from runpod import USER_AGENT
+from runpod.user_agent import USER_AGENT
 from runpod.serverless.modules import (
     rp_logger, rp_local, rp_handler, rp_ping,
     rp_scale

runpod/user_agent.py

@@ -0,0 +1,14 @@
+import os
+import platform
+
+from runpod.version import __version__ as runpod_version
+
+os_info = f"{platform.system()} {platform.release()}; {platform.machine()}"
+
+USER_AGENT = f"RunPod-Python-SDK/{runpod_version}"
+USER_AGENT += f" ({os_info})"
+USER_AGENT += f" Language/Python {platform.python_version()}"
+
+
+if os.environ.get('RUNPOD_UA_INTEGRATION') is not None:
+    USER_AGENT += f" Integration/{os.environ.get('RUNPOD_UA_INTEGRATION')}"