sunholo 0.95.0__py3-none-any.whl → 0.95.1__py3-none-any.whl

sunholo/vertex/memory_tools.py

@@ -153,42 +153,44 @@ def print_grounding_response(response):
     prev_index = 0
     markdown_text = ""
 
-    sources: dict[str, str] = {}
-    footnote = 1
-    for attribution in grounding_metadata.grounding_attributions:
-        context = attribution.web or attribution.retrieved_context
-        if not context:
-            log.info(f"Skipping Grounding Attribution {attribution}")
-            continue
-
-        title = context.title
-        uri = context.uri
-        end_index = int(attribution.segment.end_index)
+    for grounding_support in grounding_metadata.grounding_supports:
+        text_segment = text_bytes[
+            prev_index : grounding_support.segment.end_index
+        ].decode(ENCODING)
 
-        if uri not in sources:
-            sources[uri] = {"title": title, "footnote": footnote}
-            footnote += 1
+        footnotes_text = ""
+        for grounding_chunk_index in grounding_support.grounding_chunk_indices:
+            footnotes_text += f"[{grounding_chunk_index + 1}]"
 
-        text_segment = text_bytes[prev_index:end_index].decode(ENCODING)
-        markdown_text += f"{text_segment} [[{sources[uri]['footnote']}]]({uri})"
-        prev_index = end_index
+        markdown_text += f"{text_segment} {footnotes_text}\n"
+        prev_index = grounding_support.segment.end_index
 
     if prev_index < len(text_bytes):
         markdown_text += str(text_bytes[prev_index:], encoding=ENCODING)
 
-    markdown_text += "\n## Grounding Sources\n"
+    markdown_text += "\n----\n## Grounding Sources\n"
 
     if grounding_metadata.web_search_queries:
         markdown_text += (
             f"\n**Web Search Queries:** {grounding_metadata.web_search_queries}\n"
         )
+        if grounding_metadata.search_entry_point:
+            markdown_text += f"\n**Search Entry Point:**\n {grounding_metadata.search_entry_point.rendered_content}\n"
     elif grounding_metadata.retrieval_queries:
         markdown_text += (
             f"\n**Retrieval Queries:** {grounding_metadata.retrieval_queries}\n"
         )
 
-    for uri, source in sources.items():
-        markdown_text += f"{source['footnote']}. [{source['title']}]({uri})\n"
-    
-    log.info(markdown_text)
-    return markdown_text
+    markdown_text += "### Grounding Chunks\n"
+
+    for index, grounding_chunk in enumerate(
+        grounding_metadata.grounding_chunks, start=1
+    ):
+        context = grounding_chunk.web or grounding_chunk.retrieved_context
+        if not context:
+            print(f"Skipping Grounding Chunk {grounding_chunk}")
+            continue
+
+        markdown_text += f"{index}. [{context.title}]({context.uri})\n"
+
+    return markdown_text

{sunholo-0.95.0.dist-info → sunholo-0.95.1.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.95.0
+Version: 0.95.1
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.95.0.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.95.1.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.95.0.dist-info → sunholo-0.95.1.dist-info}/RECORD

@@ -141,12 +141,12 @@ sunholo/vertex/extensions_call.py,sha256=QeQbL3aAHlc4_-SynOzooZ_3xkQWAlcgNmFBSwL
 sunholo/vertex/extensions_class.py,sha256=2QGW28lNjoMEnaoVb3QcqEDwphclIsZthnpLUi5_Ivo,21033
 sunholo/vertex/genai_functions.py,sha256=2z6grM9H0Z79Yzx88l8mE1wXck3bRa0TWvnqZZ9ifDc,2051
 sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
-sunholo/vertex/memory_tools.py,sha256=q_phxgGX2TG2j2MXNULF2xGzQnQPENwjPN9nZ_A9Gh0,7526
+sunholo/vertex/memory_tools.py,sha256=x3_ESOhgMpf-gNpiOzmP7YQI0l0FAoLtbAVP2K2N0OA,7724
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.95.0.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.95.0.dist-info/METADATA,sha256=Ot1yshHTv-QqO_hNg1O7TElzvyDaopq3Ohij0_cyQJg,7889
-sunholo-0.95.0.dist-info/WHEEL,sha256=5Mi1sN9lKoFv_gxcPtisEVrJZihrm_beibeg5R6xb4I,91
-sunholo-0.95.0.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.95.0.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.95.0.dist-info/RECORD,,
+sunholo-0.95.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.95.1.dist-info/METADATA,sha256=eGcN3ki323hTYiSMUNGlgJCDQI_r1hxVLHbwksnSdXg,7889
+sunholo-0.95.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+sunholo-0.95.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.95.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.95.1.dist-info/RECORD,,

{sunholo-0.95.0.dist-info → sunholo-0.95.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.0.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any