SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Update IRealtimeHook

Browse source
This commit is contained in:
Haiping Chen 2025-08-07 15:38:16 -05:00
parent 87124c4a21
commit df0423d917
2 changed files with 22 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/Infrastructure/BotSharp.Abstraction/Realtime/IRealtimeHook.cs
View file

@ -6,8 +6,15 @@ namespace BotSharp.Abstraction.Realtime;
public interface IRealtimeHook : IHookBase public interface IRealtimeHook : IHookBase
{ {
Task OnModelReady(Agent agent, IRealTimeCompletion completer); Task OnModelReady(Agent agent, IRealTimeCompletion completer)
string[] OnModelTranscriptPrompt(Agent agent); => Task.CompletedTask;
Task OnTranscribeCompleted(RoleDialogModel message, TranscriptionData data);
Task<bool> ShouldReconnect(RealtimeHubConnection conn) => Task.FromResult(false); string[] OnModelTranscriptPrompt(Agent agent)
=> [];
Task OnTranscribeCompleted(RoleDialogModel message, TranscriptionData data)
=> Task.CompletedTask;
Task<bool> ShouldReconnect(RealtimeHubConnection conn, RoleDialogModel message)
=> Task.FromResult(false);
} }

24
src/Infrastructure/BotSharp.Core.Realtime/Services/RealtimeHub.cs
View file

@ -100,6 +100,17 @@ public class RealtimeHub : IRealtimeHub
} }
await routing.InvokeFunction(message.FunctionName, message, options: new() { From = InvokeSource.Llm }); await routing.InvokeFunction(message.FunctionName, message, options: new() { From = InvokeSource.Llm });
var hooks = _services.GetHooks<IRealtimeHook>(_conn.CurrentAgentId);
foreach (var hook in hooks)
{
if (await hook.ShouldReconnect(_conn, message))
{
await _completer.Reconnect(_conn);
_logger.LogWarning("Reconnecting to model due to function call: {FunctionName}", message.FunctionName);
break;
}
}
} }
else else
{ {
@ -117,19 +128,6 @@ public class RealtimeHub : IRealtimeHub
} }
} }
} }
var isReconnect = false;
var realtimeHooks = _services.GetHooks<IRealtimeHook>(_conn.CurrentAgentId);
foreach (var hook in realtimeHooks)
{
isReconnect = await hook.ShouldReconnect(_conn);
if (isReconnect) break;
}
if (isReconnect)
{
await _completer.Reconnect(_conn);
}
}, },
onConversationItemCreated: async response => onConversationItemCreated: async response =>
{ {