o xi @srdZddlmZddlZddlZddlZddlZddlZe Z dZ dZ gdZ dd d ZdddZdddZdS)a:Integration module for automatic Weave initialization with W&B. This module provides automatic initialization of Weave when: 1. Weave is installed 2. A W&B run is active with a project 3. Weave is imported (init-on-import) The integration can be disabled by setting the WANDB_DISABLE_WEAVE environment variable. ) annotationsNWANDB_DISABLE_WEAVEweave) anthropicautogencoherecrewaidspyz google.genaigroqzhuggingface_hub.inference instructor langchainlitellm llama_indexmcpmistral notdiamondopenaiagents smolagentsverdict verifiersvertexaientity str | NoneprojectreturnNonec CsttrdS|s dS|r|d|}n|}ttjvr!tdStdzt |WdSt yH}zt d|WYd}~dSd}~ww)zSet up automatic Weave initialization for the current W&B run. Args: project: The W&B project name to use for Weave initialization. N/zInitializing weave.z*Failed to automatically initialize weave: ) osgetenv_DISABLE_WEAVE_WEAVE_PACKAGE_NAMEsysmodules!_maybe_suggest_weave_installationwandbtermlog _weave_init Exceptiontermwarn)rr project_pather,Q/home/ubuntu/.local/lib/python3.10/site-packages/wandb/integration/weave/weave.pysetup3s"   r.cCspddtD}|s dStjt}|durd}nd}tjdd|dd d tj|d d tjd d d dS) zGSuggest Weave installation or import if any target library is imported.cSsg|] }|tjvr|qSr,)r"r#).0libr,r,r- Usz5_maybe_suggest_weave_installation..NzUse W&B Weave for improved LLM call tracing. Install Weave with `pip install weave` then add `import weave` to the top of your script.zUse W&B Weave for improved LLM call tracing. Weave is installed but not imported. Add `import weave` to the top of your script.z Detected [z, z ] in use.F)repeatzIFor more information, check out the docs at: https://weave-docs.wandb.ai/)_AVAILABLE_WEAVE_INTEGRATIONS importlibutil find_specr!r%r&join) imported_libs weave_specmsgr,r,r-r$Ss  r$r*strcCs<tddl}||WddS1swYdS)zPCall weave.init(), assuming weave has been imported. Patched in tests. rN)_weave_init_lockrinit)r*rr,r,r-r'os "r')rrrrrr)rr)r*r;rr)__doc__ __future__rimportlib.utilr4rr" threadingr%Lockr<r r!r3r.r$r'r,r,r,r-s