From d2c826759e17012fd4257cc84bbf55ad6ff1b697 Mon Sep 17 00:00:00 2001
From: darshit-s3 <119623510+darshit-s3@users.noreply.github.com>
Date: Thu, 5 Sep 2024 19:20:45 +0530
Subject: [PATCH] fix: weaviate fusion type enum handling (#328)

---
 src/langtrace_python_sdk/instrumentation/weaviate/patch.py | 7 +++++--
 src/langtrace_python_sdk/version.py                        | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/langtrace_python_sdk/instrumentation/weaviate/patch.py b/src/langtrace_python_sdk/instrumentation/weaviate/patch.py
index c2ea4b39..523b047a 100644
--- a/src/langtrace_python_sdk/instrumentation/weaviate/patch.py
+++ b/src/langtrace_python_sdk/instrumentation/weaviate/patch.py
@@ -47,9 +47,12 @@
 
 def extract_inputs(args, kwargs):
     extracted_params = {}
-    kwargs_without_properties = {k: v for k, v in kwargs.items() if k != "properties"}
+    kwargs_without_properties = {
+        k: v for k, v in kwargs.items() if k not in ["properties", "fusion_type"]
+    }
     extracted_params.update(extract_input_params(args, kwargs_without_properties))
-
+    if kwargs.get("fusion_type", None):
+        extracted_params["fusion_type"] = kwargs["fusion_type"].value
     if kwargs.get("properties", None):
         extracted_params["properties"] = []
         for each_prop in kwargs.get("properties"):
diff --git a/src/langtrace_python_sdk/version.py b/src/langtrace_python_sdk/version.py
index 4618fe65..f4bd92b2 100644
--- a/src/langtrace_python_sdk/version.py
+++ b/src/langtrace_python_sdk/version.py
@@ -1 +1 @@
-__version__ = "2.3.4"
+__version__ = "2.3.5"