Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
57 commits
Select commit Hold shift + click to select a range
e5d7d53
Checkpoint
crickman Jul 9, 2024
cfc8cd9
Markdown test
crickman Jul 9, 2024
8f40352
Sweet formatting!
crickman Jul 9, 2024
ec8fc2a
More diagrams
crickman Jul 9, 2024
f2b23c2
Tweak layout
crickman Jul 9, 2024
62e3619
VectorStore
crickman Jul 9, 2024
a7b8341
Layout
crickman Jul 9, 2024
d3e9c76
Layout
crickman Jul 9, 2024
83804d3
Sections
crickman Jul 9, 2024
199f157
Merge branch 'main' into adr_assistant_v2
crickman Jul 9, 2024
fbba826
Werds
crickman Jul 9, 2024
b4365c5
Run processing
crickman Jul 9, 2024
8546dd0
Formatting
crickman Jul 9, 2024
bdc96c4
Checkpoint
crickman Jul 10, 2024
5c91ef9
Merge branch 'main' into adr_assistant_v2
crickman Jul 11, 2024
cf3eb17
Checkpoint
crickman Jul 11, 2024
d157258
Draft complete
crickman Jul 11, 2024
a0bb01e
Typo
crickman Jul 11, 2024
98b80cd
Merge branch 'main' into adr_assistant_v2
crickman Jul 11, 2024
b8be881
Merge branch 'main' into adr_assistant_v2
crickman Jul 12, 2024
15149d8
VectorStore revision
crickman Jul 15, 2024
b02b257
Merge branch 'adr_assistant_v2' of https://github.com/microsoft/seman…
crickman Jul 15, 2024
a423330
Merge branch 'main' into adr_assistant_v2
crickman Jul 15, 2024
518ec99
Formatting
crickman Jul 15, 2024
ed6f064
Clean-up
crickman Jul 15, 2024
cee54c4
More clean-up
crickman Jul 15, 2024
89760b8
Update docs/decisions/0049-agents-assistantsV2.md
crickman Jul 15, 2024
8afd1d3
Setting precedence
crickman Jul 15, 2024
20ad996
Merge branch 'adr_assistant_v2' of https://github.com/microsoft/seman…
crickman Jul 15, 2024
15f368c
Typo
crickman Jul 15, 2024
382707e
Merge branch 'main' into adr_assistant_v2
crickman Jul 16, 2024
8a7f307
Merge branch 'main' into adr_assistant_v2
crickman Jul 17, 2024
5b89cc4
Merge branch 'main' into adr_assistant_v2
crickman Jul 17, 2024
be59a53
Add ToolCallBehavior
crickman Jul 17, 2024
24b8b87
Edits
crickman Jul 17, 2024
c177713
Merge branch 'main' into adr_assistant_v2
crickman Jul 17, 2024
906455d
Add convenience methods to table
crickman Jul 17, 2024
cdea179
Clarity (setting table)
crickman Jul 17, 2024
00d549f
Typo
crickman Jul 17, 2024
4a58353
anti-newspeak
crickman Jul 17, 2024
0b50c34
toml x2
crickman Jul 17, 2024
886469f
toml x3
crickman Jul 17, 2024
0467329
toml x3
crickman Jul 17, 2024
3d7eec6
toml x5
crickman Jul 17, 2024
5a9746c
toml x6
crickman Jul 17, 2024
1c4a022
nvm
crickman Jul 17, 2024
c0ea593
More
crickman Jul 17, 2024
8da6f41
FunctionChoiceBehavior alternative
crickman Jul 18, 2024
b771772
Format
crickman Jul 18, 2024
871874a
Merge branch 'main' into adr_assistant_v2
crickman Jul 18, 2024
3f5d290
Re-order
crickman Jul 18, 2024
7f7df48
Review / Naming Update
crickman Jul 19, 2024
a9c240a
Complete class listing (thread options)
crickman Jul 19, 2024
edee16f
Merge branch 'main' into adr_assistant_v2
crickman Jul 19, 2024
5687591
Merge branch 'main' into adr_assistant_v2
crickman Jul 22, 2024
90cf498
ThreadCreation
crickman Jul 22, 2024
172aae2
Capture tool_choice decision
crickman Jul 22, 2024
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
184 changes: 184 additions & 0 deletions docs/decisions/0049-agents-assistantsV2.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,184 @@

