Skip to content

Commit 79d51d8

Browse files
fix: weaviate datetime handling for request and response (#346)
* fix: weaviate datetime handling for request and response * bump version * bump version --------- Co-authored-by: Karthik Kalyanaraman <[email protected]>
1 parent c514af4 commit 79d51d8

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

src/langtrace_python_sdk/instrumentation/weaviate/patch.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
"""
1616

1717
import json
18+
from datetime import datetime
1819

1920
from importlib_metadata import version as v
2021
from langtrace.trace_attributes import DatabaseSpanAttributes
@@ -48,9 +49,11 @@
4849
def extract_inputs(args, kwargs):
4950
extracted_params = {}
5051
kwargs_without_properties = {
51-
k: v for k, v in kwargs.items() if k not in ["properties", "fusion_type"]
52+
k: v for k, v in kwargs.items() if k not in ["properties", "fusion_type", "filters"]
5253
}
5354
extracted_params.update(extract_input_params(args, kwargs_without_properties))
55+
if kwargs.get("filters", None):
56+
extracted_params["filters"] = str(kwargs["filters"])
5457
if kwargs.get("fusion_type", None):
5558
extracted_params["fusion_type"] = kwargs["fusion_type"].value
5659
if kwargs.get("properties", None):
@@ -95,9 +98,13 @@ def aggregate_responses(result):
9598

9699

97100
def get_response_object_attributes(response_object):
101+
def convert_value(value):
102+
if isinstance(value, datetime):
103+
return value.isoformat()
104+
return value
98105

99106
response_attributes = {
100-
**response_object.properties,
107+
**{k: convert_value(v) for k, v in response_object.properties.items()},
101108
"uuid": str(response_object.uuid) if hasattr(response_object, "uuid") else None,
102109
"collection": getattr(response_object, "collection", None),
103110
"vector": getattr(response_object, "vector", None),
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "2.3.12"
1+
__version__ = "2.3.13"

0 commit comments

Comments
 (0)