fix tests

Author: nikhilc-microsoft

tests/observability/extensions/agentframework/integration/test_observability_pipeline.py

@@ -19,7 +19,7 @@
 from opentelemetry.sdk.trace.export import SpanExporter, SpanExportResult
 
 try:
-    from agent_framework import RawAgent, ai_function
+    from agent_framework import RawAgent, tool as ai_function
     from agent_framework.azure import AzureOpenAIChatClient
     from agent_framework.observability import enable_instrumentation
     from azure.identity import AzureCliCredential
@@ -205,9 +205,8 @@ async def test_pipeline_invoke_agent_with_tool_call(
             request=request,
             scope_details=InvokeAgentScopeDetails(),
             agent_details=agent_details,
-        ) as invoke_scope:
+        ):
             result = await agent.run("What is 15 + 27?")
-            invoke_scope.set_output("42")
 
         assert result is not None
         assert len(result.text) > 0
@@ -352,9 +351,8 @@ async def test_pipeline_invoke_agent_simple_inference(
             request=request,
             scope_details=InvokeAgentScopeDetails(),
             agent_details=agent_details,
-        ) as invoke_scope:
+        ):
             result = await agent.run("Say hello in exactly 5 words.")
-            invoke_scope.set_output(result.text)
 
         assert result is not None
         assert len(result.text) > 0

tests/observability/extensions/langchain/integration/test_observability_pipeline.py

@@ -194,7 +194,7 @@ async def test_pipeline_invoke_agent_with_tool_call(
             request=request,
             scope_details=InvokeAgentScopeDetails(),
             agent_details=agent_details,
-        ) as invoke_scope:
+        ):
             messages = [
                 SystemMessage(
                     content=(
@@ -217,10 +217,8 @@ async def test_pipeline_invoke_agent_with_tool_call(
 
                 # Second LLM call with tool results
                 final_response = await llm_with_tools.ainvoke(messages)
-                invoke_scope.set_output(str(final_response.content))
             else:
                 final_response = ai_response
-                invoke_scope.set_output(str(final_response.content))
 
         assert final_response is not None
         assert len(str(final_response.content)) > 0
@@ -345,13 +343,12 @@ async def test_pipeline_invoke_agent_simple_inference(
             request=request,
             scope_details=InvokeAgentScopeDetails(),
             agent_details=agent_details,
-        ) as invoke_scope:
+        ):
             messages = [
                 SystemMessage(content="You are a helpful assistant. Reply in one sentence."),
                 HumanMessage(content="Say hello in exactly 5 words."),
             ]
             result = await llm.ainvoke(messages)
-            invoke_scope.set_output(str(result.content))
 
         assert result is not None
         assert len(str(result.content)) > 0