diff --git a/src/llmfunction.jl b/src/llmfunction.jl index 3c1e0a0..8afd118 100644 --- a/src/llmfunction.jl +++ b/src/llmfunction.jl @@ -308,11 +308,8 @@ function checkwine!(a::T, thoughtdict::AbstractDict; useSQLLLM::Bool=false # direct query with possible sql instead of SQLLLM. sql = generatesql(a, inventoryquery) - textresult, result_raw, _, _ = SQLexecution(a.context.executeSQL, sql) - thoughtdict["action_result"] = textresult - end return (thoughtdict=thoughtdict, result_raw=result_raw) @@ -482,8 +479,14 @@ function generatesql(a::T, searchterm::String, requiredKeys = ["plan", "action_name", "action_input"] errornote = "" # provide similar sql only for the first attempt - sql, _ = a.context.similarSQLVectorDB(searchterm) + sql, distance = a.context.similarSQLVectorDB(searchterm) similarSQL_ = sql !== nothing ? sql : "None" + + # if sql is really close, just use it + if distance <= 0.1 + return similarSQL_ + end + context = """