Ayomide
33 lines
761 B
Python
33 lines
761 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 = "mixtral-8x7b-32768"
|
|
|
|
# Claude
|
|
CLAUDE_API_KEY = os.getenv("CLAUDE_API_KEY")
|
|
CLAUDE_MODEL = "claude-3-5-sonnet-20240620"
|
|
|
|
# Vector Store
|
|
VECTOR_STORE_TYPE = "pinecone"
|
|
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
|
|
PINECONE_INDEX = "scp-docs"
|
|
PINECONE_ENV = "gcp-starter"
|
|
|
|
# Document Processing
|
|
MAX_DOC_SIZE = 10 * 1024 * 1024 # 10MB
|
|
ALLOWED_EXTENSIONS = {'.pdf', '.docx', '.txt'}
|
|
|
|
# Paths
|
|
UPLOAD_FOLDER = "documents/"
|