Update app.py

app.py

@@ -1,5 +1,8 @@
 import io
 import os
+import promptlayer
+promptlayer.api_key = os.environ.get("PROMPTLAYER_KEY")
+
 import ssl
 from contextlib import closing
 from typing import Optional, Tuple
@@ -17,7 +20,8 @@ from langchain import ConversationChain, LLMChain
 
 from langchain.agents import load_tools, initialize_agent
 from langchain.chains.conversation.memory import ConversationBufferMemory
-from langchain.llms import OpenAI
+#  from langchain.llms import OpenAI
+from promptlayer.langchain.llms import OpenAI
 from threading import Lock
 
 # Console to variable
@@ -48,7 +52,7 @@ tmdb_bearer_token = os.environ["TMDB_BEARER_TOKEN"]
 
 TOOLS_LIST = ['serpapi', 'wolfram-alpha', 'pal-math', 'pal-colored-objects', 'news-api']  #'google-search','news-api','tmdb-api','open-meteo-api'
 TOOLS_DEFAULT_LIST = ['serpapi', 'wolfram-alpha', 'pal-math', 'pal-colored-objects', 'news-api']
-BUG_FOUND_MSG = "Congratulations, you've found a bug in this application!"
+BUG_FOUND_MSG = "Error in the return response. Please try again."
 # AUTH_ERR_MSG = "Please paste your OpenAI key from openai.com to use this application. It is not necessary to hit a button or key after pasting it."
 AUTH_ERR_MSG = "Please paste your OpenAI key from openai.com to use this application. "
 MAX_TOKENS = 2048
@@ -285,6 +289,7 @@ def set_openai_api_key(api_key):
         return chain, express_chain, llm, embeddings, qa_chain, memory
     return None, None, None, None, None, None
 
+PROMPTLAYER_API_BASE = "https://api.promptlayer.com"
 
 def run_chain(chain, inp, capture_hidden_text):
     output = ""