chore(weave): allow importing annotation spec from weave

wandb · Jan 14, 2025 · 41b44a1 · 41b44a1
1 parent 6eb43d0
commit 41b44a1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/weave/__init__.py b/weave/__init__.py
@@ -8,6 +8,7 @@
 
 from weave.flow.agent import Agent as Agent
 from weave.flow.agent import AgentState as AgentState
+from weave.flow.annotation_spec import AnnotationSpec
 from weave.flow.dataset import Dataset
 from weave.flow.eval import Evaluation, Scorer
 from weave.flow.model import Model
@@ -40,4 +41,5 @@
     MessagesPrompt,
     Evaluation,
     Scorer,
+    AnnotationSpec,
 ]