Update chatbox.py

chatbox.py

@@ -13,7 +13,7 @@ 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 OpenAIChat
+from langchain_community.chat_models import ChatOpenAI
 import torch
 from PIL import Image, ImageDraw, ImageOps
 from transformers import pipeline, BlipProcessor, BlipForConditionalGeneration, BlipForQuestionAnswering
@@ -140,7 +140,8 @@ def build_chatbot_tools(load_dict):
 class ConversationBot:
     def __init__(self, tools, api_key=""):
         # load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
-        llm = OpenAIChat(model_name="gpt-4o", temperature=0.7, openai_api_key=api_key, model_kwargs={"api_version": "2020-11-07"})
+        print("chatbot api",api_key)
+        llm = ChatOpenAI(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