Ayomide
33 lines
788 B
Python
33 lines
788 B
Python
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
class Config:
|
|
# Cohere
|
|
COHERE_API_KEY = os.getenv("COHERE_API_KEY")
|
|
EMBED_MODEL = "embed-english-v3.0"
|
|
RERANK_MODEL = "rerank-english-v3.0"
|
|
|
|
# Groq
|
|
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
|
GROQ_MODEL = "llama3-70b-8192"
|
|
|
|
# Claude
|
|
CLAUDE_API_KEY = os.getenv("CLAUDE_API_KEY")
|
|
CLAUDE_MODEL = "claude-3-5-sonnet-20241022"
|
|
|
|
# Vector Store
|
|
VECTOR_STORE_TYPE = "pinecone"
|
|
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
|
|
PINECONE_INDEX = "scp-docs"
|
|
PINECONE_ENV = "gcp-starter"
|
|
EMBEDDING_DIMENSION = 1024
|
|
|
|
# Document Processing
|
|
MAX_DOC_SIZE = 10 * 1024 * 1024 # 10MB
|
|
ALLOWED_EXTENSIONS = {'.pdf', '.docx', '.txt'}
|
|
|
|
# Paths
|
|
UPLOAD_FOLDER = "documents/" |