Trying fix chromadb issue

langchain_helper.py

@@ -15,6 +15,7 @@ from sqlalchemy import create_engine
 from project_prompts import sqlite_prompt
 from few_shots import few_shots
 import pandas as pd
+import chromadb
 import plotly
 import plotly.express as px
 from plotly.express import bar, line, scatter, area, pie
@@ -22,6 +23,8 @@ from plotly.express import bar, line, scatter, area, pie
 # from dotenv import load_dotenv
 # load_dotenv()
 
+chromadb.api.client.SharedSystemClient.clear_system_cache()
+
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv()) # read local .env file
 
@@ -77,7 +80,7 @@ def get_few_shot_db_chain(user_message):
     
 
 def get_graph_details(usermessage:str, df=None):
-    llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.15)
+    llm = ChatOpenAI(deployment_name=current_model_id, temperature=0.15)
     template = ChatPromptTemplate.from_messages(
         [("system", "You are a visualisation expert and plotly developer, your task is to come up with best suitable \
                      chart representing user ask for the given data. please use plotly express library in python for \