wehub-resource-sync
e768098d0e
tools_continuous_delivery / Private PyPI non-main branch release (push) Has been skipped
tools_continuous_delivery / Private PyPI main branch release (push) Failing after 2m42s
Publish Promptflow Doc / Build (push) Has been cancelled
Publish Promptflow Doc / Deploy (push) Has been cancelled
Flake8 Lint / flake8 (push) Has been cancelled
Spell check CI / Spell_Check (push) Has been cancelled
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import os
|
|
from typing import Union
|
|
|
|
from promptflow.core import tool
|
|
from promptflow.connections import AzureOpenAIConnection, OpenAIConnection
|
|
|
|
from chat_with_pdf.utils.lock import acquire_lock
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__)) + "/chat_with_pdf/"
|
|
|
|
|
|
@tool
|
|
def setup_env(connection: Union[AzureOpenAIConnection, OpenAIConnection], config: dict):
|
|
if not connection or not config:
|
|
return
|
|
|
|
if isinstance(connection, AzureOpenAIConnection):
|
|
os.environ["OPENAI_API_TYPE"] = "azure"
|
|
os.environ["OPENAI_API_BASE"] = connection.api_base
|
|
os.environ["OPENAI_API_KEY"] = connection.api_key
|
|
os.environ["OPENAI_API_VERSION"] = connection.api_version
|
|
|
|
if isinstance(connection, OpenAIConnection):
|
|
os.environ["OPENAI_API_KEY"] = connection.api_key
|
|
if connection.organization is not None:
|
|
os.environ["OPENAI_ORG_ID"] = connection.organization
|
|
|
|
for key in config:
|
|
os.environ[key] = str(config[key])
|
|
|
|
with acquire_lock(BASE_DIR + "create_folder.lock"):
|
|
if not os.path.exists(BASE_DIR + ".pdfs"):
|
|
os.mkdir(BASE_DIR + ".pdfs")
|
|
if not os.path.exists(BASE_DIR + ".index/.pdfs"):
|
|
os.makedirs(BASE_DIR + ".index/.pdfs")
|
|
|
|
return "Ready"
|