TASK:lookbook上传

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'])