(span: Span, part: Response.StreamPartEncoded)
| 807 | } |
| 808 | |
| 809 | const annotateStreamResponse = (span: Span, part: Response.StreamPartEncoded): void => { |
| 810 | if (part.type === "response-metadata") { |
| 811 | addGenAIAnnotations(span, { |
| 812 | response: { |
| 813 | id: part.id, |
| 814 | model: part.modelId |
| 815 | } |
| 816 | }) |
| 817 | } |
| 818 | if (part.type === "finish") { |
| 819 | addGenAIAnnotations(span, { |
| 820 | response: { |
| 821 | finishReasons: [part.reason] |
| 822 | }, |
| 823 | usage: { |
| 824 | inputTokens: part.usage.inputTokens, |
| 825 | outputTokens: part.usage.outputTokens |
| 826 | } |
| 827 | }) |
| 828 | } |
| 829 | } |
| 830 | |
| 831 | // ============================================================================= |
| 832 | // Tool Calling |
no outgoing calls
no test coverage detected