Skip to content

Commit 36be566

Browse files
committed
fix dspy issue for handling sets
1 parent feffac2 commit 36be566

File tree

2 files changed

+6
-14
lines changed

2 files changed

+6
-14
lines changed

src/langtrace_python_sdk/instrumentation/dspy/patch.py

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,5 @@
11
import json
2-
import os
32

4-
import ujson
5-
from colorama import Fore
63
from importlib_metadata import version as v
74
from langtrace.trace_attributes import FrameworkSpanAttributes
85
from opentelemetry import baggage
@@ -11,16 +8,11 @@
118

129
from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
1310
from langtrace_python_sdk.constants.instrumentation.common import (
14-
LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
15-
SERVICE_PROVIDERS,
16-
)
11+
LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
1712
from langtrace_python_sdk.utils import set_span_attribute
18-
from langtrace_python_sdk.utils.llm import (
19-
get_extra_attributes,
20-
get_langtrace_attributes,
21-
get_span_name,
22-
set_span_attributes,
23-
)
13+
from langtrace_python_sdk.utils.llm import (get_extra_attributes,
14+
get_langtrace_attributes,
15+
get_span_name, set_span_attributes)
2416
from langtrace_python_sdk.utils.silently_fail import silently_fail
2517

2618

@@ -139,7 +131,7 @@ def traced_method(wrapped, instance, args, kwargs):
139131
set_span_attribute(
140132
span,
141133
"dspy.signature.result",
142-
json.dumps(result.toDict()),
134+
json.dumps(result.toDict(), default=lambda x: list(x) if isinstance(x, set) else x),
143135
)
144136
span.set_status(Status(StatusCode.OK))
145137

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "3.3.28"
1+
__version__ = "3.3.29"

0 commit comments

Comments
 (0)