check instructor instal

wandb · Oct 12, 2024 · 831db85 · 831db85
1 parent 7359469
commit 831db85
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/weave/flow/scorer/llm_utils.py b/weave/flow/scorer/llm_utils.py
@@ -1,7 +1,5 @@
 from typing import TYPE_CHECKING, List, Optional, Union
 
-import instructor
-
 from weave.trace.autopatch import autopatch
 
 autopatch()  # fix instrucor tracing
@@ -19,6 +17,7 @@
 DEFAULT_MAX_TOKENS = 4096
 
 if TYPE_CHECKING:
+    import instructor
     from anthropic import Anthropic, AsyncAnthropic
     from mistralai import Mistral
     from openai import AsyncOpenAI, OpenAI
@@ -28,7 +27,12 @@
     _LLM_CLIENTS = object
 
 
-def instructor_client(client: _LLM_CLIENTS) -> instructor.client:  # type: ignore
+def instructor_client(client: _LLM_CLIENTS) -> "instructor.client":  # type: ignore
+    try:
+        import instructor
+    except ImportError:
+        raise ImportError("We need instructor to use this the LLM-powered scorers")
+
     client_type = type(client).__name__.lower()
     if "mistral" in client_type:
         return instructor.from_mistral(client)