chore: reorganize constant related package

relay/relaymode/define.go

@@ -0,0 +1,14 @@
+package relaymode
+
+const (
+	Unknown = iota
+	ChatCompletions
+	Completions
+	Embeddings
+	Moderations
+	ImagesGenerations
+	Edits
+	AudioSpeech
+	AudioTranscription
+	AudioTranslation
+)

relay/relaymode/helper.go

@@ -0,0 +1,29 @@
+package relaymode
+
+import "strings"
+
+func GetByPath(path string) int {
+	relayMode := Unknown
+	if strings.HasPrefix(path, "/v1/chat/completions") {
+		relayMode = ChatCompletions
+	} else if strings.HasPrefix(path, "/v1/completions") {
+		relayMode = Completions
+	} else if strings.HasPrefix(path, "/v1/embeddings") {
+		relayMode = Embeddings
+	} else if strings.HasSuffix(path, "embeddings") {
+		relayMode = Embeddings
+	} else if strings.HasPrefix(path, "/v1/moderations") {
+		relayMode = Moderations
+	} else if strings.HasPrefix(path, "/v1/images/generations") {
+		relayMode = ImagesGenerations
+	} else if strings.HasPrefix(path, "/v1/edits") {
+		relayMode = Edits
+	} else if strings.HasPrefix(path, "/v1/audio/speech") {
+		relayMode = AudioSpeech
+	} else if strings.HasPrefix(path, "/v1/audio/transcriptions") {
+		relayMode = AudioTranscription
+	} else if strings.HasPrefix(path, "/v1/audio/translations") {
+		relayMode = AudioTranslation
+	}
+	return relayMode
+}