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 01:25:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Show title for Google grounding sources

Browse Source 
Co-authored-by: gemini-3.7-flash <noreply@google.com>
This commit is contained in:
multipleof4
2026-08-27 11:57:55 -07:00
parent ce3b3290ea
commit a32e82fac3
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
providers.js
View File

@@ -111,11 +111,13 @@ function toGoogleSchema(s) {
function collectSources(candidate, sources) {
for (const c of candidate?.groundingMetadata?.groundingChunks || []) {
const uri = c?.web?.uri
if (uri) sources.add(uri)
if (uri && !sources.has(uri)) sources.set(uri, c?.web?.title || uri)
}
for (const u of candidate?.urlContextMetadata?.urlMetadata || []) {
const uri = u?.retrievedUrl
if (uri && (!u.urlRetrievalStatus || u.urlRetrievalStatus === 'URL_RETRIEVAL_STATUS_SUCCESS')) sources.add(uri)
if (uri && (!u.urlRetrievalStatus || u.urlRetrievalStatus === 'URL_RETRIEVAL_STATUS_SUCCESS') && !sources.has(uri)) {
sources.set(uri, u?.title || uri)
}
}
}
@@ -329,7 +331,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 Set()
const sources = new Map()
let hasReasoning = false, hasContent = false
const stream = await ai.models.generateContentStream({ model, contents, config })
@@ -358,7 +360,7 @@ export async function streamGoogle({ apiKey, body, signal, onDelta, isRunning })
}
if (sources.size && isRunning()) {
const list = [...sources].map((uri, i) => `${i + 1}. [${uri}](${uri})`).join('\n')
const list = [...sources.entries()].map(([uri, title], i) => `${i + 1}. [${title}](${uri})`).join('\n')
onDelta(`\n\n---\n\n**Sources**\n\n${list}\n`)
}
}