From c06862ab76b399bf9dc586f776d5fa362f36ec51 Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Fri, 8 Jul 2022 13:38:26 +0200
Subject: [PATCH] Improve CLI

---
 great_ai/__main__.py | 5 +++++
 great_ai/context.py  | 4 +++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/great_ai/__main__.py b/great_ai/__main__.py
index f07f252..3eb7424 100644
--- a/great_ai/__main__.py
+++ b/great_ai/__main__.py
@@ -201,6 +201,11 @@ class GreatAIReload(BaseReload):
 
 
 def main() -> None:
+    import os
+    import sys
+
+    sys.path.append(os.getcwd())
+
     try:
         serve()
     except KeyboardInterrupt:
diff --git a/great_ai/context.py b/great_ai/context.py
index 56134ac..d7daf0d 100644
--- a/great_ai/context.py
+++ b/great_ai/context.py
@@ -6,6 +6,8 @@ from typing import Any, Dict, Optional, Type, Union, cast
 
 from pydantic import BaseModel
 
+from great_ai import __version__
+
 from .constants import (
     DEFAULT_LARGE_FILE_CONFIG_PATHS,
     DEFAULT_TRACING_DATABASE_CONFIG_PATHS,
@@ -109,7 +111,7 @@ def configure(
         route_config=route_config,
     )
 
-    logger.info("Settings: configured ✅")
+    logger.info(f"GreatAI (v{__version__}): configured ✅")
     for k, v in get_context().to_flat_dict().items():
         logger.info(f"{LIST_ITEM_PREFIX}{k}: {v}")