Spaces:
Running
Running
Commit
·
49c30d8
1
Parent(s):
ed60c89
Trying fix chromadb issue1
Browse files- langchain_helper.py +2 -1
langchain_helper.py
CHANGED
|
@@ -23,7 +23,7 @@ from plotly.express import bar, line, scatter, area, pie
|
|
| 23 |
# from dotenv import load_dotenv
|
| 24 |
# load_dotenv()
|
| 25 |
|
| 26 |
-
|
| 27 |
|
| 28 |
from dotenv import load_dotenv, find_dotenv
|
| 29 |
_ = load_dotenv(find_dotenv()) # read local .env file
|
|
@@ -31,6 +31,7 @@ _ = load_dotenv(find_dotenv()) # read local .env file
|
|
| 31 |
current_model_id = os.getenv('model_id')
|
| 32 |
|
| 33 |
def get_few_shot_db_chain(user_message):
|
|
|
|
| 34 |
#llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.2)
|
| 35 |
llm = ChatOpenAI(model = current_model_id)
|
| 36 |
print(llm)
|
|
|
|
| 23 |
# from dotenv import load_dotenv
|
| 24 |
# load_dotenv()
|
| 25 |
|
| 26 |
+
|
| 27 |
|
| 28 |
from dotenv import load_dotenv, find_dotenv
|
| 29 |
_ = load_dotenv(find_dotenv()) # read local .env file
|
|
|
|
| 31 |
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)
|