aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:lookbook上传

Browse Source
This commit is contained in:
shahaibo
2024-10-22 00:01:27 +08:00
parent 26360dd2b5
commit 5a26642280
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/service/lookbooks/config/config.py
View File

@@ -1,15 +1,15 @@
# config.py # config.py
import os import os
import platform # import platform
if platform.system() == 'Linux': # if platform.system() == 'Linux':
__import__('pysqlite3') # __import__('pysqlite3')
import sys # import sys
sys.modules['sqlite3'] = sys.modules.pop('pysqlite3') # sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
import chromadb import chromadb
from langchain_openai import OpenAIEmbeddings from langchain_openai import OpenAIEmbeddings
from langchain_chroma import Chroma from langchain_chroma import Chroma
import tritonclient.grpc as grpcclient # import tritonclient.grpc as grpcclient
from minio import Minio # from minio import Minio
# OpenAI settings # OpenAI settings
OPENAI_API_KEY = "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" OPENAI_API_KEY = "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"
@@ -26,7 +26,7 @@ MINIO_SECURE = False
MINIO_ACCESS = "e8zc55mzDOh4IzRrZ9Oa" MINIO_ACCESS = "e8zc55mzDOh4IzRrZ9Oa"
MINIO_SECRET = "uHfqJ7UkwA1PTDGfnA44Hp9ux5YkZTkzZLjeOYhE" MINIO_SECRET = "uHfqJ7UkwA1PTDGfnA44Hp9ux5YkZTkzZLjeOYhE"
MINIO_BUCKET = "test" 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 # Set environment variables
os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
@@ -71,4 +71,4 @@ elif OP_SERVER == "aws":
# os.environ["OOTD_URL"] = "http://18.167.251.121:10001/ootd/ootd_dc" # 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" 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'])