[Docs Agent] Bug fix - Fix the "index out of range" errors in Docs

Agent's chat web app.
google · May 12, 2024 · 0b491c3 · 0b491c3
1 parent 8824ddc
commit 0b491c3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/examples/gemini/python/docs-agent/docs_agent/agents/docs_agent.py b/examples/gemini/python/docs-agent/docs_agent/agents/docs_agent.py
@@ -439,7 +439,11 @@ def query_vector_store_to_build(
         plain_token = 0
         sources = []
         final_pages = []
-        for i in range(max_sources):
+        # Quick fix: Ensure max_sources is not larger than the array size of search_result.
+        this_range = len(search_result)
+        if this_range > max_sources:
+            this_range = max_sources
+        for i in range(this_range):
             # The current section that is being built
             # eval turns str representation of array into an array
             curr_section_id = search_result[i].section.name_id

diff --git a/examples/gemini/python/docs-agent/docs_agent/interfaces/chatbot/chatui.py b/examples/gemini/python/docs-agent/docs_agent/interfaces/chatbot/chatui.py
@@ -472,7 +472,9 @@ def ask_model(question, agent, template: str = "chatui/index.html"):
         # Log debug information.
 
         if docs_agent.config.enable_logs_for_debugging == "True":
-            top_source_url = search_result[0].section.url
+            top_source_url = ""
+            if len(search_result) > 0:
+                top_source_url = search_result[0].section.url
             source_urls = ""
             index = 1
             for result in search_result: