From b985d3a6ec18c4783ce248506ebb733f3dad4fd6 Mon Sep 17 00:00:00 2001
From: Ata Fatahi <immrata@gmail.com>
Date: Wed, 11 Dec 2024 15:37:58 -0800
Subject: [PATCH] Include version info into the router package

Signed-off-by: Ata Fatahi <immrata@gmail.com>
---
 rust/py_src/sglang_router/__init__.py | 6 ++++++
 rust/py_src/sglang_router/version.py  | 1 +
 2 files changed, 7 insertions(+)
 create mode 100644 rust/py_src/sglang_router/version.py

diff --git a/rust/py_src/sglang_router/__init__.py b/rust/py_src/sglang_router/__init__.py
index ec41b5d0c74..285ee173ba9 100644
--- a/rust/py_src/sglang_router/__init__.py
+++ b/rust/py_src/sglang_router/__init__.py
@@ -3,3 +3,9 @@
 
 # no wrapper on policy type => direct export
 from .router import Router
+
+__all__ = ["Router", "PolicyType"]
+
+from sglang_router.version import __version__
+
+__all__ += ["__version__"]
diff --git a/rust/py_src/sglang_router/version.py b/rust/py_src/sglang_router/version.py
new file mode 100644
index 00000000000..3dc1f76bc69
--- /dev/null
+++ b/rust/py_src/sglang_router/version.py
@@ -0,0 +1 @@
+__version__ = "0.1.0"