diff --git a/weave/trace/autopatch.py b/weave/trace/autopatch.py index 56cd2c57f211..04d308cb2154 100644 --- a/weave/trace/autopatch.py +++ b/weave/trace/autopatch.py @@ -65,18 +65,16 @@ def autopatch(settings: Optional[AutopatchSettings] = None) -> None: get_google_genai_patcher, ) from weave.integrations.groq.groq_sdk import get_groq_patcher + from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher from weave.integrations.instructor.instructor_sdk import get_instructor_patcher - from weave.integrations.langchain.langchain import langchain_patcher from weave.integrations.langchain_nvidia_ai_endpoints.langchain_nv_ai_endpoints import ( get_nvidia_ai_patcher, ) from weave.integrations.litellm.litellm import get_litellm_patcher - from weave.integrations.llamaindex.llamaindex import llamaindex_patcher from weave.integrations.mistral import get_mistral_patcher from weave.integrations.notdiamond.tracing import get_notdiamond_patcher from weave.integrations.openai.openai_sdk import get_openai_patcher from weave.integrations.vertexai.vertexai_sdk import get_vertexai_patcher - from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher get_openai_patcher(settings.openai).attempt_patch() get_mistral_patcher(settings.mistral).attempt_patch() @@ -103,6 +101,7 @@ def reset_autopatch() -> None: get_google_genai_patcher, ) from weave.integrations.groq.groq_sdk import get_groq_patcher + from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher from weave.integrations.instructor.instructor_sdk import get_instructor_patcher from weave.integrations.langchain.langchain import langchain_patcher from weave.integrations.langchain_nvidia_ai_endpoints.langchain_nv_ai_endpoints import ( @@ -114,7 +113,6 @@ def reset_autopatch() -> None: from weave.integrations.notdiamond.tracing import get_notdiamond_patcher from weave.integrations.openai.openai_sdk import get_openai_patcher from weave.integrations.vertexai.vertexai_sdk import get_vertexai_patcher - from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher get_openai_patcher().undo_patch() get_mistral_patcher().undo_patch()