Test with recording for ai search and function tool samples (#44249)

* Test with recording for ai search and function tool samples

* re-record

Author: howieleung

sdk/ai/azure-ai-projects/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ai/azure-ai-projects",
-  "Tag": "python/ai/azure-ai-projects_2204ef492a"
+  "Tag": "python/ai/azure-ai-projects_d20bfd0fdc"
 }

sdk/ai/azure-ai-projects/dev_requirements.txt

@@ -7,4 +7,4 @@ opentelemetry-sdk
 azure-core-tracing-opentelemetry
 azure-monitor-opentelemetry
 azure-mgmt-cognitiveservices
-
+jsonref

sdk/ai/azure-ai-projects/samples/agents/tools/computer_use_util.py

@@ -159,4 +159,4 @@ def print_final_output(response):
                 final_output += getattr(part, "text", None) or getattr(part, "refusal", None) or "" + "\n"
 
     print(f"Final status: {response.status}")
-    print(f"Final output: {final_output.strip()}")
+    print(f"==> Result: {final_output.strip()}")

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_ai_search.py

@@ -24,6 +24,7 @@
     3) AI_SEARCH_PROJECT_CONNECTION_ID - The AI Search project connection ID,
        as found in the "Connections" tab in your Microsoft Foundry project.
     4) AI_SEARCH_INDEX_NAME - The name of the AI Search index to use for searching.
+    5) AI_SEARCH_USER_INPUT - (Optional) The question to ask. If not set, you will be prompted.
 """
 
 import os
@@ -74,7 +75,10 @@
     )
     print(f"Agent created (id: {agent.id}, name: {agent.name}, version: {agent.version})")
 
-    user_input = input("Enter your question (e.g., 'Tell me about mental health services'): \n")
+    # Get user input from environment variable or prompt
+    user_input = os.environ.get("AI_SEARCH_USER_INPUT")
+    if not user_input:
+        user_input = input("Enter your question (e.g., 'Tell me about mental health services'): \n")
 
     stream_response = openai_client.responses.create(
         stream=True,
@@ -104,7 +108,7 @@
                             )
         elif event.type == "response.completed":
             print(f"\nFollow-up completed!")
-            print(f"Full response: {event.response.output_text}")
+            print(f"==> Result: {event.response.output_text}")
 
     print("\nCleaning up...")
     project_client.agents.delete_version(agent_name=agent.name, agent_version=agent.version)

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_code_interpreter.py

@@ -100,10 +100,12 @@
     # Download the generated file if available
     if file_id and filename:
         file_content = openai_client.containers.files.content.retrieve(file_id=file_id, container_id=container_id)
+        print(f"File ready for download: {filename}")
         with open(filename, "wb") as f:
             f.write(file_content.read())
-            print(f"File {filename} downloaded successfully.")
-        print(f"File ready for download: {filename}")
+
+        # Print result (should contain "file")
+        print(f"==> Result: file, {filename} downloaded successfully.")
     else:
         print("No file generated in response")
 

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_code_interpreter_async.py

@@ -103,10 +103,12 @@ async def main() -> None:
                 file_content = await openai_client.containers.files.content.retrieve(
                     file_id=file_id, container_id=container_id
                 )
+                print(f"File ready for download: {filename}")
                 with open(filename, "wb") as f:
                     f.write(file_content.read())
-                    print(f"File {filename} downloaded successfully.")
-                print(f"File ready for download: {filename}")
+
+                # Print result (should contain "file")
+                print(f"==> Result: file, {filename} downloaded successfully.")
             else:
                 print("No file generated in response")
 

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_computer_use.py

@@ -25,7 +25,7 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_MODEL_DEPLOYMENT_NAME - The deployment name of the AI model, as found under the "Name" column in
+    2) (Optional) COMPUTER_USE_MODEL_DEPLOYMENT_NAME - The deployment name of the computer-use-preview model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -70,7 +70,7 @@
     agent = project_client.agents.create_version(
         agent_name="ComputerUseAgent",
         definition=PromptAgentDefinition(
-            model=os.environ["AZURE_AI_MODEL_DEPLOYMENT_NAME"],
+            model=os.environ.get("COMPUTER_USE_MODEL_DEPLOYMENT_NAME", "computer-use-preview"),
             instructions="""
             You are a computer automation assistant. 
             

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_computer_use_async.py

@@ -25,7 +25,7 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_MODEL_DEPLOYMENT_NAME - The deployment name of the AI model, as found under the "Name" column in
+    2) (Optional) COMPUTER_USE_MODEL_DEPLOYMENT_NAME - The deployment name of the computer-use-preview model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -72,7 +72,7 @@ async def main():
         agent = await project_client.agents.create_version(
             agent_name="ComputerUseAgent",
             definition=PromptAgentDefinition(
-                model=os.environ["AZURE_AI_MODEL_DEPLOYMENT_NAME"],
+                model=os.environ.get("COMPUTER_USE_MODEL_DEPLOYMENT_NAME", "computer-use-preview"),
                 instructions="""
                 You are a computer automation assistant. 
                 

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_file_search.py

@@ -78,8 +78,7 @@
         input="Tell me about Contoso products",
         extra_body={"agent": {"name": agent.name, "type": "agent_reference"}},
     )
-    print(f"Response: {response.output_text}")
-
+    print(f"==> Result: {response.output_text}")
     print("\nCleaning up...")
     project_client.agents.delete_version(agent_name=agent.name, agent_version=agent.version)
     print("Agent deleted")

sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_file_search_in_stream.py

@@ -140,7 +140,7 @@
                             print(f"File Citation - Filename: {annotation.filename}, File ID: {annotation.file_id}")
         elif event.type == "response.completed":
             print(f"\nFollow-up completed!")
-            print(f"Full response: {event.response.output_text}")
+            print(f"==> Result: {event.response.output_text}")
 
     # Clean up resources
     print("\n" + "=" * 60)