Spaces:
Running
Running
Update langchain_helper.py
Browse files- 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 |
-
|
| 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)
|