Skip to content

Commit 371e003

Browse files
refactor: http transport gemini integration refactors
1 parent 72f5791 commit 371e003

File tree

3 files changed

+429
-42
lines changed

3 files changed

+429
-42
lines changed

transports/bifrost-http/integrations/genai/router.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,10 @@ func extractAndSetModelFromURL(ctx *fasthttp.RequestCtx, req interface{}) error
4848
}
4949

5050
modelStr := model.(string)
51-
// Remove :generateContent suffix if present
51+
// Remove Google GenAI API endpoint suffixes if present
5252
modelStr = strings.TrimSuffix(modelStr, ":generateContent")
53+
modelStr = strings.TrimSuffix(modelStr, ":streamGenerateContent")
54+
modelStr = strings.TrimSuffix(modelStr, ":countTokens")
5355
// Remove trailing colon if present
5456
if len(modelStr) > 0 && modelStr[len(modelStr)-1] == ':' {
5557
modelStr = modelStr[:len(modelStr)-1]

0 commit comments

Comments
 (0)