feat: add weave to tracing providers configs and manager

pull/14262/head
Bharat Ramanathan 1 year ago
parent 674ffd9020
commit 1bf35eaf74

@ -88,6 +88,26 @@ class OpikConfig(BaseTracingConfig):
return v
class WeaveConfig(BaseTracingConfig):
"""
Model class for Weave tracing config.
"""
api_key: str
entity: str | None = None
project: str
endpoint: str = "https://trace.wandb.ai"
@field_validator("endpoint")
@classmethod
def set_value(cls, v, info: ValidationInfo):
if v is None or v == "":
v = "https://trace.wandb.ai"
if not v.startswith("https://"):
raise ValueError("endpoint must start with https://")
return v
class WeaveConfig(BaseTracingConfig):
"""

@ -19,6 +19,7 @@ from core.ops.entities.config_entity import (
LangfuseConfig,
LangSmithConfig,
OpikConfig,
WeaveConfig,
TracingProviderEnum,
WeaveConfig,
)
@ -35,6 +36,8 @@ from core.ops.entities.trace_entity import (
)
from core.ops.langfuse_trace.langfuse_trace import LangFuseDataTrace
from core.ops.langsmith_trace.langsmith_trace import LangSmithDataTrace
from core.ops.weave_trace.weave_trace import WeaveDataTrace
from core.ops.opik_trace.opik_trace import OpikDataTrace
from core.ops.utils import get_message_data
from core.ops.weave_trace.weave_trace import WeaveDataTrace
from extensions.ext_database import db

Loading…
Cancel
Save