refine code hook

Author: Jicheng Lu

src/Infrastructure/BotSharp.Abstraction/Coding/Contexts/CodeExecutionContext.cs

@@ -4,4 +4,5 @@ public class CodeExecutionContext
 {
     public AgentCodeScript CodeScript { get; set; }
     public List<KeyValue> Arguments { get; set; } = [];
+    public string? InvokeFrom { get; set; }
 }

src/Infrastructure/BotSharp.Abstraction/Instructs/IInstructHook.cs

@@ -12,5 +12,5 @@ public interface IInstructHook : IHookBase
     Task OnResponseGenerated(InstructResponseModel response) => Task.CompletedTask;
 
     Task BeforeCodeExecution(Agent agent, CodeExecutionContext context) => Task.CompletedTask;
-    Task AfterCodeExecution(Agent agent, CodeExecutionResponseModel response) => Task.CompletedTask;
+    Task AfterCodeExecution(Agent agent, CodeExecutionContext context, CodeExecutionResponseModel response) => Task.CompletedTask;
 }

src/Infrastructure/BotSharp.Abstraction/Instructs/InstructHookBase.cs

@@ -28,7 +28,7 @@ public virtual async Task BeforeCodeExecution(Agent agent, CodeExecutionContext
         await Task.CompletedTask;
     }
 
-    public virtual async Task AfterCodeExecution(Agent agent, CodeExecutionResponseModel response)
+    public virtual async Task AfterCodeExecution(Agent agent, CodeExecutionContext context, CodeExecutionResponseModel response)
     {
         await Task.CompletedTask;
     }

src/Infrastructure/BotSharp.Core.Rules/Criteria/CodeScriptRuleCriteria.cs

@@ -53,26 +53,28 @@ public async Task<RuleCriteriaResult> ValidateAsync(Agent agent, IRuleTrigger tr
         {
             var hooks = _services.GetHooks<IInstructHook>(agent.Id);
 
-            var argName = context.Parameters.GetValueOrDefault("argument_name", null);
-            var argValue = context.Parameters.TryGetValue("argument_value", out var val) && val != null ? JsonSerializer.Deserialize<JsonElement>(val) : (JsonElement?)null;
+            var argName = context.Parameters.GetValueOrDefault("code_script_arg_name", null);
+            var argValue = context.Parameters.TryGetValue("code_script_arg_value", out var val) && val != null ? JsonSerializer.Deserialize<JsonElement>(val) : (JsonElement?)null;
             var arguments = BuildArguments(argName, argValue);
-            var codeExeContext = new CodeExecutionContext
+            var codeExecutionContext = new CodeExecutionContext
             {
                 CodeScript = codeScript,
-                Arguments = arguments
+                Arguments = arguments,
+                InvokeFrom = nameof(CodeScriptRuleCriteria)
             };
 
             foreach (var hook in hooks)
             {
-                await hook.BeforeCodeExecution(agent, codeExeContext);
+                await hook.BeforeCodeExecution(agent, codeExecutionContext);
             }
 
+            codeScript = codeExecutionContext.CodeScript;
             var (useLock, useProcess, timeoutSeconds) = CodingUtil.GetCodeExecutionConfig(_codingSettings);
             using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(timeoutSeconds));
             var response = processor.Run(codeScript.Content, options: new()
             {
                 ScriptName = scriptName,
-                Arguments = arguments,
+                Arguments = codeExecutionContext.Arguments,
                 UseLock = useLock,
                 UseProcess = useProcess
             }, cancellationToken: cts.Token);
@@ -87,7 +89,7 @@ public async Task<RuleCriteriaResult> ValidateAsync(Agent agent, IRuleTrigger tr
 
             foreach (var hook in hooks)
             {
-                await hook.AfterCodeExecution(agent, codeResponse);
+                await hook.AfterCodeExecution(agent, codeExecutionContext, codeResponse);
             }
 
             if (response == null || !response.Success)

src/Infrastructure/BotSharp.Core/Instructs/Services/InstructService.Execute.cs

@@ -195,7 +195,7 @@ public async Task<InstructResult> Execute(
         foreach (var hook in hooks)
         {
             await hook.AfterCompletion(agent, instructResult);
-            await hook.AfterCodeExecution(agent, codeExecution);
+            await hook.AfterCodeExecution(agent, context, codeExecution);
         }
 
         return instructResult;

src/Infrastructure/BotSharp.Logger/Hooks/InstructionLogHook.cs

@@ -1,9 +1,11 @@
+using BotSharp.Abstraction.Coding.Contexts;
 using BotSharp.Abstraction.Coding.Models;
 using BotSharp.Abstraction.Instructs.Models;
 using BotSharp.Abstraction.Instructs.Settings;
 using BotSharp.Abstraction.Loggers.Models;
 using BotSharp.Abstraction.Users;
 using BotSharp.Abstraction.Utilities;
+using System;
 
 namespace BotSharp.Logger.Hooks;
 
@@ -61,7 +63,7 @@ await db.SaveInstructionLogs(new List<InstructionLogModel>
         await base.OnResponseGenerated(response);
     }
 
-    public override async Task AfterCodeExecution(Agent agent, CodeExecutionResponseModel response)
+    public override async Task AfterCodeExecution(Agent agent, CodeExecutionContext context, CodeExecutionResponseModel response)
     {
         if (response == null || !IsLoggingEnabled(agent?.Id))
         {
@@ -88,7 +90,7 @@ await db.SaveInstructionLogs(new List<InstructionLogModel>
             }
         });
 
-        await base.AfterCodeExecution(agent, response);
+        await base.AfterCodeExecution(agent, context, response);
     }
 
     private bool IsLoggingEnabled(string? agentId)