From a00db1b474c3080e80733179af9a421a79758f4f Mon Sep 17 00:00:00 2001
From: drbh <david.richard.holtz@gmail.com>
Date: Fri, 28 Jun 2024 15:22:15 +0000
Subject: [PATCH] fix: adjust unwrap syntax in template

---
 router/src/infer/mod.rs | 16 ++--------------
 1 file changed, 2 insertions(+), 14 deletions(-)

diff --git a/router/src/infer/mod.rs b/router/src/infer/mod.rs
index 7a1022b8..49282eb9 100644
--- a/router/src/infer/mod.rs
+++ b/router/src/infer/mod.rs
@@ -293,20 +293,8 @@ impl ChatTemplate {
 
         Self {
             template,
-            bos_token: match bos_token {
-                Some(token) => match token {
-                    TokenizerConfigToken::String(token) => Some(token),
-                    TokenizerConfigToken::Object { content } => Some(content),
-                },
-                None => None,
-            },
-            eos_token: match eos_token {
-                Some(token) => match token {
-                    TokenizerConfigToken::String(token) => Some(token),
-                    TokenizerConfigToken::Object { content } => Some(content),
-                },
-                None => None,
-            },
+            bos_token: bos_token.map(|token| token.as_str().to_string()),
+            eos_token: eos_token.map(|token| token.as_str().to_string()),
             use_default_tool_template,
         }
     }