mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
3e8f47fd97
* feat(chat): add runner_url to payload for telemetry tracking * feat(telemetry): add runner_url to sanitized fields in telemetry payload * feat(telemetry): replace runner_url with runner_category in telemetry payload and add runner utility functions * fix:ruff
106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
class RunnerCategory:
|
|
LOCAL = 'local'
|
|
CLOUD = 'cloud'
|
|
UNKNOWN = 'unknown'
|
|
|
|
|
|
CLOUD_DOMAINS = [
|
|
'.n8n.cloud',
|
|
'.n8n.io',
|
|
'api.dify.ai',
|
|
'cloud.dify.ai',
|
|
'.coze.com',
|
|
'.coze.cn',
|
|
'cloud.langflow.ai',
|
|
'.langflow.org',
|
|
]
|
|
|
|
LOCAL_PATTERNS = [
|
|
'localhost',
|
|
'127.0.0.1',
|
|
'0.0.0.0',
|
|
'192.168.',
|
|
'10.',
|
|
'172.16.',
|
|
'172.17.',
|
|
'172.18.',
|
|
'172.19.',
|
|
'172.20.',
|
|
'172.21.',
|
|
'172.22.',
|
|
'172.23.',
|
|
'172.24.',
|
|
'172.25.',
|
|
'172.26.',
|
|
'172.27.',
|
|
'172.28.',
|
|
'172.29.',
|
|
'172.30.',
|
|
'172.31.',
|
|
]
|
|
|
|
|
|
def get_runner_category(runner_name: str, runner_url: str) -> str:
|
|
if not runner_url:
|
|
return RunnerCategory.UNKNOWN
|
|
|
|
try:
|
|
parsed_url = urlparse(runner_url)
|
|
host = parsed_url.hostname.lower() if parsed_url.hostname else ''
|
|
except Exception:
|
|
return RunnerCategory.UNKNOWN
|
|
|
|
for pattern in LOCAL_PATTERNS:
|
|
if host.startswith(pattern):
|
|
return RunnerCategory.LOCAL
|
|
|
|
for domain in CLOUD_DOMAINS:
|
|
if host.endswith(domain):
|
|
return RunnerCategory.CLOUD
|
|
|
|
return RunnerCategory.CLOUD
|
|
|
|
|
|
def get_runner_info(runner_name: str, runner_url: str) -> dict:
|
|
return {
|
|
'name': runner_name,
|
|
'url': runner_url,
|
|
'category': get_runner_category(runner_name, runner_url),
|
|
}
|
|
|
|
|
|
def is_cloud_runner(runner_name: str, runner_url: str) -> bool:
|
|
return get_runner_category(runner_name, runner_url) == RunnerCategory.CLOUD
|
|
|
|
|
|
def is_local_runner(runner_name: str, runner_url: str) -> bool:
|
|
return get_runner_category(runner_name, runner_url) == RunnerCategory.LOCAL
|
|
|
|
|
|
def extract_runner_url(runner_name: str, runner, pipeline_config: dict | None) -> str | None:
|
|
if not runner or not hasattr(runner, 'pipeline_config'):
|
|
return None
|
|
|
|
ai_config = pipeline_config.get('ai', {}) if pipeline_config else {}
|
|
|
|
if runner_name == 'dify-service-api':
|
|
return ai_config.get('dify-service-api', {}).get('base-url')
|
|
elif runner_name == 'n8n-service-api':
|
|
return ai_config.get('n8n-service-api', {}).get('webhook-url')
|
|
elif runner_name == 'coze-api':
|
|
return ai_config.get('coze-api', {}).get('api-base')
|
|
elif runner_name == 'langflow-api':
|
|
return ai_config.get('langflow-api', {}).get('base-url')
|
|
|
|
return None
|
|
|
|
|
|
def get_runner_category_from_runner(runner_name: str, runner, pipeline_config: dict | None) -> str:
|
|
runner_url = extract_runner_url(runner_name, runner, pipeline_config)
|
|
return get_runner_category(runner_name, runner_url)
|