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

update

Browse Source
This commit is contained in:
narawat lamaiin
2024-07-16 18:18:08 +07:00
parent 1ef696f206
commit f5a9454aa3
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/interface.jl
View File

@@ -1101,7 +1101,7 @@ function generatechat(a::T) where {T<:agent}
Your earlier conversation with the user: ... Your earlier conversation with the user: ...
You must follow the following guidelines (if the user interrupts, prioritize the user): You must follow the following guidelines (if the user interrupts, prioritize the user):
- Do not recommend specific wine before you checked your inventory. - Don't mention any specific wine until you've checked your inventory, as you may or may not have it.
You should then respond to the user with: You should then respond to the user with:
- chat: what do you want to say to the user based on the current situation - chat: what do you want to say to the user based on the current situation

4
test/runtest.jl
View File

@@ -42,7 +42,7 @@ function text2textInstructLLM(prompt::String)
:payload=> Dict( :payload=> Dict(
:text=> prompt, :text=> prompt,
:kwargs=> Dict( :kwargs=> Dict(
:max_tokens=> 512, :max_tokens=> 1024,
:stop=> ["<|eot_id|>"], :stop=> ["<|eot_id|>"],
:temperature=> 0.2, :temperature=> 0.2,
) )
@@ -75,7 +75,9 @@ function main()
userinput = "Hello, I would like a get a bottle of wine." userinput = "Hello, I would like a get a bottle of wine."
for i in 1:10 for i in 1:10
response = YiemAgent.conversation(a, Dict(:text=> userinput)) response = YiemAgent.conversation(a, Dict(:text=> userinput))
println("")
println("--> assistant response: \n", response) println("--> assistant response: \n", response)
println("")
println("--> user input:") println("--> user input:")
userinput = readline() userinput = readline()
end end