forked from microsoft/agent-framework
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProgram.cs
More file actions
89 lines (72 loc) · 3.46 KB
/
Copy pathProgram.cs
File metadata and controls
89 lines (72 loc) · 3.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// Copyright (c) Microsoft. All rights reserved.
// This sample shows how to download files generated by Code Interpreter using the Containers API.
// Code Interpreter generates files inside containers (cfile_ / cntr_ IDs) which cannot be
// downloaded via the standard Files API. Use ContainerClient instead.
#pragma warning disable OPENAI001
using System.ClientModel;
using Microsoft.Agents.AI;
using Microsoft.Extensions.AI;
using OpenAI;
using OpenAI.Containers;
using OpenAI.Responses;
string apiKey = Environment.GetEnvironmentVariable("OPENAI_API_KEY") ?? throw new InvalidOperationException("OPENAI_API_KEY is not set.");
string model = Environment.GetEnvironmentVariable("OPENAI_CHAT_MODEL_NAME") ?? "gpt-4o-mini";
var openAIClient = new OpenAIClient(new ApiKeyCredential(apiKey));
// Create an agent with Code Interpreter tool enabled
AIAgent agent = openAIClient
.GetResponsesClient()
.AsAIAgent(
model: model,
instructions: "You are a helpful assistant that can generate files using code.",
name: "CodeInterpreterAgent",
tools: [new HostedCodeInterpreterTool()]);
// Ask the agent to generate a file
AgentResponse response = await agent.RunAsync(
"Create a CSV file with the multiplication times tables from 1 to 12. Include headers.");
// Display the text response
foreach (TextContent textContent in response.Messages.SelectMany(x => x.Contents).OfType<TextContent>())
{
Console.WriteLine(textContent.Text);
}
// Extract container file citations from response annotations and download
ContainerClient containerClient = openAIClient.GetContainerClient();
HashSet<string> downloadedFiles = [];
bool foundContainerFiles = false;
foreach (AIContent content in response.Messages.SelectMany(x => x.Contents))
{
if (content.Annotations is null)
{
continue;
}
foreach (AIAnnotation annotation in content.Annotations)
{
// Container files from Code Interpreter have ContainerFileCitationMessageAnnotation as raw representation
if (annotation is CitationAnnotation citation
&& citation.RawRepresentation is ContainerFileCitationMessageAnnotation containerCitation)
{
foundContainerFiles = true;
// Deduplicate by container+file ID in case the same file is cited multiple times
string key = $"{containerCitation.ContainerId}/{containerCitation.FileId}";
if (!downloadedFiles.Add(key))
{
continue;
}
Console.WriteLine($"\nDownloading container file: {containerCitation.Filename}");
Console.WriteLine($" Container ID: {containerCitation.ContainerId}");
Console.WriteLine($" File ID: {containerCitation.FileId}");
BinaryData fileData = await containerClient.DownloadContainerFileAsync(
containerCitation.ContainerId,
containerCitation.FileId);
// Sanitize filename to prevent path traversal
string safeFilename = Path.GetFileName(containerCitation.Filename);
string outputPath = Path.Combine(Directory.GetCurrentDirectory(), safeFilename);
await File.WriteAllBytesAsync(outputPath, fileData.ToArray());
Console.WriteLine($" Saved to: {outputPath}");
}
}
}
if (!foundContainerFiles)
{
Console.WriteLine("\nNo container file citations found in the response.");
Console.WriteLine("The model may not have generated a downloadable file for this prompt.");
}