mirror of
https://github.com/tcsenpai/pensieve.git
synced 2025-06-07 03:35:24 +00:00
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import os
|
|
from pathlib import Path
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class VLMSettings(BaseModel):
|
|
enabled: bool = True
|
|
modelname: str = "moondream"
|
|
endpoint: str = "https://localhost:11434"
|
|
token: str = ""
|
|
concurrency: int = 4
|
|
|
|
|
|
class OCRSettings(BaseModel):
|
|
enabled: bool = True
|
|
endpoint: str = "http://localhost:5555/predict"
|
|
token: str = ""
|
|
concurrency: int = 4
|
|
|
|
|
|
class EmbeddingSettings(BaseModel):
|
|
num_dim: int = 768
|
|
ollama_endpoint: str = "http://localhost:11434"
|
|
ollama_model: str = "jina/jina-embeddings-v2-base-en"
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
model_config = SettingsConfigDict(
|
|
yaml_file=str(Path.home() / ".memos" / "config.yaml"),
|
|
yaml_file_encoding="utf-8",
|
|
env_prefix="MEMOS_",
|
|
)
|
|
|
|
base_dir: str = str(Path.home() / ".memos")
|
|
database_path: str = os.path.join(base_dir, "database.db")
|
|
typesense_host: str = "localhost"
|
|
typesense_port: str = "8108"
|
|
typesense_protocol: str = "http"
|
|
typesense_api_key: str = "xyz"
|
|
typesense_connection_timeout_seconds: int = 10
|
|
typesense_collection_name: str = "entities"
|
|
|
|
# Server settings
|
|
server_port: int = 8080
|
|
|
|
# VLM plugin settings
|
|
vlm: VLMSettings = VLMSettings()
|
|
|
|
# OCR plugin settings
|
|
ocr: OCRSettings = OCRSettings()
|
|
|
|
# Embedding settings
|
|
embedding: EmbeddingSettings = EmbeddingSettings()
|
|
|
|
|
|
settings = Settings()
|
|
|
|
# Define the default database path
|
|
os.makedirs(settings.base_dir, exist_ok=True)
|
|
|
|
# Global variable for Typesense collection name
|
|
TYPESENSE_COLLECTION_NAME = settings.typesense_collection_name
|
|
|
|
# Function to get the database path from environment variable or default
|
|
def get_database_path():
|
|
return settings.database_path |