Skip to content

Commit 63ee9b7

Browse files
committed
Fixes formatting.
1 parent f1392b1 commit 63ee9b7

File tree

2 files changed

+7
-21
lines changed

2 files changed

+7
-21
lines changed

tests/observability/core/test_execute_tool_scope.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,7 @@ def setUpClass(cls):
3535
service_namespace="test-namespace",
3636
)
3737
# Create test data
38-
cls.tenant_details = TenantDetails(
39-
tenant_id="12345678-1234-5678-1234-567812345678"
40-
)
38+
cls.tenant_details = TenantDetails(tenant_id="12345678-1234-5678-1234-567812345678")
4139
cls.agent_details = AgentDetails(
4240
agent_id="test-agent-123",
4341
agent_name="Test Agent",
@@ -52,9 +50,7 @@ def setUpClass(cls):
5250

5351
def test_record_response_method_exists(self):
5452
"""Test that record_response method exists on ExecuteToolScope."""
55-
scope = ExecuteToolScope.start(
56-
self.tool_details, self.agent_details, self.tenant_details
57-
)
53+
scope = ExecuteToolScope.start(self.tool_details, self.agent_details, self.tenant_details)
5854

5955
if scope is not None:
6056
# Test that the method exists
@@ -73,9 +69,7 @@ def test_request_metadata_set_on_span(self):
7369
content="Execute tool with request metadata",
7470
execution_type=ExecutionType.AGENT_TO_AGENT,
7571
session_id="session-xyz",
76-
source_metadata=SourceMetadata(
77-
name="Channel 1", description="Link to channel"
78-
),
72+
source_metadata=SourceMetadata(name="Channel 1", description="Link to channel"),
7973
)
8074

8175
scope = ExecuteToolScope.start(

tests/observability/core/test_inference_scope.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,7 @@ def setUpClass(cls):
3737
)
3838
# Create test agent and tenant details
3939
cls.agent_details = AgentDetails(agent_id="test-inference-agent")
40-
cls.tenant_details = TenantDetails(
41-
tenant_id="12345678-1234-5678-1234-567812345678"
42-
)
40+
cls.tenant_details = TenantDetails(tenant_id="12345678-1234-5678-1234-567812345678")
4341

4442
def test_inference_operation_type_enum(self):
4543
"""Test InferenceOperationType enum values."""
@@ -116,9 +114,7 @@ def test_inference_scope_with_request(self):
116114
session_id="test-session-123",
117115
)
118116

119-
scope = InferenceScope.start(
120-
details, self.agent_details, self.tenant_details, request
121-
)
117+
scope = InferenceScope.start(details, self.agent_details, self.tenant_details, request)
122118

123119
# Test that scope can be created with request
124120
if scope is not None:
@@ -136,19 +132,15 @@ def test_request_metadata_set_on_span(self):
136132
content="Inference request with source metadata",
137133
execution_type=ExecutionType.AGENT_TO_AGENT,
138134
session_id="session-meta",
139-
source_metadata=SourceMetadata(
140-
name="Channel 1", description="Link to channel"
141-
),
135+
source_metadata=SourceMetadata(name="Channel 1", description="Link to channel"),
142136
)
143137

144138
span_exporter = InMemorySpanExporter()
145139
tracer_provider = TracerProvider()
146140
tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter))
147141
trace.set_tracer_provider(tracer_provider)
148142

149-
scope = InferenceScope.start(
150-
details, self.agent_details, self.tenant_details, request
151-
)
143+
scope = InferenceScope.start(details, self.agent_details, self.tenant_details, request)
152144

153145
if scope is not None:
154146
scope.dispose()

0 commit comments

Comments
 (0)