Skip to content

Commit

Permalink
Remove tracer and meter providers from type hints (#321)
Browse files Browse the repository at this point in the history
  • Loading branch information
Kludex authored Jul 17, 2024
1 parent 474f525 commit 02e511e
Show file tree
Hide file tree
Showing 5 changed files with 3 additions and 17 deletions.
5 changes: 1 addition & 4 deletions logfire/_internal/integrations/flask.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,7 @@
if TYPE_CHECKING:
from wsgiref.types import WSGIEnvironment

from opentelemetry.metrics import MeterProvider
from opentelemetry.trace import Span, TracerProvider
from opentelemetry.trace import Span
from typing_extensions import Protocol, TypedDict, Unpack

class RequestHook(Protocol):
Expand All @@ -21,11 +20,9 @@ def __call__(self, span: Span, status: str, response_headers: list[tuple[str, st
class FlaskInstrumentKwargs(TypedDict, total=False):
request_hook: RequestHook | None
response_hook: RequestHook | None
tracer_provider: TracerProvider | None
excluded_urls: str | None
enable_commenter: bool | None
commenter_options: dict[str, str] | None
meter_provider: MeterProvider | None


def instrument_flask(app: Flask, **kwargs: Unpack[FlaskInstrumentKwargs]):
Expand Down
1 change: 0 additions & 1 deletion logfire/_internal/integrations/psycopg.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@ class CommenterOptions(TypedDict, total=False):
class PsycopgInstrumentKwargs(TypedDict, total=False):
enable_commenter: bool
commenter_options: CommenterOptions
skip_dep_check: bool


PACKAGE_NAMES = ('psycopg', 'psycopg2')
Expand Down
4 changes: 0 additions & 4 deletions logfire/_internal/integrations/pymongo.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,6 @@
from opentelemetry.instrumentation.pymongo import PymongoInstrumentor

if TYPE_CHECKING:
from opentelemetry.metrics import MeterProvider
from opentelemetry.trace import TracerProvider
from pymongo.monitoring import CommandFailedEvent, CommandStartedEvent, CommandSucceededEvent
from typing_extensions import Protocol, TypedDict, Unpack

Expand All @@ -24,8 +22,6 @@ class PymongoInstrumentKwargs(TypedDict, total=False):
response_hook: ResponseHook | None
failed_hook: FailedHook | None
capture_statement: bool | None
tracer_provider: TracerProvider | None
meter_provider: MeterProvider | None
skip_dep_check: bool


Expand Down
5 changes: 1 addition & 4 deletions logfire/_internal/integrations/redis.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@
from opentelemetry.instrumentation.redis import RedisInstrumentor

if TYPE_CHECKING:
from opentelemetry.metrics import MeterProvider
from opentelemetry.trace import Span, TracerProvider
from opentelemetry.trace import Span
from redis import Connection
from typing_extensions import Protocol, TypedDict, Unpack

Expand All @@ -19,8 +18,6 @@ def __call__(self, span: Span, instance: Connection, response: Any) -> None: ...
class RedisInstrumentKwargs(TypedDict, total=False):
request_hook: RequestHook | None
response_hook: ResponseHook | None
tracer_provider: TracerProvider | None
meter_provider: MeterProvider | None
skip_dep_check: bool


Expand Down
5 changes: 1 addition & 4 deletions logfire/_internal/integrations/starlette.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,7 @@
from starlette.applications import Starlette

if TYPE_CHECKING:
from opentelemetry.metrics import MeterProvider
from opentelemetry.trace import Span, TracerProvider
from opentelemetry.trace import Span
from typing_extensions import Protocol, TypedDict, Unpack

class ServerRequestHook(Protocol):
Expand All @@ -23,8 +22,6 @@ class StarletteInstrumentKwargs(TypedDict, total=False):
server_request_hook: ServerRequestHook | None
client_request_hook: ClientRequestHook | None
client_response_hook: ClientResponseHook | None
tracer_provider: TracerProvider | None
meter_provider: MeterProvider | None


def instrument_starlette(app: Starlette, **kwargs: Unpack[StarletteInstrumentKwargs]):
Expand Down

0 comments on commit 02e511e

Please # to comment.