mirror of
https://github.com/arc53/DocsGPT
synced 2024-11-02 03:40:17 +00:00
44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
from pathlib import Path
|
|
import os
|
|
|
|
from pydantic import BaseSettings
|
|
current_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
LLM_NAME: str = "openai"
|
|
EMBEDDINGS_NAME: str = "openai_text-embedding-ada-002"
|
|
CELERY_BROKER_URL: str = "redis://localhost:6379/0"
|
|
CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1"
|
|
MONGO_URI: str = "mongodb://localhost:27017/docsgpt"
|
|
MODEL_PATH: str = os.path.join(current_dir, "models/docsgpt-7b-f16.gguf")
|
|
TOKENS_MAX_HISTORY: int = 150
|
|
UPLOAD_FOLDER: str = "inputs"
|
|
VECTOR_STORE: str = "faiss" # "faiss" or "elasticsearch"
|
|
|
|
API_URL: str = "http://localhost:7091" # backend url for celery worker
|
|
|
|
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 for answering
|
|
AZURE_EMBEDDINGS_DEPLOYMENT_NAME: str = None # azure deployment name for embeddings
|
|
|
|
# elasticsearch
|
|
ELASTIC_CLOUD_ID: str = None # cloud id for elasticsearch
|
|
ELASTIC_USERNAME: str = None # username for elasticsearch
|
|
ELASTIC_PASSWORD: str = None # password for elasticsearch
|
|
ELASTIC_URL: str = None # url for elasticsearch
|
|
ELASTIC_INDEX: str = "docsgpt" # index name for elasticsearch
|
|
|
|
# SageMaker config
|
|
SAGEMAKER_ENDPOINT: str = None # SageMaker endpoint name
|
|
SAGEMAKER_REGION: str = None # SageMaker region name
|
|
SAGEMAKER_ACCESS_KEY: str = None # SageMaker access key
|
|
SAGEMAKER_SECRET_KEY: str = None # SageMaker secret key
|
|
|
|
|
|
path = Path(__file__).parent.parent.absolute()
|
|
settings = Settings(_env_file=path.joinpath(".env"), _env_file_encoding="utf-8")
|