%%capture
%pip install python-dotenv

from dotenv import load_dotenv
import os

load_dotenv()

DBX_HOST = os.environ.get("DATABRICKS_HOST", None)

UC_MANAGED_LOCATION = "s3://synaptiq-databricks-default-storage-location/"
UC_NAME = "dbr_ragchat"
SCHEMA_NAME_BRONZE = "bronze"
SCHEMA_NAME_SILVER = "silver"
SCHEMA_NAME_GOLD = "gold"

if DBX_HOST == "BAL_FAKE":
  UC_MANAGED_LOCATION = None
  UC_NAME = "dev"
  PREFIX = "ragchat_"
  SCHEMA_NAME_BRONZE = "bronze_data_products"
  SCHEMA_NAME_SILVER = "bronze_data_products"
  SCHEMA_NAME_GOLD = "bronze_data_products"
if DBX_HOST == "BAL":
  UC_MANAGED_LOCATION = None
  UC_NAME = "dev"
  PREFIX = "ragchat_"
  SCHEMA_NAME_BRONZE = "bronze_data_products"
  SCHEMA_NAME_SILVER = "bronze_data_products"
  SCHEMA_NAME_GOLD = "bronze_data_products"
  # SCHEMA_NAME_BRONZE = "bronze_ragchat"
  # SCHEMA_NAME_SILVER = "silver_ragchat"
  # SCHEMA_NAME_GOLD = "gold_ragchat"
  BRONZE_INTERCOM_CONVERSATIONS_TABLE = f"prd.bronze_intercom.conversations"

PREFIX = ""
BRONZE_FAKE_CONVERSATIONS_TABLE_NAME = f"{PREFIX}fake_intercom_conversations"
SILVER_CONVERSATIONS_TABLE_NAME = f"{PREFIX}full_intercom_conversations"
SILVER_CLEANED_CONVERSATIONS_TABLE_NAME = f"{PREFIX}full_cleaned_intercom_conversations"
SILVER_SUMMARIZED_CONVERSATIONS_TABLE_NAME = f"{PREFIX}summarized_intercom_conversations"

VECTOR_SEARCH_ENDPOINT_NAME = f"{PREFIX}vs_endpoint"
VECTOR_STORAGE_NAME = f"{PREFIX}vector_storage"
VECTOR_SEARCH_INDEX_NAME = f"{UC_NAME}.{SCHEMA_NAME_GOLD}.{VECTOR_STORAGE_NAME}"
VECTOR_SEARCH_SOURCE_TABLE_NAME = f"{UC_NAME}.{SCHEMA_NAME_SILVER}.{SILVER_SUMMARIZED_CONVERSATIONS_TABLE_NAME}"

MODEL_NAME = "dbrx_ragchat"
MODEL_NAME_FQN = f"{UC_NAME}.{SCHEMA_NAME_GOLD}.{MODEL_NAME}"