跳至內容

如何使用 RetrieveAndGenerate API 重複使用 Amazon Bedrock 中的知識庫工作階段?

1 分的閱讀內容
0

我想在 Amazon Bedrock 中多次使用知識庫工作階段。

解決方法

知識庫工作階段會持續 24 小時。您無法延長工作階段,也無法在工作階段過期後重新啟用工作階段。但是,在 24 時的期限內,您可以在下一次 API 呼叫的輸入中提供相同的工作階段 ID ,以重複使用該工作階段。如果您未包含工作階段 ID,那麼系統將自動建立一個新工作階段。

若要重複使用相同的知識庫工作階段,請呼叫 RetrieveAndGenerate API,並記下 API 呼叫回應中的 sessionId。

輸出範例:

{  
   "citations": [   
      {   
         "generatedResponsePart": {   
            "textResponsePart": {   
               "span": {   
                  "end": number,  
                  "start": number  
               },  
               "text": "string"  
            }  
         },  
         "retrievedReferences": [   
            {   
               "content": {   
                  "text": "string"  
               },  
               "location": {   
                  "s3Location": {   
                     "uri": "string"  
                  },  
                  "type": "string"  
               },  
               "metadata": {   
                  "string" : JSON value   
               }  
            }  
         ]  
      }  
   ],  
   "output": {   
      "text": "string"  
   },  
   "sessionId": "string"  
}

若要將指定的工作階段 ID 指派給目前的知識庫工作階段,請在 24 小時內呼叫 RetrieveAndGenerate API 來傳遞 sessionId。

**注意:**知識庫會記住您先前的對話並提供答案。

API 呼叫範例:

{
   "input": {
      "text": "string"
   },
   "retrieveAndGenerateConfiguration": {
      "knowledgeBaseConfiguration": {
         "knowledgeBaseId": "string",
         "modelArn": "string"
      },
      "type": "KNOWLEDGE_BASE"
   },
   "sessionConfiguration": {
      "kmsKeyArn": "string"
   },
   "sessionId": "string"
}