sune-org/us.proxy.sune.chat
1
0
Fork 0
mirror of https://github.com/sune-org/us.proxy.sune.chat.git synced 2026-08-28 09:35:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Enforce 64k tokens & full URL source citations

Browse Source 
Co-authored-by: gemini-3.7-flash <noreply@google.com>
This commit is contained in:
multipleof4
2026-08-20 10:26:16 -07:00
parent 0562254723
commit 32f8c661e4
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
providers.js
View File

@@ -97,7 +97,6 @@ function mapToGoogleContents(messages) {
if (last?.role === role) last.parts.push(...parts)
else contents.push({ role, parts })
}
// Drop empty/placeholder trailing model turns (no prefill support needed here)
while (contents.length && contents.at(-1).role === 'model' && isBlankTurn(contents.at(-1))) contents.pop()
return contents
}
@@ -112,13 +111,11 @@ function toGoogleSchema(s) {
function collectSources(candidate, sources) {
for (const c of candidate?.groundingMetadata?.groundingChunks || []) {
const uri = c?.web?.uri
if (uri && !sources.has(uri)) sources.set(uri, c.web.title || new URL(uri).hostname)
if (uri) sources.add(uri)
}
for (const u of candidate?.urlContextMetadata?.urlMetadata || []) {
const uri = u?.retrievedUrl
if (!uri || sources.has(uri)) continue
if (u.urlRetrievalStatus && u.urlRetrievalStatus !== 'URL_RETRIEVAL_STATUS_SUCCESS') continue
try { sources.set(uri, new URL(uri).hostname) } catch { sources.set(uri, uri) }
if (uri && (!u.urlRetrievalStatus || u.urlRetrievalStatus === 'URL_RETRIEVAL_STATUS_SUCCESS')) sources.add(uri)
}
}
@@ -288,10 +285,12 @@ export async function streamGoogle({ apiKey, body, signal, onDelta, isRunning })
const online = raw.endsWith(':online')
const model = (online ? raw.slice(0, -7) : raw).replace(/^models\//, '')
const config = { abortSignal: signal }
const config = {
abortSignal: signal,
maxOutputTokens: Number.isFinite(+body.max_tokens) && +body.max_tokens > 0 ? +body.max_tokens : 65536,
}
if (Number.isFinite(+body.temperature)) config.temperature = +body.temperature
if (Number.isFinite(+body.top_p)) config.topP = +body.top_p
if (Number.isFinite(+body.max_tokens) && +body.max_tokens > 0) config.maxOutputTokens = +body.max_tokens
const systemInstruction = body.messages.filter(m => m.role === 'system').map(extractText).filter(Boolean).join('\n\n')
if (systemInstruction) config.systemInstruction = systemInstruction
@@ -321,7 +320,7 @@ export async function streamGoogle({ apiKey, body, signal, onDelta, isRunning })
const contents = mapToGoogleContents(body.messages)
if (!contents.length) throw new Error('Google API error: no usable content')
const sources = new Map()
const sources = new Set()
let hasReasoning = false, hasContent = false
const stream = await ai.models.generateContentStream({ model, contents, config })
@@ -350,7 +349,7 @@ export async function streamGoogle({ apiKey, body, signal, onDelta, isRunning })
}
if (sources.size && isRunning()) {
const list = [...sources].map(([uri, title], i) => `${i + 1}. [${title}](${uri})`).join('\n')
const list = [...sources].map((uri, i) => `${i + 1}. [${uri}](${uri})`).join('\n')
onDelta(`\n\n---\n\n**Sources**\n\n${list}\n`)
}
}