Issue #, if available:
ask_bedrock_llm_with_knowledge_base function response was just the generated_text, but the citations cell below relies on the whole response
Description of changes:
Modified the response of the ask_bedrock_llm_with_knowledge_base function to return the whole response, and modified the cells below to parse out the right context.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.
Issue #, if available: ask_bedrock_llm_with_knowledge_base function response was just the generated_text, but the citations cell below relies on the whole response
Description of changes: Modified the response of the ask_bedrock_llm_with_knowledge_base function to return the whole response, and modified the cells below to parse out the right context.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.