Pavlo Paliychuk 33908da18e
feat: add OpenAI configuration options to Settings and update LLM client setup (#126)
* chore: Add model and base url customization support for graphiti svc

* fix: formatter
2024-09-19 16:35:36 -04:00

26 lines
586 B
Python

from functools import lru_cache
from typing import Annotated
from fastapi import Depends
from pydantic import Field
from pydantic_settings import BaseSettings, SettingsConfigDict # type: ignore
class Settings(BaseSettings):
openai_api_key: str
openai_base_url: str | None = Field(None)
model_name: str | None = Field(None)
neo4j_uri: str
neo4j_user: str
neo4j_password: str
model_config = SettingsConfigDict(env_file='.env', extra='ignore')
@lru_cache
def get_settings():
return Settings()
ZepEnvDep = Annotated[Settings, Depends(get_settings)]