traceAI-langchain 0.2.1__tar.gz → 0.2.2__tar.gz

{traceai_langchain-0.2.1 → traceai_langchain-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traceAI-langchain
-Version: 0.2.1
+Version: 0.2.2
 Summary: OpenTelemetry instrumentation for LangChain
 License: Apache-2.0
 Keywords: traceai,futureagi,opentelemetry,tracing,observability,llm,langchain

{traceai_langchain-0.2.1 → traceai_langchain-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "traceAI-langchain"
-version = "0.2.1"
+version = "0.2.2"
 description = "OpenTelemetry instrumentation for LangChain"
 authors = ["Future AGI <no-reply@futureagi.com>"]
 readme = "README.md"

{traceai_langchain-0.2.1 → traceai_langchain-0.2.2}/traceai_langchain/__init__.py

@@ -70,6 +70,7 @@ class LangChainInstrumentor(BaseInstrumentor):  # type: ignore
             name="BaseCallbackManager.__init__",
             wrapper=_BaseCallbackManagerInit(self._tracer),
         )
+        self._protect_class = Protect
         if Protect is not None:
             self._original_protect = Protect.protect
             wrap_function_wrapper(
@@ -83,6 +84,10 @@ class LangChainInstrumentor(BaseInstrumentor):  # type: ignore
     def _uninstrument(self, **kwargs: Any) -> None:
         langchain_core.callbacks.BaseCallbackManager.__init__ = self._original_callback_manager_init  # type: ignore
         self._original_callback_manager_init = None  # type: ignore
+        if self._protect_class is not None:
+            self._protect_class.protect = self._original_protect
+        self._original_protect = None
+        self._protect_class = None
         self._tracer = None
         self.fi_tracer = None
 

traceai_langchain-0.2.2/traceai_langchain/version.py

@@ -0,0 +1 @@
+__version__ = "0.2.2"  # x-release-please-version

traceai_langchain-0.2.1/traceai_langchain/version.py

@@ -1 +0,0 @@
-__version__ = "0.2.1"  # x-release-please-version