|
|
@@ -30,6 +30,11 @@ from openai import OpenAI
|
|
|
api_key = os.getenv('api_key')
|
|
|
base_url = os.getenv('base_url')
|
|
|
model_name = os.getenv('model')
|
|
|
+host = os.getenv('host')
|
|
|
+dbname = os.getenv('dbname')
|
|
|
+user = os.getenv('user')
|
|
|
+password = os.getenv('password')
|
|
|
+
|
|
|
|
|
|
class MyVanna(ChromaDB_VectorStore, OpenAI_Chat):
|
|
|
def __init__(self, client=None,config=None):
|
|
|
@@ -55,7 +60,7 @@ vn =MyVanna(client=client,config={'model':model_name})
|
|
|
|
|
|
# vn.connect_to_mysql(host='192.168.3.86', dbname='digitization_data', user='root', password='123456', port=3306)
|
|
|
# vn.connect_to_mysql(host='192.168.3.91', dbname='digitization_data', user='root', password='2099citu##$$**.com', port=3306)
|
|
|
-vn.connect_to_mysql(host='1.14.244.91', dbname='digitization_data', user='digitalparadigm', password='CT2099citu##$$**.com',port=3306)
|
|
|
+vn.connect_to_mysql(host=host, dbname=dbname, user=user, password=password,port=3306)
|
|
|
# NO NEED TO CHANGE ANYTHING BELOW THIS LINE
|
|
|
def requires_cache(fields):
|
|
|
def decorator(f):
|
