File size: 638 Bytes
4c945e0
 
 
 
 
 
 
 
 
 
 
 
 
 
46026e0
 
 
982c368
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from pydantic_settings import BaseSettings, SettingsConfigDict

class Settings(BaseSettings):
    LLM_PROVIDER: str = "openai"
    BLABLADOR_API_KEY: str = ""
    HELMHOLTZ_BLABLADOR_ENDPOINT: str = "https://api.helmholtz-blablador.fz-juelich.de/v1"
    MODEL_ALIAS_LARGE: str = "alias-large"
    MODEL_ALIAS_HUGE: str = "alias-huge"
    MAX_CONCURRENCY: int = 5
    PERSONA_POOL_REPO: str = "https://github.com/JsonLord/agent-notes.git"

    model_config = SettingsConfigDict(env_file=".env")

settings = Settings()

import os
if settings.BLABLADOR_API_KEY:
    os.environ['OPENAI_API_KEY'] = settings.BLABLADOR_API_KEY or 'dummy_token'