Просмотр исходного кода

Control Gemini thinking budget via Requesty (#4486)

Daniel Trugman 6 месяцев назад
Родитель
Сommit
e2516ebe02
1 измененных файлов с 8 добавлено и 2 удалено
  1. 8 2
      src/api/providers/fetchers/requesty.ts

+ 8 - 2
src/api/providers/fetchers/requesty.ts

@@ -19,8 +19,14 @@ export async function getRequestyModels(apiKey?: string): Promise<Record<string,
 		const rawModels = response.data.data
 
 		for (const rawModel of rawModels) {
-			const reasoningBudget = rawModel.supports_reasoning && rawModel.id.includes("claude")
-			const reasoningEffort = rawModel.supports_reasoning && rawModel.id.includes("openai")
+			const reasoningBudget =
+				rawModel.supports_reasoning &&
+				(rawModel.id.includes("claude") ||
+					rawModel.id.includes("coding/gemini-2.5") ||
+					rawModel.id.includes("vertex/gemini-2.5"))
+			const reasoningEffort =
+				rawModel.supports_reasoning &&
+				(rawModel.id.includes("openai") || rawModel.id.includes("google/gemini-2.5"))
 
 			const modelInfo: ModelInfo = {
 				maxTokens: rawModel.max_output_tokens,