Add support for W&B dedicated cloud instances in Weave tracing integration (#20765)

Co-authored-by: crazywoola <427733928@qq.com>
This commit is contained in:
Bharat Ramanathan
2025-06-07 20:36:23 +05:30
committed by GitHub
parent 930c4cb609
commit e6e76852d5
5 changed files with 31 additions and 4 deletions

View File

@@ -98,6 +98,7 @@ class WeaveConfig(BaseTracingConfig):
entity: str | None = None
project: str
endpoint: str = "https://trace.wandb.ai"
host: str | None = None
@field_validator("endpoint")
@classmethod
@@ -109,6 +110,14 @@ class WeaveConfig(BaseTracingConfig):
return v
@field_validator("host")
@classmethod
def validate_host(cls, v, info: ValidationInfo):
if v is not None and v != "":
if not v.startswith(("https://", "http://")):
raise ValueError("host must start with https:// or http://")
return v
OPS_FILE_PATH = "ops_trace/"
OPS_TRACE_FAILED_KEY = "FAILED_OPS_TRACE"