Skip to content

Commit 363ddf2

Browse files
Merge pull request #99 from wisedev-code/feat/accept-collection-of-messages
feat/allow-message-collection
2 parents 4b3b454 + 9db1a5b commit 363ddf2

File tree

4 files changed

+31
-3
lines changed

4 files changed

+31
-3
lines changed

Releases/0.7.1.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
# 0.7.1 release
2+
3+
- Chat/Agent context extension (accept multiple messages)

src/MaIN.Core/.nuspec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
<package>
33
<metadata>
44
<id>MaIN.NET</id>
5-
<version>0.7.0</version>
5+
<version>0.7.1</version>
66
<authors>Wisedev</authors>
77
<owners>Wisedev</owners>
88
<icon>favicon.png</icon>

src/MaIN.Core/Hub/Contexts/AgentContext.cs

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ public async Task<ChatResult> ProcessAsync(string message, bool translate = fals
237237
{
238238
Content = message,
239239
Role = "User",
240-
Type = MessageType.LocalLLM, //TODO this need an improvement - we dont know if the message is from local or cloud
240+
Type = MessageType.LocalLLM,
241241
Time = DateTime.Now
242242
});
243243
var result = await _agentService.Process(chat, _agent.Id, _knowledge, translate);
@@ -269,6 +269,26 @@ public async Task<ChatResult> ProcessAsync(Message message, bool translate = fal
269269
CreatedAt = DateTime.Now
270270
};
271271
}
272+
273+
public async Task<ChatResult> ProcessAsync(IEnumerable<Message> messages, bool translate = false)
274+
{
275+
if (_knowledge == null)
276+
{
277+
LoadExistingKnowledgeIfExists();
278+
}
279+
var chat = await _agentService.GetChatByAgent(_agent.Id);
280+
chat.Messages.Clear();
281+
chat.Messages.AddRange(messages);
282+
var result = await _agentService.Process(chat, _agent.Id, _knowledge, translate);
283+
var messageResult = result.Messages.LastOrDefault()!;
284+
return new ChatResult()
285+
{
286+
Done = true,
287+
Model = result.Model,
288+
Message = messageResult,
289+
CreatedAt = DateTime.Now
290+
};
291+
}
272292

273293
public async Task<Chat> GetChat()
274294
{

src/MaIN.Core/Hub/Contexts/ChatContext.cs

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,12 @@ public ChatContext WithBackend(BackendType backendType)
8181
return this;
8282
}
8383

84+
public ChatContext WithMessages(IEnumerable<Message> messages)
85+
{
86+
_chat.Messages.AddRange(messages);
87+
return this;
88+
}
89+
8490
public ChatContext WithMessage(string content)
8591
{
8692
var message = new Message
@@ -95,7 +101,6 @@ public ChatContext WithMessage(string content)
95101
return this;
96102
}
97103

98-
99104
public ChatContext WithMessage(string content, byte[] image)
100105
{
101106
var message = new Message

0 commit comments

Comments
 (0)