Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
nicolasgere committed Mar 6, 2024
1 parent f109565 commit 797fd33
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 6 deletions.
3 changes: 1 addition & 2 deletions chromadb/api/segment.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,8 +111,7 @@ def __init__(self, system: System):
@override
def heartbeat(self) -> int:
return int(time.time_ns())

@trace_method("SegmentAPI.create_database", OpenTelemetryGranularity.OPERATION)

@override
def create_database(self, name: str, tenant: str = DEFAULT_TENANT) -> None:
if len(name) < 3:
Expand Down
3 changes: 1 addition & 2 deletions chromadb/db/impl/grpc/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,7 @@ def __init__(self, system: System):
self._coordinator_url = system.settings.require("chroma_coordinator_host")
# TODO: break out coordinator_port into a separate setting?
self._coordinator_port = system.settings.require("chroma_server_grpc_port")
self._opentelemetry_client = system.require(OpenTelemetryClient)
super().__init__(system)
return super().__init__(system)

@overrides
def start(self) -> None:
Expand Down
2 changes: 0 additions & 2 deletions chromadb/telemetry/opentelemetry/grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,6 @@ class OtelInterceptor(
grpc.StreamStreamClientInterceptor
):
def _intercept_call(self, continuation, client_call_details, request_or_iterator):
if tracer is None:
return continuation(client_call_details, request_or_iterator)
with tracer.start_as_current_span(f"RPC {client_call_details.method}", kind=SpanKind.CLIENT) as span:
# Prepare metadata for propagation
metadata = client_call_details.metadata[:] if client_call_details.metadata else []
Expand Down

0 comments on commit 797fd33

Please sign in to comment.