Spaces:
Sleeping
Sleeping
File size: 669 Bytes
d2d5a16 a92ae62 d2d5a16 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """
Configuration settings for the RAG Agent.
These settings are used by the various RAG tools.
Vertex AI initialization is performed in the package's __init__.py
"""
import os
# Load environment variables (this is redundant if __init__.py is imported first,
# but included for safety when importing config directly)
# Vertex AI settings
PROJECT_ID = os.getenv('GOOGLE_CLOUD_PROJECT')
LOCATION = os.getenv('GOOGLE_CLOUD_LOCATION')
# RAG settings
DEFAULT_CHUNK_SIZE = 512
DEFAULT_CHUNK_OVERLAP = 100
DEFAULT_TOP_K = 3
DEFAULT_DISTANCE_THRESHOLD = 0.5
DEFAULT_EMBEDDING_MODEL = "publishers/google/models/text-embedding-005"
DEFAULT_EMBEDDING_REQUESTS_PER_MIN = 1000
|