Skip to content
Merged
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
Original file line number Diff line number Diff line change
Expand Up @@ -134,8 +134,7 @@ private void ProcessBatch(IList<PromptStorageEventArgs> eventArgs)
{
switch (eventArg.Type)
{
case PromptStorageEventType.Added:
case PromptStorageEventType.Modified:
case PromptStorageEventType.Updated:
using (_diagnosticEvents.UpdatePrompts())
{
var prompt = PromptFactory.CreatePrompt(eventArg.PromptDefinition!);
Expand Down Expand Up @@ -175,8 +174,7 @@ private void ProcessBatch(IList<OperationToolStorageEventArgs> eventArgs)
{
switch (eventArg.Type)
{
case OperationToolStorageEventType.Added:
case OperationToolStorageEventType.Modified:
case OperationToolStorageEventType.Updated:
using (_diagnosticEvents.UpdateTools())
{
var validationResult =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ namespace HotChocolate.Adapters.Mcp.Storage;
/// </summary>
/// <param name="Name">The name of the tool that changed.</param>
/// <param name="Type">The type of change that occurred.</param>
/// <param name="ToolDefinition">The tool definition. Required for Added/Modified, null for Removed.</param>
/// <param name="ToolDefinition">The tool definition. Required for Updated, null for Removed.</param>
public record OperationToolStorageEventArgs(
string Name,
OperationToolStorageEventType Type,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,8 @@ namespace HotChocolate.Adapters.Mcp.Storage;
/// </summary>
public enum OperationToolStorageEventType
{
/// <summary>A new tool was added to storage.</summary>
Added,

/// <summary>An existing tool was modified in storage.</summary>
Modified,
/// <summary>A tool was added to or updated in storage.</summary>
Updated,

/// <summary>A tool was removed from storage.</summary>
Removed
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ namespace HotChocolate.Adapters.Mcp.Storage;
/// </summary>
/// <param name="Name">The name of the prompt that changed.</param>
/// <param name="Type">The type of change that occurred.</param>
/// <param name="PromptDefinition">The prompt definition. Required for Added/Modified, null for Removed.</param>
/// <param name="PromptDefinition">The prompt definition. Required for Updated, null for Removed.</param>
public record PromptStorageEventArgs(
string Name,
PromptStorageEventType Type,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,8 @@ namespace HotChocolate.Adapters.Mcp.Storage;
/// </summary>
public enum PromptStorageEventType
{
/// <summary>A new prompt was added to storage.</summary>
Added,

/// <summary>An existing prompt was modified in storage.</summary>
Modified,
/// <summary>A prompt was added to or updated in storage.</summary>
Updated,

/// <summary>A prompt was removed from storage.</summary>
Removed
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,46 +48,32 @@ public async Task AddOrUpdatePromptAsync(
PromptDefinition promptDefinition,
CancellationToken cancellationToken = default)
{
PromptStorageEventType type;

await _promptSemaphore.WaitAsync(cancellationToken);

if (_prompts.TryAdd(promptDefinition.Name, promptDefinition))
{
type = PromptStorageEventType.Added;
}
else
if (!_prompts.TryAdd(promptDefinition.Name, promptDefinition))
{
_prompts[promptDefinition.Name] = promptDefinition;
type = PromptStorageEventType.Modified;
}

_promptSemaphore.Release();

NotifySubscribers(promptDefinition.Name, promptDefinition, type);
NotifySubscribers(promptDefinition.Name, promptDefinition, PromptStorageEventType.Updated);
}

public async Task AddOrUpdateToolAsync(
OperationToolDefinition toolDefinition,
CancellationToken cancellationToken = default)
{
OperationToolStorageEventType type;

await _toolSemaphore.WaitAsync(cancellationToken);

if (_tools.TryAdd(toolDefinition.Name, toolDefinition))
{
type = OperationToolStorageEventType.Added;
}
else
if (!_tools.TryAdd(toolDefinition.Name, toolDefinition))
{
_tools[toolDefinition.Name] = toolDefinition;
type = OperationToolStorageEventType.Modified;
}

_toolSemaphore.Release();

NotifySubscribers(toolDefinition.Name, toolDefinition, type);
NotifySubscribers(toolDefinition.Name, toolDefinition, OperationToolStorageEventType.Updated);
}

public IDisposable Subscribe(IObserver<PromptStorageEventArgs> observer)
Expand All @@ -105,7 +91,7 @@ private void NotifySubscribers(
PromptDefinition? promptDefinition,
PromptStorageEventType type)
{
if (type is PromptStorageEventType.Added or PromptStorageEventType.Modified)
if (type is PromptStorageEventType.Updated)
{
ArgumentNullException.ThrowIfNull(promptDefinition);
}
Expand All @@ -129,7 +115,7 @@ private void NotifySubscribers(
OperationToolDefinition? toolDefinition,
OperationToolStorageEventType type)
{
if (type is OperationToolStorageEventType.Added or OperationToolStorageEventType.Modified)
if (type is OperationToolStorageEventType.Updated)
{
ArgumentNullException.ThrowIfNull(toolDefinition);
}
Expand Down
Loading