From fb39b6b30e7f7f69c55770059f71d65eaa0f5d2f Mon Sep 17 00:00:00 2001
From: "1808837298@qq.com" <1808837298@qq.com>
Date: Tue, 5 Mar 2024 23:04:57 +0800
Subject: [PATCH] fix: Fix the issue of 'unknown relay mode' when making an
 embedding request (close #93)

---
 relay/relay-text.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/relay/relay-text.go b/relay/relay-text.go
index c0d089a9..d38afaa8 100644
--- a/relay/relay-text.go
+++ b/relay/relay-text.go
@@ -169,6 +169,8 @@ func getPromptTokens(textRequest *dto.GeneralOpenAIRequest, info *relaycommon.Re
 		promptTokens, err = service.CountTokenInput(textRequest.Prompt, textRequest.Model), nil
 	case relayconstant.RelayModeModerations:
 		promptTokens, err = service.CountTokenInput(textRequest.Input, textRequest.Model), nil
+	case relayconstant.RelayModeEmbeddings:
+		promptTokens, err = service.CountTokenInput(textRequest.Input, textRequest.Model), nil
 	default:
 		err = errors.New("unknown relay mode")
 		promptTokens = 0