Update chatbox.py

chatbox.py

@@ -13,11 +13,14 @@ import inspect
 from langchain.agents.initialize import initialize_agent
 from langchain.agents.tools import Tool
 from langchain.chains.conversation.memory import ConversationBufferMemory
-from langchain.llms.openai import OpenAI
+from langchain.llms.openai import OpenAIChat
 import torch
 from PIL import Image, ImageDraw, ImageOps
 from transformers import pipeline, BlipProcessor, BlipForConditionalGeneration, BlipForQuestionAnswering
 
+# openai.api_version = '2020-11-07' 
+os.environ["OPENAI_API_VERSION"] = '2020-11-07'
+
 VISUAL_CHATGPT_PREFIX = """
     I want you act as Caption Anything Chatbox (short as CATchat), which is designed to be able to assist with a wide range of text and visual related tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. You are able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
 
@@ -139,7 +142,7 @@ def build_chatbot_tools(load_dict):
 class ConversationBot:
     def __init__(self, tools, api_key="",save_history=False):
         # load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
-        llm = OpenAI(model_name="gpt-4o", temperature=0.7, openai_api_key=api_key)
+        llm = OpenAIChat(model_name="gpt-4o", temperature=0.7, openai_api_key=api_key, model_kwargs={"api_version": "2020-11-07"})
         self.llm = llm
         self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
         self.tools = tools