|
4 | 4 | author_url: https://github.com/owndev/ |
5 | 5 | project_url: https://github.com/owndev/Open-WebUI-Functions |
6 | 6 | funding_url: https://github.com/sponsors/owndev |
7 | | -version: 1.9.0 |
| 7 | +version: 1.9.1 |
8 | 8 | required_open_webui_version: 0.6.26 |
9 | 9 | license: Apache License 2.0 |
10 | 10 | description: Highly optimized Google Gemini pipeline with advanced image generation capabilities, intelligent compression, and streamlined processing workflows. |
@@ -1786,8 +1786,6 @@ async def _process_grounding_metadata( |
1786 | 1786 | grounding_metadata_list: List[types.GroundingMetadata], |
1787 | 1787 | text: str, |
1788 | 1788 | __event_emitter__: Callable, |
1789 | | - *, |
1790 | | - emit_replace: bool = True, |
1791 | 1789 | ): |
1792 | 1790 | """Process and emit grounding metadata events.""" |
1793 | 1791 | grounding_chunks = [] |
@@ -1855,17 +1853,8 @@ async def _process_grounding_metadata( |
1855 | 1853 | cited_chunks.append(text_bytes[last_byte_index:].decode(ENCODING)) |
1856 | 1854 |
|
1857 | 1855 | replaced_text = "".join(cited_chunks) |
1858 | | - if emit_replace: |
1859 | | - await __event_emitter__( |
1860 | | - { |
1861 | | - "type": "replace", |
1862 | | - "data": {"content": replaced_text}, |
1863 | | - } |
1864 | | - ) |
1865 | 1856 |
|
1866 | | - # Return the transformed text when requested by caller |
1867 | | - if not emit_replace: |
1868 | | - return replaced_text if replaced_text is not None else text |
| 1857 | + return replaced_text if replaced_text is not None else text |
1869 | 1858 |
|
1870 | 1859 | async def _handle_streaming_response( |
1871 | 1860 | self, |
@@ -2003,12 +1992,10 @@ async def emit_chat_event(event_type: str, data: Dict[str, Any]) -> None: |
2003 | 1992 | # After processing all chunks, handle grounding data |
2004 | 1993 | final_answer_text = "".join(answer_chunks) |
2005 | 1994 | if grounding_metadata_list and __event_emitter__: |
2006 | | - # Don't emit replace here; we'll compose final content below |
2007 | 1995 | cited = await self._process_grounding_metadata( |
2008 | 1996 | grounding_metadata_list, |
2009 | 1997 | final_answer_text, |
2010 | 1998 | __event_emitter__, |
2011 | | - emit_replace=False, |
2012 | 1999 | ) |
2013 | 2000 | final_answer_text = cited or final_answer_text |
2014 | 2001 |
|
@@ -2386,7 +2373,6 @@ async def get_response(): |
2386 | 2373 | grounding_metadata_list, |
2387 | 2374 | final_answer, |
2388 | 2375 | __event_emitter__, |
2389 | | - emit_replace=False, |
2390 | 2376 | ) |
2391 | 2377 | final_answer = cited or final_answer |
2392 | 2378 |
|
|
0 commit comments