-
Notifications
You must be signed in to change notification settings - Fork 441
/
settings.py
44 lines (33 loc) · 1.34 KB
/
settings.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
from pydantic_settings import BaseSettings, SettingsConfigDict
class AppSettings(BaseSettings):
model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8")
# Embeddings config
EMBEDDING_MODEL_ID: str = "sentence-transformers/all-MiniLM-L6-v2"
EMBEDDING_MODEL_MAX_INPUT_LENGTH: int = 256
EMBEDDING_SIZE: int = 384
EMBEDDING_MODEL_DEVICE: str = "cpu"
OPENAI_MODEL_ID: str = "gpt-4-1106-preview"
OPENAI_API_KEY: str | None = None
# MongoDB configs
MONGO_DATABASE_HOST: str = "mongodb://localhost:30001,localhost:30002,localhost:30003/?replicaSet=my-replica-set"
MONGO_DATABASE_NAME: str = "scrabble"
# QdrantDB config
QDRANT_DATABASE_HOST: str = "localhost"
QDRANT_DATABASE_PORT: int = 6333
QDRANT_DATABASE_URL: str = "http://localhost:6333"
QDRANT_CLOUD_URL: str = "str"
USE_QDRANT_CLOUD: bool = False
QDRANT_APIKEY: str | None = None
# MQ config
RABBITMQ_DEFAULT_USERNAME: str = "guest"
RABBITMQ_DEFAULT_PASSWORD: str = "guest"
RABBITMQ_HOST: str = "localhost"
RABBITMQ_PORT: int = 5673
# CometML config
COMET_API_KEY: str | None = None
COMET_WORKSPACE: str | None = None
COMET_PROJECT: str | None = None
# LinkedIn credentials
LINKEDIN_USERNAME: str | None = None
LINKEDIN_PASSWORD: str | None = None
settings = AppSettings()