update

2024-08-25 11:50:50 +07:00
parent 2e90665fb6
commit 232202d1ca
4 changed files with 149 additions and 21 deletions
--- a/src/type.jl
+++ b/src/type.jl
@@ -98,11 +98,15 @@ mutable struct sommelier <: agent
  # communication function
  text2textInstructLLM::Function
  executeSQL::Function
+  querySQLVectorDB::Function
+  addSQLVectorDB::Function
 end

 function sommelier(
  text2textInstructLLM::Function,
-  executeSQL::Function
+  executeSQL::Function,
+  querySQLVectorDB::Function,
+  addSQLVectorDB::Function
  ;
  name::String= "Assistant",
  id::String= string(uuid4()),
@@ -143,7 +147,9 @@ function sommelier(
    chathistory,
    memory,
    text2textInstructLLM,
-    executeSQL
+    executeSQL,
+    querySQLVectorDB,
+    addSQLVectorDB
    )
  
  return newAgent