Merge pull request #361 from Scale3-Labs/development

Release: Bugfix for weaviate module not found

Author: alizenhom

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

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.3.20"
+__version__ = "2.3.21"