Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-25 03:10:08 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix(GeminiProvider): relocate grounding metadata handling (#5490)

Browse Source 
* fix(GeminiProvider): relocate grounding metadata handling to improve code clarity and maintainability

* refactor(GeminiProvider): enhance conditional checks for chunk processing
This commit is contained in:
SuYao 2025-04-29 17:36:35 +08:00 committed by GitHub
parent 44ee736a10
commit 3f1045b936
1 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/renderer/src/providers/AiProvider/GeminiProvider.ts
View File

@ -447,26 +447,27 @@ export default class GeminiProvider extends BaseProvider {
final_time_completion_millsec = new Date().getTime() - start_time_millsec
}
// 3. Grounding/Search Metadata
const groundingMetadata = chunk.candidates?.[0]?.groundingMetadata
if (groundingMetadata) {
onChunk({
type: ChunkType.LLM_WEB_SEARCH_COMPLETE,
llm_web_search: {
results: groundingMetadata,
source: WebSearchSource.GEMINI
}
} as LLMWebSearchCompleteChunk)
}
// 4. Image Generation
const generateImage = this.processGeminiImageResponse(chunk)
if (generateImage?.images?.length) {
onChunk({ type: ChunkType.IMAGE_COMPLETE, image: generateImage })
}
if (chunk.candidates?.[0]?.finishReason && chunk.text) {
onChunk({ type: ChunkType.TEXT_COMPLETE, text: content })
if (chunk.candidates?.[0]?.finishReason) {
if (chunk.text) {
onChunk({ type: ChunkType.TEXT_COMPLETE, text: content })
}
if (chunk.candidates?.[0]?.groundingMetadata) {
// 3. Grounding/Search Metadata
const groundingMetadata = chunk.candidates?.[0]?.groundingMetadata
onChunk({
type: ChunkType.LLM_WEB_SEARCH_COMPLETE,
llm_web_search: {
results: groundingMetadata,
source: WebSearchSource.GEMINI
}
} as LLMWebSearchCompleteChunk)
}
onChunk({
type: ChunkType.BLOCK_COMPLETE,
response: {