This commit is contained in:
narawat lamaiin
2024-04-29 20:55:02 +07:00
parent 8f95f08695
commit 3316ec4ecf
4 changed files with 84 additions and 11 deletions

View File

@@ -136,7 +136,7 @@ function expand(a::T1, node::MCTSNode, state::T2, decisionMaker::Function, state
action = _action[:action]
actioninput = _action[:input]
newState = transition(a, node.state, action, actioninput) #[] Implement your transition function
newState = MCTStransition(a, node.state, action, actioninput) #[] Implement your transition function
if newState keys(node.children)
node.children[newState] = MCTSNode(newState, 0, 0.0, Dict{T, MCTSNode}())
@@ -230,14 +230,14 @@ julia>
# Signature
"""
function transition(a::T1, state::T2, action::T3,
function MCTStransition(a::T1, state::T2, action::T3,
actioninput::T3) where {T1<:agent, T2<:AbstractDict, T3<:AbstractString}
# map action and input() to llm function
result =
if action == "chatbox"
chatbox(a, input)
virtualWineCustomerChatbox(a, input) # user virtu
elseif action == "winestock"
elseif action == "finish"