update

src/interface.jl

@@ -318,7 +318,7 @@ julia>
 # Signature
 """
 function evaluator(state::T1, text2textInstructLLM::Function;
-  addSQLVectorDB::Union{Function, Nothing}=nothing
+  insertSQLVectorDB::Union{Function, Nothing}=nothing
   ) where {T1<:AbstractDict}
   
   # systemmsg = 
@@ -784,7 +784,7 @@ function transition(state::T, args::NamedTuple
   context = args[:context]
   executeSQL::Function = args[:executeSQL]
   text2textInstructLLM::Function = args[:text2textInstructLLM]
-  addSQLVectorDBF::Function = args[:addSQLVectorDB]
+  insertSQLVectorDB::Function = args[:insertSQLVectorDB]
   querySQLVectorDBF::Function = args[:querySQLVectorDB]
 
   # getting SQL from vectorDB
@@ -820,7 +820,7 @@ function transition(state::T, args::NamedTuple
   isterminal::Bool = haskey(response, :isterminal) ? response[:isterminal] : false
   newNodeKey, newstate = makeNewState(state, thoughtDict, rawresponse, JSON3.write(result), select, reward, isterminal)
   progressvalue::Integer = evaluatorF(newstate, text2textInstructLLM; 
-                                      addSQLVectorDB=addSQLVectorDBF)
+                                    insertSQLVectorDB=insertSQLVectorDB)
 
   return (newNodeKey=newNodeKey, newstate=newstate, progressvalue=progressvalue)
 end
@@ -903,7 +903,7 @@ julia> println(result)
 # Signature
 """
 function query(query::T, executeSQL::Function, text2textInstructLLM::Function;
-  addSQLVectorDB::Union{Function, Nothing}=nothing,
+  insertSQLVectorDB::Union{Function, Nothing}=nothing,
   similarSQLVectorDB::Union{Function, Nothing}=nothing,
   ) where {T<:AbstractString}
 
@@ -947,7 +947,7 @@ function query(query::T, executeSQL::Function, text2textInstructLLM::Function;
                     executeSQL=executeSQL, 
                     text2textInstructLLM=text2textInstructLLM,
                     querySQLVectorDB=similarSQLVectorDB,
-                    addSQLVectorDB=addSQLVectorDB,
+                    insertSQLVectorDB=insertSQLVectorDB,
                   )
 
   earlystop(state) = state[:reward] >= 8 ? true : false
@@ -961,10 +961,10 @@ function query(query::T, executeSQL::Function, text2textInstructLLM::Function;
   extracted = resultState[:thoughtHistory][latestKey]
 
   # add to vectorDB only if the answer is achieved and the state is terminal 
-  if addSQLVectorDB !== nothing && resultState[:isterminal] == true &&
+  if insertSQLVectorDB !== nothing && resultState[:isterminal] == true &&
     resultState[:rawresponse] !== nothing
 
-    addSQLVectorDB(resultState[:thoughtHistory][:question], sql)
+    insertSQLVectorDB(resultState[:thoughtHistory][:question], sql)
   end
 
   return (text=extracted, rawresponse=resultState[:rawresponse])