Skip to content

Commit fbbdcf9

Browse files
committed
refactor(client, session): optional ClientTelemetry, default retrieval from singleton instance
1 parent fb11e95 commit fbbdcf9

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

agentops/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,7 @@ def start_session(
236236
session_tags.update(tags)
237237

238238
session = Session(
239+
client=self._telemetry,
239240
session_id=session_id,
240241
tags=list(session_tags),
241242
host_env=self.host_env,

agentops/session.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from typing import Any, Dict, List, Optional, Sequence, Union
1010
from uuid import UUID, uuid4
1111

12+
from agentops.client import Client
1213
from opentelemetry import trace
1314
from opentelemetry.context import attach, detach, set_value
1415
from opentelemetry.sdk.resources import SERVICE_NAME, Resource
@@ -201,7 +202,7 @@ def __init__(
201202
self,
202203
session_id: UUID,
203204
config: Configuration,
204-
client: ClientTelemetry,
205+
client: Optional[ClientTelemetry] = Client()._telemetry, # Not mandatory, we can use the Client singleton to retrieve the telemetry client
205206
tags: Optional[List[str]] = None,
206207
host_env: Optional[dict] = None,
207208
):

0 commit comments

Comments
 (0)