initial commit
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
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/"
|
||||
Reference in New Issue
Block a user