51 lines
1.7 KiB
C#
51 lines
1.7 KiB
C#
using BotSharp.Abstraction.Agents.Models;
|
|
using BotSharp.Abstraction.Repositories;
|
|
using System.IO;
|
|
|
|
namespace BotSharp.Core.Agents.Services;
|
|
|
|
public partial class AgentService
|
|
{
|
|
public async Task UpdateAgent(Agent agent)
|
|
{
|
|
var db = _services.GetRequiredService<IBotSharpRepository>();
|
|
|
|
db.Transaction<IBotSharpTable>(delegate
|
|
{
|
|
var record = (from a in db.Agent
|
|
join ua in db.UserAgent on a.Id equals ua.AgentId
|
|
join u in db.User on ua.UserId equals u.Id
|
|
where (ua.UserId == _user.Id || u.ExternalId == _user.Id) &&
|
|
a.Id == agent.Id
|
|
select a).FirstOrDefault();
|
|
|
|
if (record == null) return;
|
|
|
|
record.Name = agent.Name;
|
|
|
|
if (!string.IsNullOrEmpty(agent.Description))
|
|
record.Description = agent.Description;
|
|
|
|
if (!string.IsNullOrEmpty(agent.Instruction))
|
|
record.Instruction = agent.Instruction;
|
|
|
|
if (!string.IsNullOrEmpty(agent.Functions))
|
|
record.Functions = agent.Functions;
|
|
|
|
if (!agent.Routes.IsEmpty())
|
|
record.Routes = agent.Routes;
|
|
|
|
record.UpdatedTime = DateTime.UtcNow;
|
|
db.Add<IBotSharpTable>(record);
|
|
});
|
|
|
|
// Save instruction to file
|
|
//var dir = GetAgentDataDir(agent.Id);
|
|
//var instructionFile = Path.Combine(dir, "instruction.txt");
|
|
//File.WriteAllText(instructionFile, agent.Instruction);
|
|
|
|
//var samplesFile = Path.Combine(dir, "samples.txt");
|
|
//File.WriteAllText(samplesFile, agent.Samples);
|
|
}
|
|
}
|