/
blanchefort
/
embedder_service
Обзор
Документация
Войти
/
blanchefort
/
embedder_service
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
src/settings.py
34 строки
1 KB
Igor Shatalin
fix
30 июл 2026, 01:10
30 июл 2026, 01:10
2ca924a
Код
Авторство
О чём код?
from functools import lru_cache from pathlib import Path from typing import Literal from pydantic import Field, SecretStr from pydantic_settings import BaseSettings, SettingsConfigDict class Settings(BaseSettings): model_config = SettingsConfigDict(env_file=".env", case_sensitive=False, extra="ignore") base_path: Path = Path(__file__).parent.parent hf_token: SecretStr | None = None secret_auth_token: SecretStr text_embedder_model_id: str = "deepvk/USER2-base" text_embedder_max_batch_size: int = Field(default=20, ge=1, le=100) reranker_model_id: str = "Alibaba-NLP/gte-multilingual-reranker-base" reranker_device: Literal["cuda", "cpu"] = "cpu" reranker_max_length: int = Field(default=4096, ge=512, le=8192) reranker_max_query_tokens: int = Field(default=512, ge=1, le=8192) reranker_batch_size: int = Field(default=16, ge=1, le=64) reranker_max_candidates: int = Field(default=100, ge=1, le=100) reranker_max_concurrency: int = Field(default=1, ge=1) reranker_queue_timeout_seconds: float = Field(default=2.0, gt=0) reranker_inference_timeout_seconds: float = Field(default=30.0, gt=0) reranker_warmup: bool = True reranker_local_files_only: bool = False reranker_max_body_bytes: int = Field(default=2_000_000, ge=1024) @lru_cache def get_settings() -> Settings: return Settings() # type: ignore[call-arg] # populated by pydantic-settings environment