Skip to content

Commit d2c8267

Browse files
authored
fix: weaviate fusion type enum handling (#328)
1 parent abd75f3 commit d2c8267

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

src/langtrace_python_sdk/instrumentation/weaviate/patch.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,12 @@
4747

4848
def extract_inputs(args, kwargs):
4949
extracted_params = {}
50-
kwargs_without_properties = {k: v for k, v in kwargs.items() if k != "properties"}
50+
kwargs_without_properties = {
51+
k: v for k, v in kwargs.items() if k not in ["properties", "fusion_type"]
52+
}
5153
extracted_params.update(extract_input_params(args, kwargs_without_properties))
52-
54+
if kwargs.get("fusion_type", None):
55+
extracted_params["fusion_type"] = kwargs["fusion_type"].value
5356
if kwargs.get("properties", None):
5457
extracted_params["properties"] = []
5558
for each_prop in kwargs.get("properties"):
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "2.3.4"
1+
__version__ = "2.3.5"

0 commit comments

Comments
 (0)