Update app.py

app.py

@@ -134,8 +134,8 @@ def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_out
                   "prompt": prompt if (str(err_msg) == "") else "",
         },
         outputs = {"result": result if (str(err_msg) == "") else "",
-                   "generation_info": generation_info if (str(err_msg) == "") else "",
-                   "llm_output": llm_output if (str(err_msg) == "") else "",
+                   "generation_info": str(generation_info) if (str(err_msg) == "") else "",
+                   "llm_output": str(llm_output) if (str(err_msg) == "") else "",
                    "completion": str(completion) if (str(err_msg) == "") else "",
         },
         model_dict = {"llm_client": (str(chain.llm.client) if (rag_option == RAG_OFF) else
@@ -188,13 +188,8 @@ def invoke(openai_api_key, rag_option, prompt):
             completion, chain = llm_chain(llm, prompt)
             if (completion.generations[0] != None and completion.generations[0][0] != None):
                 result = completion.generations[0][0].text
-                print("###")
-                print(completion)
-                print("###")
-                print(completion.generations[0])
-                print("###")
                 generation_info = completion.generations[0][0].generation_info
-                #llm_output = completion.generations.llm_output
+                llm_output = completion.llm_output
     except Exception as e:
         err_msg = e
         raise gr.Error(e)