@@ -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