Skip to content

Commit bdc9562

Browse files
committed
Fix AgentService.Restart()
1 parent 4710d44 commit bdc9562

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

src/MaIN.Services/Services/AgentService.cs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,10 @@ public async Task<Chat> Restart(string agentId)
156156
var agent = await agentRepository.GetAgentById(agentId) ?? throw new AgentNotFoundException(agentId);
157157

158158
var chat = (await chatRepository.GetChatById(agent.ChatId))!.ToDomain();
159-
var llmService = llmServiceFactory.CreateService(agent.Backend ?? maInSettings.BackendType);
159+
var backend = ModelRegistry.TryGetById(chat.ModelId, out var model)
160+
? model!.Backend
161+
: maInSettings.BackendType;
162+
var llmService = llmServiceFactory.CreateService(backend);
160163
await llmService.CleanSessionCache(chat.Id!);
161164
AgentStateManager.ClearState(agent, chat);
162165

0 commit comments

Comments
 (0)