Do you have any integration with Amazon Bedrock Knowledgebase?

[pratip-bagchi]

Do you have any integration with Amazon Bedrock Knowledgebase? If yes, can you please provide some example, I was able to find RetrievalQA class, but struggling to understand how to use it. In Python I can do like

 retriever = AmazonKnowledgeBasesRetriever(
        credentials_profile_name=os.getenv("profile_name"),
        knowledge_base_id=kbId,
        retrieval_config={"vectorSearchConfiguration": {"numberOfResults": numberOfResults}},

then

 qa = RetrievalQA.from_chain_type(
        llm=llm, retriever=retriever, return_source_documents=True
    )

however, I could not able to figure out on how to use RetrievalQaChain class, from this repo.

[HavenDV]

Not at the moment, maybe it's a good idea for a new integration for @curlyfro

[curlyfro]

hi @pratip-bagchi,
i was going to add this but i haven't had the chance yet.

it's pretty simple:

        var request = new RetrieveAndGenerateRequest
            {
                Input = new RetrieveAndGenerateInput { Text = question },
                RetrieveAndGenerateConfiguration = new RetrieveAndGenerateConfiguration
                {
                    Type = RetrieveAndGenerateType.KNOWLEDGE_BASE,
                    KnowledgeBaseConfiguration = new KnowledgeBaseRetrieveAndGenerateConfiguration
                    {
                        KnowledgeBaseId = _knowledgeBase?.KnowledgeBaseId,
                        ModelArn = _textModelId,
                        RetrievalConfiguration = new KnowledgeBaseRetrievalConfiguration
                        {
                            VectorSearchConfiguration = new KnowledgeBaseVectorSearchConfiguration
                            {
                                OverrideSearchType = SelectedSearchType,
                                // Filter = new RetrievalFilter
                                // {
                                //     Equals = new FilterAttribute
                                //     {
                                //         Key = "year",
                                //         Value = new Document(2020)
                                //     }
                                // }
                            }
                        }
                    }
                }
            };
        var result = await _agentClient!.RetrieveAndGenerateAsync(request);

[pratip-bagchi]

Where is this class RetrieveAndGenerateRequest ? I did not find in codes or any documentations on this, can you please share ?

[curlyfro]

it's another nuget package. it's part of the Amazon.BedrockAgentRuntime

[curlyfro]

i created a pull request to this feature.
#301