Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
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
Prev Previous commit
Next Next commit
fb
  • Loading branch information
christothes committed May 5, 2025
commit 5ce1d026f0cc7cd0f1dd86d0a6b7c373a1c83800
6 changes: 3 additions & 3 deletions src/Utility/ChatTools.cs
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ public class ChatTools
private readonly Dictionary<string, Func<string, BinaryData, Task<BinaryData>>> _mcpMethods = [];
private readonly List<ChatTool> _tools = [];
private readonly EmbeddingClient _client;
private readonly List<VectorbaseEntry> _entries = [];
private readonly List<VectorDatabaseEntry> _entries = [];
private readonly List<McpClient> _mcpClients = [];
private readonly Dictionary<string, McpClient> _mcpClientsByEndpoint = [];

Expand Down Expand Up @@ -133,7 +133,7 @@ private async Task AddMcpToolsAsync(BinaryData toolDefinitions, McpClient client
var vector = embedding.ToFloats();
var item = toolsToVectorize[embedding.Index];
var toolDefinition = SerializeTool(item);
_entries.Add(new VectorbaseEntry(vector, toolDefinition));
_entries.Add(new VectorDatabaseEntry(vector, toolDefinition));
}
}
}
Expand Down Expand Up @@ -211,7 +211,7 @@ private async Task<IEnumerable<ChatTool>> FindRelatedTools(bool async, string pr
.Select(e => ParseToolDefinition(e.Data));
}

private async Task<IEnumerable<VectorbaseEntry>> FindVectorMatches(bool async, string prompt, int maxTools, float minVectorDistance)
private async Task<IEnumerable<VectorDatabaseEntry>> FindVectorMatches(bool async, string prompt, int maxTools, float minVectorDistance)
{
var vector = async ?
await ToolsUtility.GetEmbeddingAsync(_client, prompt).ConfigureAwait(false) :
Expand Down
6 changes: 3 additions & 3 deletions src/Utility/ResponseTools.cs
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ public class ResponseTools
private readonly Dictionary<string, Func<string, BinaryData, Task<BinaryData>>> _mcpMethods = [];
private readonly List<ResponseTool> _tools = [];
private readonly EmbeddingClient _client;
private readonly List<VectorbaseEntry> _entries = [];
private readonly List<VectorDatabaseEntry> _entries = [];
private readonly List<McpClient> _mcpClients = [];
private readonly Dictionary<string, McpClient> _mcpClientsByEndpoint = [];

Expand Down Expand Up @@ -133,7 +133,7 @@ private async Task AddMcpToolsAsync(BinaryData toolDefinitions, McpClient client
var vector = embedding.ToFloats();
var item = toolsToVectorize[embedding.Index];
var toolDefinition = SerializeTool(item);
_entries.Add(new VectorbaseEntry(vector, toolDefinition));
_entries.Add(new VectorDatabaseEntry(vector, toolDefinition));
}
}
}
Expand Down Expand Up @@ -214,7 +214,7 @@ private async Task<IEnumerable<ResponseTool>> FindRelatedTools(bool async, strin
.Select(e => ParseToolDefinition(e.Data));
}

private async Task<IEnumerable<VectorbaseEntry>> FindVectorMatches(bool async, string prompt, int maxTools, float minVectorDistance)
private async Task<IEnumerable<VectorDatabaseEntry>> FindVectorMatches(bool async, string prompt, int maxTools, float minVectorDistance)
{
var vector = async ?
await ToolsUtility.GetEmbeddingAsync(_client, prompt).ConfigureAwait(false) :
Expand Down
2 changes: 1 addition & 1 deletion src/Utility/ToolsUtility.cs
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,7 @@ internal static void ParseFunctionCallArgs(MethodInfo method, BinaryData functio
}
}

internal static IEnumerable<VectorbaseEntry> GetClosestEntries(List<VectorbaseEntry> entries, int maxTools, float minVectorDistance, ReadOnlyMemory<float> vector)
internal static IEnumerable<VectorDatabaseEntry> GetClosestEntries(List<VectorDatabaseEntry> entries, int maxTools, float minVectorDistance, ReadOnlyMemory<float> vector)
{
var distances = entries
.Select((e, i) => (Distance: 1f - ToolsUtility.CosineSimilarity(e.Vector.Span, vector.Span), Index: i))
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,17 @@
namespace OpenAI;

/// <summary>
/// A vectorbase entry.
/// A vector database entry.
/// </summary>
internal readonly struct VectorbaseEntry
internal readonly struct VectorDatabaseEntry
{
/// <summary>
/// Initializes a new instance of the <see cref="VectorbaseEntry"/> class.
/// Initializes a new instance of the <see cref="VectorDatabaseEntry"/> class.
/// </summary>
/// <param name="vector"></param>
/// <param name="data"></param>
/// <param name="id"></param>
public VectorbaseEntry(ReadOnlyMemory<float> vector, BinaryData data, int? id = default)
public VectorDatabaseEntry(ReadOnlyMemory<float> vector, BinaryData data, int? id = default)
{
Vector = vector;
Data = data;
Expand Down