# Agent Framework - Assistant V2 Migration

## Context and Problem Statement

Open AI has release the _Assistants V2_ API. This builds on top of the V1 _assistant_ concept, but also invalidates certain V1 features. In addition, the _dotnet_ API that supports _Assistant V2_ features is entirely divergent on the `Azure.AI.OpenAI.Assistants` SDK that is currently in use.

### Open Issues
- **Streaming:** To be addressed as a discrete feature


## Design

Migrating to Assistant V2 API is a breaking change to the existing package due to:
- Underlying capability differences (e.g. `file-search` vs `retrieval`)
- Underlying V2 SDK is version incompatible with V1 (`OpenAI` and `Azure.AI.OpenAI`)

### Agent Implementation

The `OpenAIAssistant` agent is roughly equivalent to its V1 form save for:

- Supports options for _assistant_, _thread_, and _run_
- Agent definition shifts to `Definition` property
- Convenience methods for producing an OpenAI client

Previously, the agent definition as exposed via direct properties such as:

- `FileIds`
- `Metadata`

This has all been shifted and expanded upon via the `Definition` property which is of the same type (`OpenAIAssistantDefinition`) utilized to create and query an assistant.

<p align="center">
<kbd><img src="diagrams/assistant-agent.png" style="width: 720pt;"></kbd>
</p>

The following table describes the purpose of diagramed methods on the `OpenAIAssistantAgent`.

|Method Name|Description|
---|---
**Create**|Create a new assistant agent
**ListDefinitions**|List existing assistant definitions
**Retrieve**|Retrieve an existing assistant
**CreateThread**|Create an assistant thread
**DeleteThread**|Delete an assistant thread
**AddChatMessage**|Add a message to an assistant thread
**GetThreadMessages**|Retrieve all messages from an assistant thread
**Delete**|Delete the assistant agent's definition (puts agent into a terminal state)
**Invoke**|Invoke the assistant agent (no chat)
**GetChannelKeys**|Inherited from `Agent`
**CreateChannel**|Inherited from `Agent`


### Class Inventory
Comment thread
crickman marked this conversation as resolved.
This section provides an overview / inventory of all the public surface area described in this ADR.

|Class Name|Description|
Comment thread
crickman marked this conversation as resolved.
---|---
**OpenAIAssistantAgent**|An `Agent` based on the Open AI Assistant API
**OpenAIAssistantChannel**|An 'AgentChannel' for `OpenAIAssistantAgent` (associated with a _thread-id_.)
**OpenAIAssistantDefinition**|All of the metadata / definition for an Open AI Assistant. Unable to use the _Open AI API_ model due to implementation constraints (constructor not public).
**OpenAIAssistantExecutionOptions**|Options that affect the _run_, but defined globally for the agent/assistant.
**OpenAIAssistantInvocationOptions**|Options bound to a discrete run, used for direct (no chat) invocation.
**OpenAIThreadCreationOptions**|Options for creating a thread that take precedence over assistant definition, when specified.
**OpenAIServiceConfiguration**|Describes the service connection and used to create the `OpenAIClient`


### Run Processing
Comment thread
crickman marked this conversation as resolved.

The heart of supporting an _assistant_ agent is creating and processing a `Run`.

A `Run` is effectively a discrete _assistant_ interaction on a `Thread` (or conversation).

- https://platform.openai.com/docs/api-reference/runs
- https://platform.openai.com/docs/api-reference/run-steps

