in src/psearch/ingestion/main.py [0:0]
def __init__(self):
# Get environment variables
self.project_id = os.environ["PROJECT_ID"]
self.location = os.environ["REGION"]
self.enrichiment = os.environ.get("ENRICHIMENT", False)
self.model_name = os.environ.get(
"DENSE_MODEL_NAME", "text-multilingual-embedding-002"
)
self.dimensions = int(os.environ.get("DENSE_DIMENSIONS", "768"))
# Initialize services
self.bq_service = BigQueryService(self.project_id)
self.spanner_service = SpannerService(self.project_id)
self.gemini_service = GeminiService(self.project_id, self.location)
# Initialize the embedding client
self._init_embedding_client()