From d99407127fb15a1d95a493cbee18c1e6fac05414 Mon Sep 17 00:00:00 2001
From: Sakurasan <1173092237@qq.com>
Date: Sat, 1 Jul 2023 21:11:39 +0800
Subject: [PATCH] fix bug

---
 router/router.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/router/router.go b/router/router.go
index 9368602..000b52d 100644
--- a/router/router.go
+++ b/router/router.go
@@ -773,7 +773,7 @@ func NumTokensFromStr(messages string, model string) (num_tokens int) {
 
 func modelmap(in string) string {
 	// gpt-3.5-turbo -> gpt-35-turbo
-	if strings.HasSuffix(in, ".") {
+	if strings.Contains(in, ".") {
 		return strings.ReplaceAll(in, ".", "")
 	}
 	return in