This `Run` processing is implemented as internal logic within the _OpenAI Agent Framework_ that is outlined here:

Initiate processing using:

- `agent` -> `OpenAIAssistantAgent`
- `client` -> `AssistantClient`
- `threadid` -> `string`
- `options` -> `OpenAIAssistantInvocationOptions` (optional)


Perform processing:

- Verify `agent` not deleted
- Define `RunCreationOptions`
- Create the `run` (based on `threadid` and `agent.Id`)
- Process the run:

do

- Poll `run` status until is not _queued_, _in-progress_, or _cancelling_
- Throw if `run` status is _expired_, _failed_, or _cancelled_
- Query `steps` for `run`

- if `run` status is _requires-action_

- process function `steps`

- post function results

- foreach (`step` is completed)

- if (`step` is tool-call) generate and yield tool content

- else if (`step` is message) generate and yield message content

while (`run` status is not completed)


### Vector Store Support

_Vector Store_ support is required in order to enable usage of the `file-search` tool.

In alignment with V2 streaming of the `FileClient`, the caller may also directly target `VectorStoreClient` from the _OpenAI SDK_.


### Definition / Options Classes

Specific configuration/options classes are introduced to support the ability to define assistant behavior at each of the supported articulation points (i.e. _assistant_, _thread_, & _run_).

|Class|Purpose|
|---|---|
|`OpenAIAssistantDefinition`|Definition of the assistant. Used when creating a new assistant, inspecting an assistant-agent instance, or querying assistant definitions.|
|`OpenAIAssistantExecutionOptions`|Options that affect run execution, defined within assistant scope.|
|`OpenAIAssistantInvocationOptions`|Run level options that take precedence over assistant definition, when specified.|
|`OpenAIAssistantToolCallBehavior`|Informs tool-call behavior for the associated scope: assistant or run.|
|`OpenAIThreadCreationOptions`|Thread scoped options that take precedence over assistant definition, when specified.|
|`OpenAIServiceConfiguration`|Informs the which service to target, and how.|


#### Assistant Definition

The `OpenAIAssistantDefinition` was previously used only when enumerating a list of stored agents. It has been evolved to also be used as input for creating and agent and exposed as a discrete property on the `OpenAIAssistantAgent` instance.

This includes optional `ExecutionOptions` which define default _run_ behavior. Since these execution options are not part of the remote assistant definition, they are persisted in the assistant metadata for when an existing agent is retrieved. `OpenAIAssistantToolCallBehavior` is included as part of the _execution options_ and modeled in alignment with the `ToolCallBehavior` associated with _AI Connectors_.

> Note: Manual function calling isn't currently supported for `OpenAIAssistantAgent` or `AgentChat` and is planned to be addressed as an enhancement. When this supported is introduced, `OpenAIAssistantToolCallBehavior` will determine the function calling behavior (also in alignment with the `ToolCallBehavior` associated with _AI Connectors_).

**Alternative (Future?)**

A pending change has been authored that introduces `FunctionChoiceBehavior` as a property of the base / abstract `PromptExecutionSettings`. Once realized, it may make sense to evaluate integrating this pattern for `OpenAIAssistantAgent`. This may also imply in inheritance relationship of `PromptExecutionSettings` for both `OpenAIAssistantExecutionOptions` and `OpenAIAssistantInvocationOptions` (next section).

**DECISION**: Do not support `tool_choice` until the `FunctionChoiceBehavior` is realized.

<p align="center">
<kbd><img src="diagrams/assistant-definition.png" style="width: 500pt;"></kbd>
</p>


#### Assistant Invocation Options

When invoking an `OpenAIAssistantAgent` directly (no-chat), definition that only apply to a discrete run may be specified. These definition are defined as `OpenAIAssistantInvocationOptions` and ovetake precedence over any corresponding assistant or thread definition.

> Note: These definition are also impacted by the `ToolCallBehavior` / `FunctionChoiceBehavior` quadary.

