exclude unused/private methods from langchain core

Author: alizenhom

src/langtrace_python_sdk/instrumentation/langchain_core/instrumentation.py

@@ -66,8 +66,14 @@ def patch_module_classes(
         if name.startswith("_") or name in exclude_classes:
             continue
         # loop through all public methods of the class
-        for method_name, method in inspect.getmembers(obj, predicate=inspect.isfunction):
-            if method_name in exclude_methods or method.__qualname__.split('.')[0] != name:
+        for method_name, method in inspect.getmembers(
+            obj, predicate=inspect.isfunction
+        ):
+            if (
+                method_name in exclude_methods
+                or method.__qualname__.split(".")[0] != name
+                or method_name.startswith("_")
+            ):
                 continue
             try:
                 method_path = f"{name}.{method_name}"
@@ -108,6 +114,9 @@ def _instrument(self, **kwargs):
             "format",
             "format_messages",
             "format_prompt",
+            "__or__",
+            "__init__",
+            "__repr__",
         ]
         exclude_classes = [
             "BaseChatPromptTemplate",
@@ -125,7 +134,13 @@ def _instrument(self, **kwargs):
         modules_to_patch = [
             ("langchain_core.retrievers", "retriever", generic_patch, True, True),
             ("langchain_core.prompts.chat", "prompt", generic_patch, True, True),
-            ("langchain_core.language_models.llms", "generate", generic_patch, True, True),
+            (
+                "langchain_core.language_models.llms",
+                "generate",
+                generic_patch,
+                True,
+                True,
+            ),
             ("langchain_core.runnables.base", "runnable", runnable_patch, True, True),
             (
                 "langchain_core.runnables.passthrough",