Skip to content

Commit 31b875f

Browse files
committed
running tox and isort to format imports
1 parent 13944b3 commit 31b875f

File tree

6 files changed

+29
-35
lines changed

6 files changed

+29
-35
lines changed

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

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from typing import Collection
2-
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
3-
from wrapt import wrap_function_wrapper
42

5-
from opentelemetry.trace import get_tracer
6-
from opentelemetry.instrumentation.utils import unwrap
3+
from wrapt import wrap_function_wrapper
74

8-
from opentelemetry.instrumentation.langchain_v2.version import __version__
5+
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
96
from opentelemetry.instrumentation.langchain_v2.callback_handler import OpenTelemetryCallbackHandler
7+
from opentelemetry.instrumentation.langchain_v2.version import __version__
8+
from opentelemetry.instrumentation.utils import unwrap
9+
from opentelemetry.trace import get_tracer
1010

1111
__all__ = ["OpenTelemetryCallbackHandler"]
1212

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

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,20 @@
11
import time
22
from dataclasses import dataclass, field
33
from typing import Any, Optional
4-
from langchain_core.callbacks import (
5-
BaseCallbackHandler,
6-
)
4+
from uuid import UUID
75

6+
from langchain_core.agents import AgentAction, AgentFinish
7+
from langchain_core.callbacks import BaseCallbackHandler
88
from langchain_core.messages import BaseMessage
99
from langchain_core.outputs import LLMResult
10-
from opentelemetry.context.context import Context
11-
from opentelemetry.trace import SpanKind, set_span_in_context
12-
from opentelemetry.trace.span import Span
13-
from opentelemetry.util.types import AttributeValue
14-
from uuid import UUID
1510

1611
from opentelemetry import context as context_api
12+
from opentelemetry.instrumentation.langchain_v2.span_attributes import GenAIOperationValues, Span_Attributes
1713
from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY
18-
19-
from langchain_core.agents import AgentAction, AgentFinish
20-
21-
from opentelemetry.instrumentation.langchain_v2.span_attributes import Span_Attributes, GenAIOperationValues
14+
from opentelemetry.trace import SpanKind, set_span_in_context
15+
from opentelemetry.trace.span import Span
2216
from opentelemetry.trace.status import Status, StatusCode
17+
from opentelemetry.util.types import AttributeValue
2318

2419

2520
@dataclass

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/tests/conftest.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,12 @@
1-
from opentelemetry.sdk.trace import TracerProvider
2-
from opentelemetry.sdk.trace.export import SimpleSpanProcessor
3-
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
4-
51
import os
6-
import pytest
7-
8-
from opentelemetry.sdk._logs.export import (
9-
InMemoryLogExporter,
10-
)
112

3+
import pytest
124
from src.opentelemetry.instrumentation.langchain_v2 import LangChainInstrumentor
135

6+
from opentelemetry.sdk._logs.export import InMemoryLogExporter
7+
from opentelemetry.sdk.trace import TracerProvider
8+
from opentelemetry.sdk.trace.export import SimpleSpanProcessor
9+
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
1410

1511
OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT = "OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT"
1612

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/tests/test_agents.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
11
import os
2-
from typing import Tuple
32

3+
import boto3
44
import pytest
55
from langchain import hub
6-
from langchain_aws import ChatBedrock
76
from langchain.agents import AgentExecutor, create_tool_calling_agent
7+
from langchain_aws import ChatBedrock
88
from langchain_community.tools import DuckDuckGoSearchResults
9-
import boto3
109

1110

1211
@pytest.mark.vcr(filter_headers=["Authorization", "X-Amz-Date", "X-Amz-Security-Token"], record_mode="all")

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/tests/test_chains.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
1-
import os
21
import ast
2+
import os
3+
4+
import boto3
35
import pytest
6+
from langchain.chains import LLMChain, SequentialChain
47
from langchain.prompts import PromptTemplate
5-
import boto3
68
from langchain_aws import BedrockLLM
7-
from langchain.chains import LLMChain, SequentialChain
9+
810
from opentelemetry.trace import SpanKind
911

1012

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/tests/test_langgraph_agent.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
1-
import pytest
21
import os
3-
from langchain_aws import ChatBedrock
4-
import boto3
52
from typing import TypedDict
3+
4+
import boto3
5+
import pytest
6+
from langchain_aws import ChatBedrock
67
from langgraph.graph import StateGraph
8+
79
from opentelemetry import trace
810
from opentelemetry.trace import INVALID_SPAN
911

0 commit comments

Comments
 (0)