Add settings config
Browse files- graphrag/configs/settings.py +18 -0
graphrag/configs/settings.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Configuration settings for the GraphRAG Inference Hackathon."""
|
| 2 |
+
import os
|
| 3 |
+
from dataclasses import dataclass
|
| 4 |
+
|
| 5 |
+
@dataclass
|
| 6 |
+
class AppConfig:
|
| 7 |
+
"""Top-level application config."""
|
| 8 |
+
llm_model: str = os.getenv("LLM_MODEL", "gpt-4o-mini")
|
| 9 |
+
embed_model: str = os.getenv("EMBED_MODEL", "text-embedding-3-small")
|
| 10 |
+
openai_api_key: str = os.getenv("OPENAI_API_KEY", "")
|
| 11 |
+
tg_host: str = os.getenv("TG_HOST", "")
|
| 12 |
+
tg_graph: str = os.getenv("TG_GRAPH", "GraphRAG")
|
| 13 |
+
tg_username: str = os.getenv("TG_USERNAME", "tigergraph")
|
| 14 |
+
tg_password: str = os.getenv("TG_PASSWORD", "")
|
| 15 |
+
dashboard_port: int = 7860
|
| 16 |
+
debug: bool = os.getenv("DEBUG", "false").lower() == "true"
|
| 17 |
+
|
| 18 |
+
config = AppConfig()
|