object-memory / core /config.py
russ4stall
fresh history
24f3fb6
raw
history blame contribute delete
927 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
# Qdrant
QDRANT_HOST = os.getenv("QDRANT_HOST")
QDRANT_API = os.getenv("QDRANT_API")
QDRANT_COLLECTION = os.getenv("QDRANT_COLLECTION", "object_collection")
# S3
S3_BUCKET = os.getenv("S3_BUCKET", "object-mem")
AWS_KEY = os.getenv("S3_ACCESS_KEY")
AWS_SECRET = os.getenv("S3_SECRET_KEY")
AWS_REGION = os.getenv("S3_REGION", "us-east-1")
# Neo4j
NEO4J_URI = os.getenv("NEO4J_URI")
NEO4J_USER = os.getenv("NEO4J_USER")
NEO4J_PASS = os.getenv("NEO4J_PASSWORD")
# SAM2
SAM_CHECKPT = os.getenv("SAM_CHECKPT", "./checkpoints/sam2.1_hiera_small.pt")
SAM_CFG = os.getenv("SAM_CFG", "configs/sam2.1/sam2.1_hiera_s.yaml")
# GroundingDINO
DINO_CHECKPT = os.getenv("DINO_CHECKPT", "./checkpoints/groundingdino_swint_ogc.pth")
DINO_CFG = os.getenv("DINO_CFG", "./configs/GroundingDINO_SwinT_OGC.py")
# CLIP
CLIP_MODEL = os.getenv("CLIP_MODEL", "ViT-B/32")