Update pages/Chat_Mode.py

pages/Chat_Mode.py

@@ -119,7 +119,7 @@ if prompt := st.chat_input("Type some text here!"):
         
         created = response["created"]
         model = response["model"]
-        object_value = response["object"]
+        object = response["object"]
         completion_tokens = response["usage"]["completion_tokens"]
         prompt_tokens = response["usage"]["prompt_tokens"]
         total_tokens = response["usage"]["total_tokens"]
@@ -128,7 +128,7 @@ if prompt := st.chat_input("Type some text here!"):
 
         # Insert data into the table
         insert_data_query = """
-        INSERT INTO chatbot3 (id, finish_reason, index_value, logprobs, text_line1, text_line2, created, model, object_value, completion_tokens, prompt_tokens, total_tokens)
+        INSERT INTO chatbot3 (id, finish_reason, index_value, logprobs, text_line1, text_line2, created, model, object, completion_tokens, prompt_tokens, total_tokens)
         VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s);
         """
         cursor.execute(insert_data_query, (id_value, finish_reason, index_value, logprobs, text_line1, text_line2, created, model, object_value, completion_tokens, prompt_tokens, total_tokens))