fix upstage api key check error

app.py

@@ -10,8 +10,7 @@ from autorag.data.qa.schema import Raw
 from llama_index.llms.openai import OpenAI
 
 from src.create import default_create, fast_create, advanced_create
-from src.util import on_submit_openai_key
-
+from src.util import on_submit_openai_key, on_submit_llama_cloud_key, on_submit_upstage_key
 
 root_dir = os.path.dirname(os.path.realpath(__file__))
 FILE_DIR = os.path.join(root_dir, "file_cache")
@@ -233,6 +232,8 @@ with gr.Blocks(theme="earneleh/paris") as demo:
 	# API Key visibility
 	parse_choice.change(change_visible_status_api_key, inputs=[parse_choice],
 						outputs=[llama_cloud_api_key_row, upstage_api_key_row])
+	llama_key_textbox.submit(on_submit_llama_cloud_key, inputs=[llama_key_textbox], outputs=llama_key_status_box)
+	upstage_key_textbox.submit(on_submit_upstage_key, inputs=[upstage_key_textbox], outputs=upstage_key_status_box)
 
 
 demo.launch(share=False, debug=True, allowed_paths=[FILE_DIR, DATA_DIR])

src/util.py

@@ -52,10 +52,10 @@ def on_submit_upstage_key(upstage_key):
         response: ChatResponse = llm.chat(messages=[
             ChatMessage(role="system", content="You are a helpful assistant."),
             ChatMessage(role="user", content="Hi, how are you?")
-        ], max_token=3)
+        ], max_tokens=3)
 
-        assert isinstance(response.message, str)
-        assert bool(response.message)
+        assert isinstance(response.message.content, str)
+        assert bool(response.message.content)
         return "Setting complete."
     except:
         gr.Error("Upstage API key is invalid.", duration=3)