wehub-resource-sync
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from backend.rag.index_manager import IndexManager
|
|
from backend.rag.prompts import Prompts
|
|
from common.utils import load_config
|
|
|
|
config = load_config()
|
|
|
|
PROJECT_ID = config["project_id"]
|
|
LOCATION = config["location"]
|
|
DOCSTORE_BUCKET_NAME = config["docstore_bucket_name"]
|
|
VECTOR_INDEX_NAME = config["vector_index_name"]
|
|
INDEX_ENDPOINT_NAME = config["index_endpoint_name"]
|
|
EMBEDDINGS_MODEL_NAME = config["embeddings_model_name"]
|
|
VECTOR_DATA_PREFIX = config["vector_data_prefix"]
|
|
FIRESTORE_DB_NAME = config["firestore_db_name"]
|
|
FIRESTORE_NAMESPACE = config["firestore_namespace"]
|
|
QA_INDEX_NAME = config.get("qa_index_name")
|
|
QA_ENDPOINT_NAME = config.get("qa_endpoint_name")
|
|
FIRESTORE_DB_NAME = config.get("firestore_db_name")
|
|
FIRESTORE_NAMESPACE = config.get("firestore_namespace")
|
|
BUCKET_NAME = config.get("docstore_bucket_name")
|
|
|
|
# Initialize State of Prompts and Indexes
|
|
|
|
prompts = Prompts()
|
|
index_manager = IndexManager(
|
|
project_id=PROJECT_ID,
|
|
location=LOCATION,
|
|
base_index_name=VECTOR_INDEX_NAME,
|
|
base_endpoint_name=INDEX_ENDPOINT_NAME,
|
|
qa_index_name=QA_INDEX_NAME,
|
|
qa_endpoint_name=QA_ENDPOINT_NAME,
|
|
embeddings_model_name=EMBEDDINGS_MODEL_NAME,
|
|
firestore_db_name=FIRESTORE_DB_NAME,
|
|
firestore_namespace=FIRESTORE_NAMESPACE,
|
|
vs_bucket_name=BUCKET_NAME,
|
|
)
|