wehub-resource-sync
23 lines
645 B
Python
23 lines
645 B
Python
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# API Keys
|
|
ASSEMBLYAI_API_KEY = os.getenv("ASSEMBLYAI_API_KEY", "")
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
|
|
|
# LLM Configuration
|
|
LLM_MODEL = os.getenv("LLM_MODEL", "gpt-4o-mini")
|
|
|
|
# Milvus Configuration
|
|
MILVUS_HOST = os.getenv("MILVUS_HOST", "localhost")
|
|
MILVUS_PORT = int(os.getenv("MILVUS_PORT", "19530"))
|
|
COLLECTION_NAME = os.getenv("MILVUS_COLLECTION_NAME", "multimodal_rag")
|
|
|
|
# Embeddings Configuration
|
|
EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "text-embedding-3-small")
|
|
EMBEDDING_DIM = int(os.getenv("EMBEDDING_DIMENSION", "1536"))
|
|
|
|
# Data Directory
|
|
DATA_DIR = "data" |