Capture file IDs from code interpreter in streaming responses (#2741)

python/samples/getting_started/agents/azure_ai_agent/README.md

@@ -9,6 +9,7 @@ This folder contains examples demonstrating different ways to create and use age
 | [`azure_ai_basic.py`](azure_ai_basic.py) | The simplest way to create an agent using `ChatAgent` with `AzureAIAgentClient`. It automatically handles all configuration using environment variables. |
 | [`azure_ai_with_bing_custom_search.py`](azure_ai_with_bing_custom_search.py) | Shows how to use Bing Custom Search with Azure AI agents to find real-time information from the web using custom search configurations. Demonstrates how to set up and use HostedWebSearchTool with custom search instances. |
 | [`azure_ai_with_bing_grounding.py`](azure_ai_with_bing_grounding.py) | Shows how to use Bing Grounding search with Azure AI agents to find real-time information from the web. Demonstrates web search capabilities with proper source citations and comprehensive error handling. |
+| [`azure_ai_with_code_interpreter_file_generation.py`](azure_ai_with_code_interpreter_file_generation.py) | Shows how to retrieve file IDs from code interpreter generated files using both streaming and non-streaming approaches. |
 | [`azure_ai_with_code_interpreter.py`](azure_ai_with_code_interpreter.py) | Shows how to use the HostedCodeInterpreterTool with Azure AI agents to write and execute Python code. Includes helper methods for accessing code interpreter data from response chunks. |
 | [`azure_ai_with_existing_agent.py`](azure_ai_with_existing_agent.py) | Shows how to work with a pre-existing agent by providing the agent ID to the Azure AI chat client. This example also demonstrates proper cleanup of manually created agents. |
 | [`azure_ai_with_existing_thread.py`](azure_ai_with_existing_thread.py) | Shows how to work with a pre-existing thread by providing the thread ID to the Azure AI chat client. This example also demonstrates proper cleanup of manually created threads. |

python/samples/getting_started/agents/azure_ai_agent/azure_ai_with_code_interpreter_file_generation.py

@@ -0,0 +1,102 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+import asyncio
+
+from agent_framework import AgentRunResponseUpdate, ChatAgent, HostedCodeInterpreterTool, HostedFileContent
+from agent_framework.azure import AzureAIAgentClient
+from azure.identity.aio import AzureCliCredential
+
+"""
+Azure AI Agent Code Interpreter File Generation Example
+
+This sample demonstrates using HostedCodeInterpreterTool with AzureAIAgentClient
+to generate a text file and then retrieve it.
+
+The test flow:
+1. Create an agent with code interpreter tool
+2. Ask the agent to generate a txt file using Python code
+3. Capture the file_id from HostedFileContent in the response
+4. Retrieve the file using the agents_client.files API
+"""
+
+
+async def main() -> None:
+    """Test file generation and retrieval with code interpreter."""
+
+    async with AzureCliCredential() as credential:
+        client = AzureAIAgentClient(credential=credential)
+
+        try:
+            async with ChatAgent(
+                chat_client=client,
+                instructions=(
+                    "You are a Python code execution assistant. "
+                    "ALWAYS use the code interpreter tool to execute Python code when asked to create files. "
+                    "Write actual Python code to create files, do not just describe what you would do."
+                ),
+                tools=[HostedCodeInterpreterTool()],
+            ) as agent:
+                # Be very explicit about wanting code execution and a download link
+                query = (
+                    "Use the code interpreter to execute this Python code and then provide me "
+                    "with a download link for the generated file:\n"
+                    "```python\n"
+                    "with open('/mnt/data/sample.txt', 'w') as f:\n"
+                    "    f.write('Hello, World! This is a test file.')\n"
+                    "'/mnt/data/sample.txt'\n"  # Return the path so it becomes downloadable
+                    "```"
+                )
+                print(f"User: {query}\n")
+                print("=" * 60)
+
+                # Collect file_ids from the response
+                file_ids: list[str] = []
+
+                async for chunk in agent.run_stream(query):
+                    if not isinstance(chunk, AgentRunResponseUpdate):
+                        continue
+
+                    for content in chunk.contents:
+                        if content.type == "text":
+                            print(content.text, end="", flush=True)
+                        elif content.type == "hosted_file":
+                            if isinstance(content, HostedFileContent):
+                                file_ids.append(content.file_id)
+                                print(f"\n[File generated: {content.file_id}]")
+
+                print("\n" + "=" * 60)
+
+                # Attempt to retrieve discovered files
+                if file_ids:
+                    print(f"\nAttempting to retrieve {len(file_ids)} file(s):")
+                    for file_id in file_ids:
+                        try:
+                            file_info = await client.agents_client.files.get(file_id)
+                            print(f"  File {file_id}: Retrieved successfully")
+                            print(f"    Filename: {file_info.filename}")
+                            print(f"    Purpose: {file_info.purpose}")
+                            print(f"    Bytes: {file_info.bytes}")
+                        except Exception as e:
+                            print(f"  File {file_id}: FAILED to retrieve - {e}")
+                else:
+                    print("No file IDs were captured from the response.")
+
+                # List all files to see if any exist
+                print("\nListing all files in the agent service:")
+                try:
+                    files_list = await client.agents_client.files.list()
+                    count = 0
+                    for file_info in files_list.data:
+                        count += 1
+                        print(f"  - {file_info.id}: {file_info.filename} ({file_info.purpose})")
+                    if count == 0:
+                        print("  No files found.")
+                except Exception as e:
+                    print(f"  Failed to list files: {e}")
+
+        finally:
+            await client.close()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())