feat:remove thread pool wrap

Change-Id: Idf5110cfde8789737ab0f984ba7c36ac2eae151c
Co-developed-by: Cursor <noreply@cursor.com>

Author: sunym1991

instrumentation-loongsuite/loongsuite-instrumentation-mem0/README.md

@@ -114,7 +114,7 @@ conventions** and should be treated as experimental.
 
 ## Compatibility
 
-- Python: `>= 3.8, < 3.13`
+- Python: `>= 3.8, <= 3.13`
 - Mem0 / `mem0ai`: `>= 1.0.0`
 - OpenTelemetry API: `>= 1.20.0`
 

instrumentation-loongsuite/loongsuite-instrumentation-mem0/pyproject.toml

@@ -25,10 +25,10 @@ classifiers = [
     "Programming Language :: Python :: 3.13",
 ]
 dependencies = [
-    "wrapt",
-    "opentelemetry-api",
-    "opentelemetry-instrumentation",
-    "opentelemetry-semantic-conventions",
+    "wrapt >=1.17.3,<2",
+    "opentelemetry-api ~=1.37",
+    "opentelemetry-instrumentation ~=0.58b0",
+    "opentelemetry-semantic-conventions ~=0.58b0",
 ]
 
 [project.optional-dependencies]

instrumentation-loongsuite/loongsuite-instrumentation-mem0/src/opentelemetry/instrumentation/mem0/__init__.py

@@ -16,9 +16,6 @@
 from opentelemetry.instrumentation.mem0.internal._extractors import (
     METHOD_EXTRACTION_RULES,
 )
-from opentelemetry.instrumentation.mem0.internal._thread_pool_handler import (
-    ThreadPoolContextPropagationHandler,
-)
 from opentelemetry.instrumentation.mem0.internal._wrapper import (
     GraphStoreWrapper,
     MemoryOperationWrapper,
@@ -133,17 +130,6 @@ def _instrument(self, **kwargs: Any) -> None:
         )
 
         # Wrap ThreadPoolExecutor.submit to support context propagation
-        # Some mem0 methods (add, get_all, search) use ThreadPoolExecutor,
-        # causing OpenTelemetry context to not auto-propagate to child threads.
-        # We use wrapt's standard wrapping mechanism for proper cleanup support.
-
-        self._threadpool_handler = ThreadPoolContextPropagationHandler()
-        wrap_function_wrapper(
-            module="concurrent.futures",
-            name="ThreadPoolExecutor.submit",
-            wrapper=self._threadpool_handler,
-        )
-
         # Execute instrumentation (traces only, metrics removed)
         self._instrument_memory_operations(tracer)
         self._instrument_memory_client_operations(tracer)
@@ -159,9 +145,6 @@ def _uninstrument(self, **kwargs: Any) -> None:
         if not self._is_instrumented:
             return
 
-        # Unwrap ThreadPoolExecutor.submit
-        self._uninstrument_threadpool()
-
         # Uninstrument Memory and MemoryClient operations (symmetric to _instrument)
         self._uninstrument_memory_operations()
         self._uninstrument_memory_client_operations()
@@ -174,16 +157,6 @@ def _uninstrument(self, **kwargs: Any) -> None:
         # Reset instrumentation state
         self._is_instrumented = False
 
-    def _uninstrument_threadpool(self) -> None:
-        """Unwrap ThreadPoolExecutor.submit."""
-        try:
-            import concurrent.futures
-
-            unwrap(concurrent.futures.ThreadPoolExecutor, "submit")
-            logger.debug("Successfully unwrapped ThreadPoolExecutor.submit")
-        except Exception as e:
-            logger.debug(f"Failed to unwrap ThreadPoolExecutor.submit: {e}")
-
     def _uninstrument_memory_operations(self) -> None:
         """Uninstrument Memory and AsyncMemory operations."""
         if not _MEM0_CORE_AVAILABLE or Memory is None or AsyncMemory is None:

instrumentation-loongsuite/loongsuite-instrumentation-mem0/src/opentelemetry/instrumentation/mem0/internal/_extractors.py

