From d8118140194bd01ee9e39258f98816098daf81fd Mon Sep 17 00:00:00 2001 From: Andrew Truong Date: Wed, 11 Dec 2024 16:28:44 -0500 Subject: [PATCH] test --- weave/trace/autopatch.py | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/weave/trace/autopatch.py b/weave/trace/autopatch.py index 1828ae1128f..0619194a224 100644 --- a/weave/trace/autopatch.py +++ b/weave/trace/autopatch.py @@ -32,20 +32,22 @@ class IntegrationSettings(BaseModel): class AutopatchSettings(BaseModel): """Settings for auto-patching integrations.""" - anthropic: IntegrationSettings = Field(default_factory=IntegrationSettings) - cerebras: IntegrationSettings = Field(default_factory=IntegrationSettings) - cohere: IntegrationSettings = Field(default_factory=IntegrationSettings) - dspy: IntegrationSettings = Field(default_factory=IntegrationSettings) - google_ai_studio: IntegrationSettings = Field(default_factory=IntegrationSettings) - groq: IntegrationSettings = Field(default_factory=IntegrationSettings) - instructor: IntegrationSettings = Field(default_factory=IntegrationSettings) - langchain: IntegrationSettings = Field(default_factory=IntegrationSettings) - litellm: IntegrationSettings = Field(default_factory=IntegrationSettings) - llamaindex: IntegrationSettings = Field(default_factory=IntegrationSettings) - mistral: IntegrationSettings = Field(default_factory=IntegrationSettings) - notdiamond: IntegrationSettings = Field(default_factory=IntegrationSettings) + # These will be uncommented as we add support for more integrations. Note that + + # anthropic: IntegrationSettings = Field(default_factory=IntegrationSettings) + # cerebras: IntegrationSettings = Field(default_factory=IntegrationSettings) + # cohere: IntegrationSettings = Field(default_factory=IntegrationSettings) + # dspy: IntegrationSettings = Field(default_factory=IntegrationSettings) + # google_ai_studio: IntegrationSettings = Field(default_factory=IntegrationSettings) + # groq: IntegrationSettings = Field(default_factory=IntegrationSettings) + # instructor: IntegrationSettings = Field(default_factory=IntegrationSettings) + # langchain: IntegrationSettings = Field(default_factory=IntegrationSettings) + # litellm: IntegrationSettings = Field(default_factory=IntegrationSettings) + # llamaindex: IntegrationSettings = Field(default_factory=IntegrationSettings) + # mistral: IntegrationSettings = Field(default_factory=IntegrationSettings) + # notdiamond: IntegrationSettings = Field(default_factory=IntegrationSettings) openai: IntegrationSettings = Field(default_factory=IntegrationSettings) - vertexai: IntegrationSettings = Field(default_factory=IntegrationSettings) + # vertexai: IntegrationSettings = Field(default_factory=IntegrationSettings) @validate_call