From 8d613169482fbc9d7cb396c8a83fe4b5bb83a9c5 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil.hafeez@gmail.com>
Date: Thu, 25 Dec 2025 21:33:31 -0800
Subject: [PATCH] fix tests

---
 cli/planoai/__init__.py | 3 +++
 cli/planoai/main.py     | 8 +++++++-
 cli/pyproject.toml      | 3 +++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/cli/planoai/__init__.py b/cli/planoai/__init__.py
index e69de29b..07ae31b5 100644
--- a/cli/planoai/__init__.py
+++ b/cli/planoai/__init__.py
@@ -0,0 +1,3 @@
+"""Plano CLI - Intelligent Prompt Gateway."""
+
+__version__ = "0.4.0"
diff --git a/cli/planoai/main.py b/cli/planoai/main.py
index 4d69b7f7..3c69242f 100644
--- a/cli/planoai/main.py
+++ b/cli/planoai/main.py
@@ -50,10 +50,16 @@ ARCHGW_DOCKERFILE = "./Dockerfile"
 
 def get_version():
     try:
+        # First try to get version from package metadata (for installed packages)
         version = importlib.metadata.version("planoai")
         return version
     except importlib.metadata.PackageNotFoundError:
-        return "version not found"
+        # Fallback to version defined in __init__.py (for development)
+        try:
+            from planoai import __version__
+            return __version__
+        except ImportError:
+            return "version not found"
 
 
 @click.group(invoke_without_command=True)
diff --git a/cli/pyproject.toml b/cli/pyproject.toml
index a16ce2af..882453c6 100644
--- a/cli/pyproject.toml
+++ b/cli/pyproject.toml
@@ -25,6 +25,9 @@ planoai = "planoai.main:main"
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.hatch.version]
+path = "planoai/__init__.py"
+
 [tool.hatch.build.targets.wheel]
 packages = ["planoai"]