DocsGPT/application/core/settings.py

26 lines
970 B
Python
Raw Normal View History

2023-04-29 14:40:55 +00:00
from pathlib import Path
from pydantic import BaseSettings
2023-04-29 14:40:55 +00:00
class Settings(BaseSettings):
2023-04-29 14:44:47 +00:00
LLM_NAME: str = "openai_chat"
2023-04-29 14:46:09 +00:00
EMBEDDINGS_NAME: str = "openai_text-embedding-ada-002"
2023-04-30 10:03:09 +00:00
CELERY_BROKER_URL: str = "redis://localhost:6379/0"
CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1"
MONGO_URI: str = "mongodb://localhost:27017/docsgpt"
2023-05-25 18:33:37 +00:00
MODEL_PATH: str = "./models/gpt4all-model.bin"
TOKENS_MAX_HISTORY: int = 150
2023-04-30 09:46:52 +00:00
API_URL: str = "http://localhost:5001" # backend url for celery worker
2023-04-29 14:56:32 +00:00
2023-04-30 09:46:52 +00:00
API_KEY: str = None # LLM api key
EMBEDDINGS_KEY: str = None # api key for embeddings (if using openai, just copy API_KEY
OPENAI_API_BASE: str = None # azure openai api base url
OPENAI_API_VERSION: str = None # azure openai api version
AZURE_DEPLOYMENT_NAME: str = None # azure deployment name
2023-04-29 14:40:55 +00:00
path = Path(__file__).parent.parent.absolute()
settings = Settings(_env_file=path.joinpath(".env"), _env_file_encoding="utf-8")