Skip to content

Commit 95464c3

Browse files
gausekhaDavidSoutherbeqqrry-awsrlhagerm
authored
Added KB examples and tests and updated metadata file (#7451)
--------- Co-authored-by: David Souther <davidsouther+github@gmail.com> Co-authored-by: Jason Q <81179619+beqqrry-aws@users.noreply.github.com> Co-authored-by: Rachel Hagerman <110480692+rlhagerm@users.noreply.github.com>
1 parent b791808 commit 95464c3

File tree

8 files changed

+1320
-1
lines changed

8 files changed

+1320
-1
lines changed

.doc_gen/metadata/bedrock-agent_metadata.yaml

Lines changed: 65 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -521,3 +521,68 @@ bedrock-agent_GettingStartedWithBedrockPrompts:
521521
services:
522522
bedrock-agent: {CreatePrompt, CreatePromptVersion, DeletePrompt}
523523
bedrock-runtime: {Converse}
524+
525+
bedrock-agent_CreateKnowledgeBase:
526+
languages:
527+
Python:
528+
versions:
529+
- sdk_version: 3
530+
github: python/example_code/bedrock-agent
531+
excerpts:
532+
- description: Create an Amazon Bedrock knowledge base.
533+
snippet_tags:
534+
- python.example_code.bedrock-agent.create_knowledge_base
535+
services:
536+
bedrock-agent: {CreateKnowledgeBase}
537+
538+
bedrock-agent_GetKnowledgeBase:
539+
languages:
540+
Python:
541+
versions:
542+
- sdk_version: 3
543+
github: python/example_code/bedrock-agent
544+
excerpts:
545+
- description: Get an Amazon Bedrock knowledge base.
546+
snippet_tags:
547+
- python.example_code.bedrock-agent.get_knowledge_base
548+
services:
549+
bedrock-agent: {GetKnowledgeBase}
550+
551+
bedrock-agent_UpdateKnowledgeBase:
552+
languages:
553+
Python:
554+
versions:
555+
- sdk_version: 3
556+
github: python/example_code/bedrock-agent
557+
excerpts:
558+
- description: Update an Amazon Bedrock knowledge base.
559+
snippet_tags:
560+
- python.example_code.bedrock-agent.update_knowledge_base
561+
services:
562+
bedrock-agent: {UpdateKnowledgeBase}
563+
564+
bedrock-agent_DeleteKnowledgeBase:
565+
languages:
566+
Python:
567+
versions:
568+
- sdk_version: 3
569+
github: python/example_code/bedrock-agent
570+
excerpts:
571+
- description: Delete an Amazon Bedrock knowledge base.
572+
snippet_tags:
573+
- python.example_code.bedrock-agent.delete_knowledge_base
574+
services:
575+
bedrock-agent: {DeleteKnowledgeBase}
576+
577+
bedrock-agent_ListKnowledgeBases:
578+
languages:
579+
Python:
580+
versions:
581+
- sdk_version: 3
582+
github: python/example_code/bedrock-agent
583+
excerpts:
584+
- description: List Amazon Bedrock knowledge Bases.
585+
snippet_tags:
586+
- python.example_code.bedrock-agent.list_knowledge_bases
587+
services:
588+
bedrock-agent: {ListKnowledgeBases}

python/example_code/bedrock-agent/README.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,29 +44,34 @@ Code excerpts that show you how to call individual service functions.
4444
- [CreateFlow](flows/flow.py#L18)
4545
- [CreateFlowAlias](flows/flow_alias.py#L15)
4646
- [CreateFlowVersion](flows/flow_version.py#L18)
47+
- [CreateKnowledgeBase](knowledge_bases/knowledge_base.py#L30)
4748
- [CreatePrompt](prompts/prompt.py#L18)
4849
- [CreatePromptVersion](prompts/prompt.py#L84)
4950
- [DeleteAgent](bedrock_agent_wrapper.py#L118)
5051
- [DeleteAgentAlias](bedrock_agent_wrapper.py#L139)
5152
- [DeleteFlow](flows/flow.py#L155)
5253
- [DeleteFlowAlias](flows/flow_alias.py#L98)
5354
- [DeleteFlowVersion](flows/flow_version.py#L91)
55+
- [DeleteKnowledgeBase](knowledge_bases/knowledge_base.py#L169)
5456
- [DeletePrompt](prompts/prompt.py#L159)
5557
- [GetAgent](bedrock_agent_wrapper.py#L161)
5658
- [GetFlow](flows/flow.py#L192)
5759
- [GetFlowVersion](flows/flow_version.py#L54)
60+
- [GetKnowledgeBase](knowledge_bases/knowledge_base.py#L90)
5861
- [GetPrompt](prompts/prompt.py#L124)
5962
- [ListAgentActionGroups](bedrock_agent_wrapper.py#L208)
6063
- [ListAgentKnowledgeBases](bedrock_agent_wrapper.py#L237)
6164
- [ListAgents](bedrock_agent_wrapper.py#L185)
6265
- [ListFlowAliases](flows/flow_alias.py#L132)
6366
- [ListFlowVersions](flows/flow_version.py#L128)
6467
- [ListFlows](flows/flow.py#L229)
68+
- [ListKnowledgeBases](knowledge_bases/knowledge_base.py#L199)
6569
- [ListPrompts](prompts/prompt.py#L191)
6670
- [PrepareAgent](bedrock_agent_wrapper.py#L266)
6771
- [PrepareFlow](flows/flow.py#L58)
6872
- [UpdateFlow](flows/flow.py#L112)
6973
- [UpdateFlowAlias](flows/flow_alias.py#L55)
74+
- [UpdateKnowledgeBase](knowledge_bases/knowledge_base.py#L120)
7075

7176
### Scenarios
7277

@@ -102,6 +107,11 @@ This example shows you how to do the following:
102107
- Delete all created resources.
103108

104109
<!--custom.scenario_prereqs.bedrock-agent_GettingStartedWithBedrockFlows.start-->
110+
111+
The flow includes a prompt node that generates a playlist for a chosen genre
112+
and number of songs. The example creates the nodes and permissions
113+
for the flow.
114+
105115
<!--custom.scenario_prereqs.bedrock-agent_GettingStartedWithBedrockFlows.end-->
106116

107117
Start the example by running the following at a command prompt:

0 commit comments

Comments
 (0)