diff --git a/app/service/lookbooks/config/config.py b/app/service/lookbooks/config/config.py index 7f47d6a..1ff18fc 100644 --- a/app/service/lookbooks/config/config.py +++ b/app/service/lookbooks/config/config.py @@ -1,15 +1,14 @@ -# config.py import os -# import platform -# if platform.system() == 'Linux': -# __import__('pysqlite3') -# import sys -# sys.modules['sqlite3'] = sys.modules.pop('pysqlite3') +import platform +if platform.system() == 'Linux': + __import__('pysqlite3') + import sys + sys.modules['sqlite3'] = sys.modules.pop('pysqlite3') import chromadb from langchain_openai import OpenAIEmbeddings from langchain_chroma import Chroma -# import tritonclient.grpc as grpcclient -# from minio import Minio +import tritonclient.grpc as grpcclient +from minio import Minio # OpenAI settings OPENAI_API_KEY = "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" @@ -26,7 +25,7 @@ MINIO_SECURE = False MINIO_ACCESS = "e8zc55mzDOh4IzRrZ9Oa" MINIO_SECRET = "uHfqJ7UkwA1PTDGfnA44Hp9ux5YkZTkzZLjeOYhE" MINIO_BUCKET = "test" -# MINIO_CLIENT = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE) +MINIO_CLIENT = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE) # Set environment variables os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY @@ -71,4 +70,4 @@ elif OP_SERVER == "aws": # os.environ["OOTD_URL"] = "http://18.167.251.121:10001/ootd/ootd_dc" os.environ["OOTD_URL"] = "https://muskox-many-bluegill.ngrok-free.app/ootd_dc" -# triton_client = grpcclient.InferenceServerClient(url=os.environ['GRPCCLIENT_URL']) +triton_client = grpcclient.InferenceServerClient(url=os.environ['GRPCCLIENT_URL']) diff --git a/requirements.txt b/requirements.txt index a240d8b..3a595e6 100644 Binary files a/requirements.txt and b/requirements.txt differ