added chatopenai

langchain_helper.py

@@ -1,5 +1,5 @@
 import os
-from langchain_openai import AzureOpenAI, OpenAI
+from langchain_openai import AzureOpenAI, ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents import create_pandas_dataframe_agent
@@ -29,7 +29,7 @@ current_model_id = os.getenv('model_id')
 
 def get_few_shot_db_chain(user_message):
     #llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.2)
-    llm = OpenAI(model = current_model_id)
+    llm = ChatOpenAI(model = current_model_id)
     
     engine = create_engine("sqlite:///ecomm.db")
     db = SQLDatabase(engine=engine, sample_rows_in_table_info=3)