From 14c881741edf27fed6e28b18b5b5835ba35054b6 Mon Sep 17 00:00:00 2001
From: tonaerospace <tonaerospace.etc@gmail.com>
Date: Sun, 16 Mar 2025 22:11:23 +0700
Subject: [PATCH] update

---
 src/interface.jl |  2 +-
 test/runtest.jl  |  2 +-
 test/test_2.jl   | 43 +++++++++++++++++++++++++------------------
 3 files changed, 27 insertions(+), 20 deletions(-)

diff --git a/src/interface.jl b/src/interface.jl
index 87b1c8d..20d5cf6 100644
--- a/src/interface.jl
+++ b/src/interface.jl
@@ -871,7 +871,7 @@ function query(query::T, executeSQL::Function, text2textInstructLLM::Function;
 
   earlystop(state) = state[:reward] >= 8 ? true : false
 
-  _, _, resultState = LLMMCTS.runMCTS(initialstate, transition, transitionargs;
+  root, _, resultState = LLMMCTS.runMCTS(initialstate, transition, transitionargs;
                       horizontalSampleExpansionPhase=5, 
                       horizontalSampleSimulationPhase=2,
                       maxSimulationDepth=5, 
diff --git a/test/runtest.jl b/test/runtest.jl
index 31b0b20..00bfb58 100644
--- a/test/runtest.jl
+++ b/test/runtest.jl
@@ -157,7 +157,7 @@ sessionId = "555"
 
 
 # query = Dict(:text=> "How many wines from France do you have that can be paired with lamb?")
-query = "How many French wines do you have?"
+query = "How many German wines do you have?"
 # query = "retailer: Yiem, wine_type: red, sweetness: 1-2, intensity: 4-5, wine price: 20-40"
 # query = "wine_type: white, country: United States, sweetness: 1-2, tannin: 3, food to be served with wine: pizza"
 # query = "wine_type: white, country: Austria, food to be served with wine: pork"
diff --git a/test/test_2.jl b/test/test_2.jl
index a099cf6..26b94b1 100644
--- a/test/test_2.jl
+++ b/test/test_2.jl
@@ -1,23 +1,30 @@
 using Revise
+using SQLLLM, LLMMCTS, DataStructures
 
-function testf(a)::NamedTuple{(:a, :b), Tuple{Union{Nothing, Int}, Int}}
-  if a == 1
-    return (a=nothing, b=5)
-  else
-    return (a=5, b=5)
-  end
+initialstate = Dict{Symbol, Any}(
+    :reward=> 0,
+    :isterminal=> false,
+    :evaluation=> "None",
+    :evaluationscore=> 0,
+    :suggestion=> "None",
+    :accepted_as_answer=> "No",
+    :lesson=> nothing,
+
+    # contain question, thought_1, action_1, observation_1, thought_2, ...
+    :thoughtHistory=> OrderedDict{Symbol, Any}( 
+      #[] :recap=>,
+      :question=> "query",
+      ),
+    )
+
+root = MCTSNode("root", initialstate, 0, 0, 0, 0, false, nothing, Dict{String,MCTSNode}(),
+  Dict{Symbol,Any}())
+
+field = fieldnames(typeof(root))
+for f in field
+  println(getfield(root, f))
+  setfield!(root, f, 1)
 end
 
 
-q = testf(1)
-w = testf(2)
-
-
-
-
-
-
-
-
-
-
+println("\n done")
\ No newline at end of file