add name and revision

tabris2015 · Nov 29, 2023 · 7462ac2 · 7462ac2
1 parent 075f789
commit 7462ac2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/config.py b/src/config.py
@@ -9,6 +9,8 @@ class GPTModel(str, Enum):
 
 
 class Settings(BaseSettings):
+    service_name: str = "Awesome Projects"
+    k_revision: str = "Local"
     log_level: str = "DEBUG"
     openai_key: str
     model: GPTModel = GPTModel.gpt_3_5_turbo

diff --git a/src/main.py b/src/main.py
@@ -3,9 +3,14 @@
 from src.llm_service import TemplateLLM
 from src.prompts import ProjectParams
 from src.parsers import ProjectIdeas
+from src.config import get_settings
 
+_SETTINGS = get_settings()
 
-app = FastAPI()
+app = FastAPI(
+    title=_SETTINGS.service_name,
+    version=_SETTINGS.k_revision
+)
 
 app.add_middleware(
     CORSMiddleware,