Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions sdk/ai/azure-ai-agents/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -723,7 +723,7 @@ The synchronous session log method returns `IterableStream<SessionLogEvent>`, wh

```java com.azure.ai.agents.session_logs_sync
IterableStream<SessionLogEvent> sessionLogs =
betaAgentsClient.getSessionLogStream(agentName, agentVersion, sessionId);
agentsClient.getSessionLogStream(agentName, agentVersion, sessionId);

int logsRead = 0;
for (SessionLogEvent event : sessionLogs) {
Expand All @@ -741,7 +741,7 @@ for (SessionLogEvent event : sessionLogs) {
The asynchronous session log method returns `Flux<SessionLogEvent>`, integrating naturally with Reactor pipelines:

```java com.azure.ai.agents.session_logs_async
betaAgentsAsyncClient.getSessionLogStream(agentName, agentVersion, sessionId)
agentsAsyncClient.getSessionLogStream(agentName, agentVersion, sessionId)
.take(100)
.doOnNext(event -> System.out.printf("[%s] %s%n", event.getEvent(), event.getData()))
.blockLast();
Expand Down

Large diffs are not rendered by default.

1,736 changes: 1,736 additions & 0 deletions sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/AgentsClient.java

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -60,13 +60,13 @@
@ServiceClientBuilder(
serviceClients = {
BetaMemoryStoresClient.class,
BetaToolboxesClient.class,
BetaAgentsClient.class,
AgentsClient.class,
ToolboxesClient.class,
BetaMemoryStoresAsyncClient.class,
BetaToolboxesAsyncClient.class,
BetaAgentsAsyncClient.class,
AgentsAsyncClient.class })
AgentsAsyncClient.class,
ToolboxesAsyncClient.class })
public final class AgentsClientBuilder
implements HttpTrait<AgentsClientBuilder>, ConfigurationTrait<AgentsClientBuilder>,
TokenCredentialTrait<AgentsClientBuilder>, EndpointTrait<AgentsClientBuilder> {
Expand All @@ -85,14 +85,12 @@ public final class AgentsClientBuilder

private static final String AGENT_PREVIEW_FEATURES = Stream
.concat(Arrays.stream(AgentDefinitionOptInKeys.values()).map(AgentDefinitionOptInKeys::toString),
Stream.of(FoundryFeaturesOptInKeys.AGENTS_OPTIMIZATION_V1_PREVIEW.toString()))
Stream.of(FoundryFeaturesOptInKeys.AGENTS_OPTIMIZATION_V2_PREVIEW.toString()))
.collect(Collectors.joining(","));

private static final String MEMORY_STORES_PREVIEW_FEATURES
= FoundryFeaturesOptInKeys.MEMORY_STORES_V1_PREVIEW.toString();

private static final String TOOLBOXES_PREVIEW_FEATURES = FoundryFeaturesOptInKeys.TOOLBOXES_V1_PREVIEW.toString();

private boolean allowPreview;

@Generated
Expand Down Expand Up @@ -540,10 +538,8 @@ public BetaAgentsClientBuilder beta() {
serviceClients = {
BetaAgentsClient.class,
BetaMemoryStoresClient.class,
BetaToolboxesClient.class,
BetaAgentsAsyncClient.class,
BetaMemoryStoresAsyncClient.class,
BetaToolboxesAsyncClient.class })
BetaMemoryStoresAsyncClient.class })
public final class BetaAgentsClientBuilder {

/**
Expand Down Expand Up @@ -571,15 +567,6 @@ public BetaMemoryStoresAsyncClient buildBetaMemoryStoresAsyncClient() {
buildInnerClient(MEMORY_STORES_PREVIEW_FEATURES).getBetaMemoryStores());
}

/**
* Builds an instance of BetaToolboxesAsyncClient class.
*
* @return an instance of BetaToolboxesAsyncClient.
*/
public BetaToolboxesAsyncClient buildBetaToolboxesAsyncClient() {
return new BetaToolboxesAsyncClient(buildInnerClient(TOOLBOXES_PREVIEW_FEATURES).getBetaToolboxes());
}

/**
* Builds an instance of BetaAgentsClient class.
*
Expand All @@ -597,15 +584,6 @@ public BetaAgentsClient buildBetaAgentsClient() {
public BetaMemoryStoresClient buildBetaMemoryStoresClient() {
return new BetaMemoryStoresClient(buildInnerClient(MEMORY_STORES_PREVIEW_FEATURES).getBetaMemoryStores());
}

/**
* Builds an instance of BetaToolboxesClient class.
*
* @return an instance of BetaToolboxesClient.
*/
public BetaToolboxesClient buildBetaToolboxesClient() {
return new BetaToolboxesClient(buildInnerClient(TOOLBOXES_PREVIEW_FEATURES).getBetaToolboxes());
}
}

/**
Expand All @@ -626,15 +604,6 @@ private BetaMemoryStoresAsyncClient buildBetaMemoryStoresAsyncClient() {
return new BetaMemoryStoresAsyncClient(buildInnerClient(MEMORY_STORES_PREVIEW_FEATURES).getBetaMemoryStores());
}

/**
* Builds an instance of BetaToolboxesAsyncClient class.
*
* @return an instance of BetaToolboxesAsyncClient.
*/
private BetaToolboxesAsyncClient buildBetaToolboxesAsyncClient() {
return new BetaToolboxesAsyncClient(buildInnerClient(TOOLBOXES_PREVIEW_FEATURES).getBetaToolboxes());
}

/**
* Builds an instance of BetaAgentsClient class.
*
Expand All @@ -654,11 +623,22 @@ private BetaMemoryStoresClient buildBetaMemoryStoresClient() {
}

/**
* Builds an instance of BetaToolboxesClient class.
* Builds an instance of ToolboxesAsyncClient class.
*
* @return an instance of BetaToolboxesClient.
* @return an instance of ToolboxesAsyncClient.
*/
private BetaToolboxesClient buildBetaToolboxesClient() {
return new BetaToolboxesClient(buildInnerClient(TOOLBOXES_PREVIEW_FEATURES).getBetaToolboxes());
@Generated
public ToolboxesAsyncClient buildToolboxesAsyncClient() {
return new ToolboxesAsyncClient(buildInnerClient().getToolboxes());
}

/**
* Builds an instance of ToolboxesClient class.
*
* @return an instance of ToolboxesClient.
*/
@Generated
public ToolboxesClient buildToolboxesClient() {
return new ToolboxesClient(buildInnerClient().getToolboxes());
}
}
Loading