fix test

ServiceNow · Jan 24, 2025 · f579233 · f579233
1 parent 4aaedad
commit f579233
Show file tree

Hide file tree

Showing 4 changed files with 21,314 additions and 2,729 deletions.
diff --git a/tapeagents/core.py b/tapeagents/core.py
@@ -381,7 +381,7 @@ class LLMCall(BaseModel):
     cached: bool
     llm_info: dict = {}
     cost: float = 0
-    logprobs: list = Field(default_factory=list, exclude=True)
+    logprobs: list[TokenLogprob] = Field(default_factory=list, exclude=True)
 
 
 AnnotatorTape = Tape[TapeType, StepType]