muthuk1 commited on
Commit
29c2dcc
·
verified ·
1 Parent(s): 53f589b

Add settings config

Browse files
Files changed (1) hide show
  1. 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()