Skip to content

Commit 0709d38

Browse files
refactor: http transport gemini integration refactors
1 parent d68ae37 commit 0709d38

File tree

3 files changed

+355
-43
lines changed

3 files changed

+355
-43
lines changed

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

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

5050
modelStr := model.(string)
51-
// Remove :generateContent suffix if present
52-
modelStr = strings.TrimSuffix(modelStr, ":generateContent")
51+
// Remove Google GenAI API endpoint suffixes if present
52+
for _, sfx := range []string{
53+
":streamGenerateContent",
54+
":generateContent",
55+
":countTokens",
56+
} {
57+
modelStr = strings.TrimSuffix(modelStr, sfx)
58+
}
59+
5360
// Remove trailing colon if present
5461
if len(modelStr) > 0 && modelStr[len(modelStr)-1] == ':' {
5562
modelStr = modelStr[:len(modelStr)-1]

0 commit comments

Comments
 (0)