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: 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) { function collectSources(candidate, sources) {
for (const c of candidate?.groundingMetadata?.groundingChunks || []) { for (const c of candidate?.groundingMetadata?.groundingChunks || []) {
const uri = c?.web?.uri 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 || []) { for (const u of candidate?.urlContextMetadata?.urlMetadata || []) {
const uri = u?.retrievedUrl 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) const contents = mapToGoogleContents(body.messages)
if (!contents.length) throw new Error('Google API error: no usable content') 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 let hasReasoning = false, hasContent = false
const stream = await ai.models.generateContentStream({ model, contents, config }) 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()) { 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`) onDelta(`\n\n---\n\n**Sources**\n\n${list}\n`)
} }
} }