ton/YiemAgent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 6 Wiki Activity

update

Browse Source
This commit is contained in:
narawat
2026-07-12 05:54:47 +07:00
parent 688a8c4df2
commit 3c72373b85
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/llmfunction.jl
+7 -4
View File
@@ -308,11 +308,8 @@ function checkwine!(a::T, thoughtdict::AbstractDict; useSQLLLM::Bool=false
# direct query with possible sql instead of SQLLLM. # direct query with possible sql instead of SQLLLM.
sql = generatesql(a, inventoryquery) sql = generatesql(a, inventoryquery)
textresult, result_raw, _, _ = SQLexecution(a.context.executeSQL, sql) textresult, result_raw, _, _ = SQLexecution(a.context.executeSQL, sql)
thoughtdict["action_result"] = textresult thoughtdict["action_result"] = textresult
end end
return (thoughtdict=thoughtdict, result_raw=result_raw) return (thoughtdict=thoughtdict, result_raw=result_raw)
@@ -482,8 +479,14 @@ function generatesql(a::T, searchterm::String,
requiredKeys = ["plan", "action_name", "action_input"] requiredKeys = ["plan", "action_name", "action_input"]
errornote = "" errornote = ""
# provide similar sql only for the first attempt # provide similar sql only for the first attempt
sql, _ = a.context.similarSQLVectorDB(searchterm) sql, distance = a.context.similarSQLVectorDB(searchterm)
similarSQL_ = sql !== nothing ? sql : "None" similarSQL_ = sql !== nothing ? sql : "None"
# if sql is really close, just use it
if distance <= 0.1
return similarSQL_
end
context = context =
""" """
<internal_context_for_assistant> <internal_context_for_assistant>