update

2024-07-18 11:22:53 +07:00
parent 0ddc98a447
commit 6e2391e0e3
4 changed files with 168 additions and 138 deletions
--- a/src/type.jl
+++ b/src/type.jl
@@ -97,10 +97,12 @@ mutable struct sommelier <: agent

  # communication function
  text2textInstructLLM::Function
+  executeSQL::Function
 end

 function sommelier(
-  text2textInstructLLM::Function
+  text2textInstructLLM::Function,
+  executeSQL::Function
  ;
  name::String= "Assistant",
  id::String= string(uuid4()),
@@ -140,6 +142,7 @@ function sommelier(
    chathistory,
    memory,
    text2textInstructLLM,
+    executeSQL
    )
  
  return newAgent