@@ -88,6 +88,41 @@ class MethodExtractionRule:
 }
 
 
+GENERIC_MEMORY_ID_PAGINATION_SEARCH_SPEC: List[Tuple[str, str, Any]] = [
+    ("infer", SemanticAttributes.GEN_AI_MEMORY_INFER, bool),
+    (
+        "memory_type",
+        SemanticAttributes.GEN_AI_MEMORY_MEMORY_TYPE,
+        safe_str,
+    ),
+    ("memory_id", SemanticAttributes.GEN_AI_MEMORY_ID, safe_str),
+    ("limit", SemanticAttributes.GEN_AI_MEMORY_LIMIT, safe_int),
+    ("page", SemanticAttributes.GEN_AI_MEMORY_PAGE, safe_int),
+    (
+        "page_size",
+        SemanticAttributes.GEN_AI_MEMORY_PAGE_SIZE,
+        safe_int,
+    ),
+    ("top_k", SemanticAttributes.GEN_AI_MEMORY_TOP_K, safe_int),
+    (
+        "threshold",
+        SemanticAttributes.GEN_AI_MEMORY_THRESHOLD,
+        safe_float,
+    ),
+    ("rerank", SemanticAttributes.GEN_AI_MEMORY_RERANK, bool),
+    (
+        "only_metadata_based_search",
+        SemanticAttributes.GEN_AI_MEMORY_ONLY_METADATA_BASED_SEARCH,
+        bool,
+    ),
+    (
+        "keyword_search",
+        SemanticAttributes.GEN_AI_MEMORY_KEYWORD_SEARCH,
+        bool,
+    ),
+]
+
+
 def _extract_input_content(
     operation_name: str, kwargs: Dict[str, Any]
 ) -> Optional[str]:
@@ -583,39 +618,7 @@ def extract_generic_attributes(
             attributes,
             instance=None,
             kwargs=kwargs,
-            spec=[
-                ("infer", SemanticAttributes.GEN_AI_MEMORY_INFER, bool),
-                (
-                    "memory_type",
-                    SemanticAttributes.GEN_AI_MEMORY_MEMORY_TYPE,
-                    safe_str,
-                ),
-                ("memory_id", SemanticAttributes.GEN_AI_MEMORY_ID, safe_str),
-                ("limit", SemanticAttributes.GEN_AI_MEMORY_LIMIT, safe_int),
-                ("page", SemanticAttributes.GEN_AI_MEMORY_PAGE, safe_int),
-                (
-                    "page_size",
-                    SemanticAttributes.GEN_AI_MEMORY_PAGE_SIZE,
-                    safe_int,
-                ),
-                ("top_k", SemanticAttributes.GEN_AI_MEMORY_TOP_K, safe_int),
-                (
-                    "threshold",
-                    SemanticAttributes.GEN_AI_MEMORY_THRESHOLD,
-                    safe_float,
-                ),
-                ("rerank", SemanticAttributes.GEN_AI_MEMORY_RERANK, bool),
-                (
-                    "only_metadata_based_search",
-                    SemanticAttributes.GEN_AI_MEMORY_ONLY_METADATA_BASED_SEARCH,
-                    bool,
-                ),
-                (
-                    "keyword_search",
-                    SemanticAttributes.GEN_AI_MEMORY_KEYWORD_SEARCH,
-                    bool,
-                ),
-            ],
+            spec=GENERIC_MEMORY_ID_PAGINATION_SEARCH_SPEC,
         )
 
         # List parameters: fields / categories

instrumentation-loongsuite/loongsuite-instrumentation-mem0/src/opentelemetry/instrumentation/mem0/internal/_thread_pool_handler.py

@@ -12,6 +12,7 @@ wrapt==1.17.3
 opentelemetry-api==1.37
 opentelemetry-sdk==1.37
 opentelemetry-semantic-conventions==0.58b0
+opentelemetry-instrumentation-threading==0.58b0
 
 # Mock and utilities
 mock>=4.0.0