From 6f7f143e7d658313ed1e56ea4a9a188d1f44383f Mon Sep 17 00:00:00 2001
From: Jicheng Lu <103353@smsassist.com>
Date: Wed, 25 Oct 2023 15:09:21 -0500
Subject: [PATCH] add impact

---
 .../BotSharp.Core/Repository/FileRepository.cs           | 9 +++++++--
 .../Models/AgentResponseMongoElement.cs                  | 1 +
 .../Models/AgentTemplateMongoElement.cs                  | 1 +
 .../Models/FunctionDefMongoElement.cs                    | 1 +
 .../Models/RoutingRuleMongoElement.cs                    | 1 +
 src/Plugins/BotSharp.Plugin.MongoStorage/Using.cs        | 3 ++-
 6 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/Infrastructure/BotSharp.Core/Repository/FileRepository.cs b/src/Infrastructure/BotSharp.Core/Repository/FileRepository.cs
index 4d862e7a..f873a827 100644
--- a/src/Infrastructure/BotSharp.Core/Repository/FileRepository.cs
+++ b/src/Infrastructure/BotSharp.Core/Repository/FileRepository.cs
@@ -5,7 +5,6 @@ using BotSharp.Abstraction.Users.Models;
 using BotSharp.Abstraction.Agents.Models;
 using MongoDB.Driver;
 using BotSharp.Abstraction.Routing.Models;
-using Amazon.Util;
 
 namespace BotSharp.Core.Repository;
 
@@ -380,7 +379,13 @@ public class FileRepository : IBotSharpRepository
         }
 
         var functionText = JsonSerializer.Serialize(functions, _options);
-        File.WriteAllText(functionFile, functionText);
+
+        using (var sw = File.CreateText(functionFile))
+        {
+            sw.Write(functionText);
+        }
+
+        //File.WriteAllText(functionFile, functionText);
     }
 
     private void UpdateAgentTemplates(string agentId, List<AgentTemplate> templates)
diff --git a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentResponseMongoElement.cs b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentResponseMongoElement.cs
index ee556549..26ddebbc 100644
--- a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentResponseMongoElement.cs
+++ b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentResponseMongoElement.cs
@@ -2,6 +2,7 @@ using BotSharp.Abstraction.Agents.Models;
 
 namespace BotSharp.Plugin.MongoStorage.Models;
 
+[BsonIgnoreExtraElements]
 public class AgentResponseMongoElement
 {
     public string Prefix { get; set; }
diff --git a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentTemplateMongoElement.cs b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentTemplateMongoElement.cs
index 719a11d9..847ec5c9 100644
--- a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentTemplateMongoElement.cs
+++ b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/AgentTemplateMongoElement.cs
@@ -2,6 +2,7 @@ using BotSharp.Abstraction.Agents.Models;
 
 namespace BotSharp.Plugin.MongoStorage.Models;
 
+[BsonIgnoreExtraElements]
 public class AgentTemplateMongoElement
 {
     public string Name { get; set; }
diff --git a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/FunctionDefMongoElement.cs b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/FunctionDefMongoElement.cs
index 08caa2a4..5910661f 100644
--- a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/FunctionDefMongoElement.cs
+++ b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/FunctionDefMongoElement.cs
@@ -3,6 +3,7 @@ using System.Text.Json;
 
 namespace BotSharp.Plugin.MongoStorage.Models;
 
+[BsonIgnoreExtraElements]
 public class FunctionDefMongoElement
 {
     public string Name { get; set; }
diff --git a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/RoutingRuleMongoElement.cs b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/RoutingRuleMongoElement.cs
index 9f644c68..1a40b0f7 100644
--- a/src/Plugins/BotSharp.Plugin.MongoStorage/Models/RoutingRuleMongoElement.cs
+++ b/src/Plugins/BotSharp.Plugin.MongoStorage/Models/RoutingRuleMongoElement.cs
@@ -2,6 +2,7 @@ using BotSharp.Abstraction.Routing.Models;
 
 namespace BotSharp.Plugin.MongoStorage.Models;
 
+[BsonIgnoreExtraElements]
 public class RoutingRuleMongoElement
 {
     public string Field { get; set; }
diff --git a/src/Plugins/BotSharp.Plugin.MongoStorage/Using.cs b/src/Plugins/BotSharp.Plugin.MongoStorage/Using.cs
index 5debd75f..ca13f204 100644
--- a/src/Plugins/BotSharp.Plugin.MongoStorage/Using.cs
+++ b/src/Plugins/BotSharp.Plugin.MongoStorage/Using.cs
@@ -10,4 +10,5 @@ global using BotSharp.Abstraction.Plugins;
 global using Microsoft.Extensions.Configuration;
 global using Microsoft.Extensions.DependencyInjection;
 global using MongoDB.Bson;
-global using MongoDB.Driver;
\ No newline at end of file
+global using MongoDB.Driver;
+global using MongoDB.Bson.Serialization.Attributes;
\ No newline at end of file