<p align="center">
<kbd><img src="diagrams/assistant-invocationsettings.png" style="width: 370pt;"></kbd>
</p>


#### Thread Creation Options

When invoking an `OpenAIAssistantAgent` directly (no-chat), a thread must be explicitly managed. When doing so, thread specific options may be specified. These options are defined as `OpenAIThreadCreationOptions` and take precedence over any corresponding assistant definition.

<p align="center">
<kbd><img src="diagrams/assistant-threadcreationsettings.png" style="width: 132pt;"></kbd>
</p>


#### Service Configuration

The `OpenAIServiceConfiguration` defines how to connect to a specific remote service, whether it be OpenAI, Azure, or proxy. This eliminates the need to define multiple overloads for each call site that results in a connection to the remote API service (i.e. create a _client)_.

> Note: This was previously named `OpenAIAssistantConfiguration`, but is not necessarily assistant specific.

<p align="center">
<kbd><img src="diagrams/assistant-serviceconfig.png" style="width: 520pt;"></kbd>
</p>

44 changes: 44 additions & 0 deletions docs/decisions/diagrams/assistant-agent.mmd
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
classDiagram

KernelAgent
OpenAIAssistantDefinition
OpenAIAssistantDefinition --> OpenAIExecutionOptions
OpenAIExecutionOptions
OpenAIExecutionOptions --> AssistantToolCallBehavior
OpenAIServiceConfiguration
OpenAIAssistantInvocationOptions
OpenAIAssistantInvocationOptions --> AssistantToolCallBehavior
OpenAIThreadCreationOptions

KernelAgent <|-- OpenAIAssistantAgent
OpenAIAssistantAgent ..> OpenAIServiceConfiguration
OpenAIAssistantAgent -- OpenAIAssistantChannel
OpenAIAssistantAgent --> OpenAIAssistantDefinition
OpenAIAssistantAgent ..> OpenAIAssistantInvocationOptions
OpenAIAssistantAgent ..> OpenAIThreadCreationOptions
class OpenAIAssistantAgent {
+OpenAIAssistantDefinition Definition
+bool IsDeleted
+RunPollingConfiguration Polling
+Task~OpenAIAssistantAgent~ Create(Kernel kernel, OpenAIServiceConfiguration config, OpenAIAssistantDefinition definition)$
+AsyncEnumerable~OpenAIAssistantDefinition~ ListDefinitions(OpenAIServiceConfiguration config)$
+Task~OpenAIAssistantAgent~ Retrieve(Kernel kernel, OpenAIServiceConfiguration config, string id)$
+Task~string~ CreateThread()
+Task~string~ CreateThread(OpenAIThreadCreationOptions? Options)
+Task~bool~ DeleteThread(string threadId)
+Task AddChatMessage(string threadId, ChatMessageContent message)
+AsyncEnumerable~ChatMessageContent~ GetThreadMessages(string threadId)
+Task~bool~ Delete()
+AsyncEnumerable~ChatMessageContent~ Invoke(string threadId)
+AsyncEnumerable~ChatMessageContent~ Invoke(string threadId, OpenAIAssistantInvocationOptions? Options)
#AsyncEnumerable~string~ GetChannelKeys()
#Task~AgentChannel~ CreateChannel()
}

OpenAIAssistantChannel ..> OpenAIAssistantAgent
class OpenAIAssistantChannel {
#Task Receive(IReadOnlyList<ChatMessageContent> history)
#AsyncEnumerable<ChatMessageContent> Invoke(OpenAIAssistantAgent agent)
#AsyncEnumerable<ChatMessageContent> GetHistory()
}

Binary file added docs/decisions/diagrams/assistant-agent.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
33 changes: 33 additions & 0 deletions docs/decisions/diagrams/assistant-definition.mmd
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
classDiagram

