binaychandra commited on
Commit
3be028e
·
verified ·
1 Parent(s): c43f8c9

Update langchain_helper.py

Browse files
Files changed (1) hide show
  1. langchain_helper.py +2 -2
langchain_helper.py CHANGED
@@ -32,8 +32,8 @@ current_model_id = os.getenv('MODEL_ID')
32
 
33
  def get_few_shot_db_chain(user_message):
34
  chromadb.api.client.SharedSystemClient.clear_system_cache()
35
- #llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.2)
36
- llm = ChatOpenAI(model = current_model_id)
37
  print(llm)
38
  engine = create_engine("sqlite:///ecomm.db")
39
  db = SQLDatabase(engine=engine, sample_rows_in_table_info=3)
 
32
 
33
  def get_few_shot_db_chain(user_message):
34
  chromadb.api.client.SharedSystemClient.clear_system_cache()
35
+ llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.2)
36
+ #llm = ChatOpenAI(model = current_model_id)
37
  print(llm)
38
  engine = create_engine("sqlite:///ecomm.db")
39
  db = SQLDatabase(engine=engine, sample_rows_in_table_info=3)