Update lang_assistant/langhelper.py

lang_assistant/langhelper.py

@@ -22,11 +22,13 @@ from langchain.chat_models import ChatOpenAI
 from langchain.prompts.chat import SystemMessage, HumanMessagePromptTemplate, HumanMessage, AIMessage
 from langchain.agents import create_pandas_dataframe_agent
 
-model_id = 'gpt-4o-mini'
+#current_model_id = 'gpt-4o-mini'
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv()) # read local .env file
 
+current_model_id = os.getenv('model_id')
+
 def summary_extractor_from_df(df:pd.DataFrame)-> str:
 
     try:
@@ -62,7 +64,7 @@ def chat_with_df(query, table_name = None):
         df = pd.read_csv(filename)
 
     agent = create_pandas_dataframe_agent(                      
-                ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613"),
+                ChatOpenAI(temperature=0, model=current_model_id),
                 df,
                 agent_type=AgentType.OPENAI_FUNCTIONS,
                 verbose=True
@@ -71,7 +73,7 @@ def chat_with_df(query, table_name = None):
         res_query = agent.run(query)
     except Exception as e:
         print(str(e))
-        res_query = "ERROR"
+        res_query = f"ERROR : {str(e)[:150]}"
 
     return res_query
 
@@ -225,12 +227,12 @@ def generate_graphdata(query, table_name = None):
                                                     reference_data=jsondf, 
                                                     format_instructions=format_instructions)
 
-    llm = ChatOpenAI(model = 'gpt-3.5-turbo-0613', temperature = 0)
+    llm = ChatOpenAI(model = current_model_id, temperature = 0)
     response = llm(customer_messages)
     json_pandassteps_charttype = output_parser.parse(response.content)
     print("pandas steps and charttype" , json_pandassteps_charttype)
     agent = create_pandas_dataframe_agent(                 
-                ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613"),
+                ChatOpenAI(temperature=0, model=current_model_id),
                 df, 
                 agent_type=AgentType.OPENAI_FUNCTIONS,
                 verbose=True,