From 9c3673e226c1707dc1d6947fcbb34a5582395be2 Mon Sep 17 00:00:00 2001
From: Ali Waleed <ali@scale3labs.com>
Date: Tue, 24 Sep 2024 18:42:35 +0300
Subject: [PATCH] Weaviate try and catch

---
 .../weaviate/instrumentation.py               | 34 +++++++++++--------
 src/langtrace_python_sdk/version.py           |  2 +-
 2 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/src/langtrace_python_sdk/instrumentation/weaviate/instrumentation.py b/src/langtrace_python_sdk/instrumentation/weaviate/instrumentation.py
index b51a3acd..ba31d5a9 100644
--- a/src/langtrace_python_sdk/instrumentation/weaviate/instrumentation.py
+++ b/src/langtrace_python_sdk/instrumentation/weaviate/instrumentation.py
@@ -43,20 +43,26 @@ def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
         tracer = get_tracer(__name__, "", tracer_provider)
         version = importlib.metadata.version("weaviate-client")
-
-        for api_name, api_config in APIS.items():
-            if api_config.get("OPERATION") in ["query", "generate"]:
-                wrap_function_wrapper(
-                    api_config["MODULE"],
-                    api_config["METHOD"],
-                    generic_query_patch(api_name, version, tracer),
-                )
-            elif api_config.get("OPERATION") == "create":
-                wrap_function_wrapper(
-                    api_config["MODULE"],
-                    api_config["METHOD"],
-                    generic_collection_patch(api_name, version, tracer),
-                )
+        try:
+            for api_name, api_config in APIS.items():
+                if api_config.get("OPERATION") in ["query", "generate"]:
+                    wrap_function_wrapper(
+                        api_config["MODULE"],
+                        api_config["METHOD"],
+                        generic_query_patch(api_name, version, tracer),
+                    )
+                elif api_config.get("OPERATION") == "create":
+                    print(
+                        api_config["MODULE"],
+                        api_config["METHOD"],
+                    )
+                    wrap_function_wrapper(
+                        api_config["MODULE"],
+                        api_config["METHOD"],
+                        generic_collection_patch(api_name, version, tracer),
+                    )
+        except ModuleNotFoundError as e:
+            pass
 
     def _instrument_module(self, module_name):
         pass
diff --git a/src/langtrace_python_sdk/version.py b/src/langtrace_python_sdk/version.py
index 2ac9e8ce..61e5ea59 100644
--- a/src/langtrace_python_sdk/version.py
+++ b/src/langtrace_python_sdk/version.py
@@ -1 +1 @@
-__version__ = "2.3.20"
+__version__ = "2.3.21"