|
12 | 12 |
|
13 | 13 | import json |
14 | 14 | from datetime import datetime |
15 | | -from typing import Any, Dict, List, Literal, Optional, Union |
| 15 | +from typing import Any, Dict, Literal, Optional, Union |
16 | 16 |
|
17 | 17 | from langfuse._client.constants import ( |
18 | 18 | ObservationTypeGenerationLike, |
19 | 19 | ObservationTypeSpanLike, |
20 | 20 | ) |
21 | 21 | from langfuse._utils.serializer import EventSerializer |
| 22 | +from langfuse.api import MapValue |
22 | 23 | from langfuse.model import PromptClient |
23 | | -from langfuse.types import MapValue, SpanLevel |
| 24 | +from langfuse.types import SpanLevel |
24 | 25 |
|
25 | 26 |
|
26 | 27 | class LangfuseOtelSpanAttributes: |
@@ -73,28 +74,14 @@ class LangfuseOtelSpanAttributes: |
73 | 74 |
|
74 | 75 | def create_trace_attributes( |
75 | 76 | *, |
76 | | - name: Optional[str] = None, |
77 | | - user_id: Optional[str] = None, |
78 | | - session_id: Optional[str] = None, |
79 | | - version: Optional[str] = None, |
80 | | - release: Optional[str] = None, |
81 | 77 | input: Optional[Any] = None, |
82 | 78 | output: Optional[Any] = None, |
83 | | - metadata: Optional[Any] = None, |
84 | | - tags: Optional[List[str]] = None, |
85 | 79 | public: Optional[bool] = None, |
86 | 80 | ) -> dict: |
87 | 81 | attributes = { |
88 | | - LangfuseOtelSpanAttributes.TRACE_NAME: name, |
89 | | - LangfuseOtelSpanAttributes.TRACE_USER_ID: user_id, |
90 | | - LangfuseOtelSpanAttributes.TRACE_SESSION_ID: session_id, |
91 | | - LangfuseOtelSpanAttributes.VERSION: version, |
92 | | - LangfuseOtelSpanAttributes.RELEASE: release, |
93 | 82 | LangfuseOtelSpanAttributes.TRACE_INPUT: _serialize(input), |
94 | 83 | LangfuseOtelSpanAttributes.TRACE_OUTPUT: _serialize(output), |
95 | | - LangfuseOtelSpanAttributes.TRACE_TAGS: tags, |
96 | 84 | LangfuseOtelSpanAttributes.TRACE_PUBLIC: public, |
97 | | - **_flatten_and_serialize_metadata(metadata, "trace"), |
98 | 85 | } |
99 | 86 |
|
100 | 87 | return {k: v for k, v in attributes.items() if v is not None} |
|
0 commit comments