Skip to content

Commit 9dab168

Browse files
committed
formatted using black
1 parent 1f549b0 commit 9dab168

File tree

10 files changed

+318
-434
lines changed

10 files changed

+318
-434
lines changed

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/pyproject.toml

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,16 @@ instruments = [
3434
"langchain >= 0.3.21",
3535
]
3636

37-
[tool.hatch.build.targets.wheel]
38-
packages = ["/src/amazon/opentelemetry"]
37+
3938

4039
[tool.hatch.version]
41-
path = "version.py"
40+
path = "src/opentelemetry/instrumentation/langchain_v2/version.py"
4241

42+
[tool.hatch.build.targets.sdist]
43+
include = [
44+
"/src",
45+
"/tests",
46+
]
47+
48+
[tool.hatch.build.targets.wheel]
49+
packages = ["src"]

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/src/opentelemetry/instrumentation/langchain_v2/__init__.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,35 +12,36 @@
1212

1313
_instruments = ("langchain >= 0.1.0",)
1414

15+
1516
class LangChainInstrumentor(BaseInstrumentor):
16-
17+
1718
def instrumentation_dependencies(cls) -> Collection[str]:
1819
return _instruments
19-
20+
2021
def _instrument(self, **kwargs):
2122
tracer_provider = kwargs.get("tracer_provider")
2223
tracer = get_tracer(__name__, __version__, tracer_provider)
2324

2425
otelCallbackHandler = OpenTelemetryCallbackHandler(tracer)
25-
26+
2627
wrap_function_wrapper(
2728
module="langchain_core.callbacks",
2829
name="BaseCallbackManager.__init__",
2930
wrapper=_BaseCallbackManagerInitWrapper(otelCallbackHandler),
3031
)
31-
32+
3233
def _uninstrument(self, **kwargs):
3334
unwrap("langchain_core.callbacks", "BaseCallbackManager.__init__")
3435
if hasattr(self, "_wrapped"):
3536
for module, name in self._wrapped:
3637
unwrap(module, name)
37-
38-
38+
39+
3940
class _BaseCallbackManagerInitWrapper:
4041
def __init__(self, callback_handler: "OpenTelemetryCallbackHandler"):
4142
self.callback_handler = callback_handler
4243
self._wrapped = []
43-
44+
4445
def __call__(
4546
self,
4647
wrapped,
@@ -53,4 +54,4 @@ def __call__(
5354
if isinstance(handler, OpenTelemetryCallbackHandler):
5455
return None
5556
else:
56-
instance.add_handler(self.callback_handler, True)
57+
instance.add_handler(self.callback_handler, True)

0 commit comments

Comments
 (0)