From 32c8774a0a9781eb4377662a674ad0a6a9161016 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nielson=20Jann=C3=A9?= <nielsonj@gmail.com>
Date: Wed, 12 Mar 2025 14:45:25 +0100
Subject: [PATCH] Makes backend enum pretty printable

---
 generic_rag/backend/models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/generic_rag/backend/models.py b/generic_rag/backend/models.py
index ebcf45c..e09e5f5 100644
--- a/generic_rag/backend/models.py
+++ b/generic_rag/backend/models.py
@@ -19,6 +19,10 @@ class BackendType(Enum):
     aws = "aws"
     local = "local"
 
+    # make the enum pretty printable for argparse
+    def __str__(self):
+        return self.value
+
 
 def get_chat_model(backend_type: BackendType) -> BaseChatModel:
     if backend_type == BackendType.azure: