|
68 | 68 |
|
69 | 69 |
|
70 | 70 | def _set_response_attributes( |
71 | | - invocation: LLMInvocation, |
| 71 | + invocation: LLMInvocation, # pyright: ignore[reportDeprecated] |
72 | 72 | result: "Message | None", |
73 | 73 | capture_content: bool, |
74 | 74 | ) -> None: |
@@ -203,14 +203,14 @@ def _stop(self) -> None: |
203 | 203 | self.invocation, self._message, self._capture_content |
204 | 204 | ) |
205 | 205 | with self._safe_instrumentation("stop_llm"): |
206 | | - self.handler.stop_llm(self.invocation) |
| 206 | + self.handler.stop_llm(self.invocation) # pyright: ignore[reportDeprecated] |
207 | 207 | self._finalized = True |
208 | 208 |
|
209 | 209 | def _fail(self, message: str, error_type: type[BaseException]) -> None: |
210 | 210 | if self._finalized: |
211 | 211 | return |
212 | 212 | with self._safe_instrumentation("fail_llm"): |
213 | | - self.handler.fail_llm( |
| 213 | + self.handler.fail_llm( # pyright: ignore[reportDeprecated] |
214 | 214 | self.invocation, Error(message=message, type=error_type) |
215 | 215 | ) |
216 | 216 | self._finalized = True |
@@ -258,7 +258,7 @@ def __init__( |
258 | 258 | self, |
259 | 259 | stream: "AsyncStream[RawMessageStreamEvent] | AsyncMessageStream[ResponseFormatT]", |
260 | 260 | handler: TelemetryHandler, |
261 | | - invocation: LLMInvocation, |
| 261 | + invocation: LLMInvocation, # pyright: ignore[reportDeprecated] |
262 | 262 | capture_content: bool, |
263 | 263 | ): |
264 | 264 | self.stream = stream |
@@ -324,7 +324,7 @@ def __init__( |
324 | 324 | self, |
325 | 325 | manager: "MessageStreamManager[ResponseFormatT]", |
326 | 326 | handler: TelemetryHandler, |
327 | | - invocation: LLMInvocation, |
| 327 | + invocation: LLMInvocation, # pyright: ignore[reportDeprecated] |
328 | 328 | capture_content: bool, |
329 | 329 | ): |
330 | 330 | self._manager = manager |
@@ -382,7 +382,7 @@ def __init__( |
382 | 382 | self, |
383 | 383 | manager: "AsyncMessageStreamManager[ResponseFormatT]", |
384 | 384 | handler: TelemetryHandler, |
385 | | - invocation: LLMInvocation, |
| 385 | + invocation: LLMInvocation, # pyright: ignore[reportDeprecated] |
386 | 386 | capture_content: bool, |
387 | 387 | ): |
388 | 388 | self._manager = manager |
|
0 commit comments