diff --git a/docs/package.json b/docs/package.json
index 75b9870..f24107d 100644
--- a/docs/package.json
+++ b/docs/package.json
@@ -1,6 +1,6 @@
 {
   "name": "ai-backend-docs",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "Documentation for the AI Backend project",
   "scripts": {
     "docs:dev": "vitepress dev src",
diff --git a/src/internal/models.py b/src/internal/models.py
index 4e305b0..563fc05 100644
--- a/src/internal/models.py
+++ b/src/internal/models.py
@@ -7,6 +7,8 @@ class TextModel(Enum):
     """
 
     LLAMA3 = "llama3"
+    LLAMA3_1 = "llama3.1"
+    GEMMA2 = "gemma2"
 
 
 class ImageModel(Enum):
diff --git a/src/main.py b/src/main.py
index c1da546..603adb0 100644
--- a/src/main.py
+++ b/src/main.py
@@ -6,7 +6,7 @@
 app = FastAPI(
     title="AI Backend",
     description="ai backend for your app powered by llama3",
-    version="0.2.1",
+    version="0.2.2",
 )
 
 origins = ["*"]