From f5a9454aa392feab6de0d6cca1ecb439e5229af4 Mon Sep 17 00:00:00 2001
From: narawat lamaiin <narawat@outlook.com>
Date: Tue, 16 Jul 2024 18:18:08 +0700
Subject: [PATCH] update

---
 src/interface.jl | 2 +-
 test/runtest.jl  | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/interface.jl b/src/interface.jl
index 7e9b5de..5981262 100644
--- a/src/interface.jl
+++ b/src/interface.jl
@@ -1101,7 +1101,7 @@ function generatechat(a::T) where {T<:agent}
   Your earlier conversation with 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:
   - chat: what do you want to say to the user based on the current situation
diff --git a/test/runtest.jl b/test/runtest.jl
index ca98b5d..6dc7bab 100644
--- a/test/runtest.jl
+++ b/test/runtest.jl
@@ -42,7 +42,7 @@ function text2textInstructLLM(prompt::String)
     :payload=> Dict(
       :text=> prompt,
       :kwargs=> Dict(
-        :max_tokens=> 512,
+        :max_tokens=> 1024,
         :stop=> ["<|eot_id|>"],
         :temperature=> 0.2,
         )
@@ -75,7 +75,9 @@ function main()
   userinput = "Hello, I would like a get a bottle of wine."
   for i in 1:10
     response = YiemAgent.conversation(a, Dict(:text=> userinput))
+    println("")
     println("--> assistant response: \n", response)
+    println("")
     println("--> user input:")
     userinput = readline()
   end