OpenAIAssistantDefinition --> OpenAIAssistantExecutionOptions
class OpenAIAssistantDefinition {
string ModelName
string? Description
string Id
string? Instructions
string? Name
List~string~? CodeInterpterFileIds
bool EnableCodeInterpreter
bool EnableJsonResponse
Dictionary~string, string~? Metadata
float? Temperature
float? TopP
string? VectorStoreId
OpenAIAssistantExecutionOptions? ExecutionOptions
}

OpenAIAssistantExecutionOptions --> OpenAIAssistantToolCallBehavior
class OpenAIAssistantExecutionOptions {
int? MaxCompletionTokens
int? MaxPromptTokens
bool? ParallelToolCallsEnabled
int? TruncationMessageCount
OpenAIAssistantToolCallBehavior? ToolCallBehavior
}

class OpenAIAssistantToolCallBehavior {
AssistantToolCallBehavior RequireCodeInterpreter()$
AssistantToolCallBehavior RequireFunction(KernelFunction function)$
AssistantToolCallBehavior RequireFileSearch()$
}
Binary file added docs/decisions/diagrams/assistant-definition.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
23 changes: 23 additions & 0 deletions docs/decisions/diagrams/assistant-invocationsettings.mmd
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
classDiagram

OpenAIAssistantInvocationOptions --> OpenAIAssistantToolCallBehavior
class OpenAIAssistantInvocationOptions {
string? ModelName
bool? EnableCodeInterpreter
bool? EnableFileSearch
bool? EnableJsonResponse
int? MaxCompletionTokens
int? MaxPromptTokens
bool? ParallelToolCallsEnabled
int? TruncationMessageCount
float? Temperature
float? TopP
Dictionary~string, string~? Metadata
OpenAIAssistantToolCallBehavior? ToolCallBehavior
}

class OpenAIAssistantToolCallBehavior {
AssistantToolCallBehavior RequireCodeInterpreter()$
AssistantToolCallBehavior RequireFunction(KernelFunction function)$
AssistantToolCallBehavior RequireFileSearch()$
}
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
40 changes: 40 additions & 0 deletions docs/decisions/diagrams/assistant-serviceconfig.mmd
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
classDiagram

OpenAIClientFactory ..> OpenAIServiceConfiguration
class OpenAIClientFactory {
<<internal>>
}

OpenAIServiceConfiguration --> OpenAIServiceType
class OpenAIServiceConfiguration {
OpenAIServiceConfiguration ForAzureOpenAI(string? apiKey, Uri? endpoint, HttpClient httpClient)$
OpenAIServiceConfiguration ForAzureOpenAI(TokenCredential credential, Uri? endpoint, HttpClient httpClient)$
OpenAIServiceConfiguration OpenAI(string? apiKey, Uri? endpoint, HttpClient httpClient)$
-string? ApiKey
-TokenCredential? TokenCredential
-Uri? Endpoint
-HttpClient? HttpClient
-OpenAIServiceType ServiceType
}

OpenAIServiceConfigurationExtensions ..> OpenAIServiceConfiguration
OpenAIServiceConfigurationExtensions ..> FileClient
OpenAIServiceConfigurationExtensions ..> VectorStoreClient
class OpenAIServiceConfigurationExtensions {
+FileClient CreateFileClient(this OpenAIServiceConfiguration config)$
+VectorStoreClient CreateVectorStoreClient(this OpenAIServiceConfiguration config)$
}

class OpenAIServiceType {
<<enumeration>>
AzureOpenAI
OpenAI
}

class FileClient {
<<OpenAI>>
}

class VectorStoreClient {
<<OpenAI>>
}
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
8 changes: 8 additions & 0 deletions docs/decisions/diagrams/assistant-threadcreationsettings.mmd
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
classDiagram

class OpenAIThreadCreationOptions {
List~string~? CodeInterpterFileIds
IReadOnlyList<ChatMessageContent>? Messages
string? VectorStoreId
Dictionary~string, string~? Metadata
}
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.