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

Support states for instruct mode.

Browse source
This commit is contained in:
Haiping Chen 2023-10-22 19:37:33 -05:00
parent a4c1b56740
commit 8672214c39
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/Infrastructure/BotSharp.OpenAPI/Controllers/InstructModeController.cs
View file

@ -35,6 +35,7 @@ public class InstructModeController : ControllerBase, IApiAdapter
}
var conv = _services.GetRequiredService<IConversationService>();
input.States.ForEach(x => conv.States.SetState(x.Split('=')[0], x.Split('=')[1]));
conv.States.SetState("provider", input.Provider)
.SetState("model", input.Model);
@ -46,9 +47,14 @@ public class InstructModeController : ControllerBase, IApiAdapter
}
[HttpPost("/instruct/text-completion")]
public async Task<string> TextCompletion([FromBody] IncomingMessageModel message)
public async Task<string> TextCompletion([FromBody] IncomingMessageModel input)
{
var conv = _services.GetRequiredService<IConversationService>();
input.States.ForEach(x => conv.States.SetState(x.Split('=')[0], x.Split('=')[1]));
conv.States.SetState("provider", input.Provider)
.SetState("model", input.Model);
var textCompletion = CompletionProvider.GetTextCompletion(_services);
return await textCompletion.GetCompletion(message.Text);
return await textCompletion.GetCompletion(input.Text);
}
}