Как заставить мой помощник по рэг -рэг -рэг -ай спрашивать, о каком продукте спрашивает пользователь ⇐ Python

[Anonymous]

Я добавил помощника Rag AI, использующего AWS Bedrock на наш веб -сайт, чтобы клиенты могли задавать вопросы и получить ответы о четырех продуктах, которые мы продаем. Каждый продукт имеет одинаковые два документа, такие как: Product_a_user_guide.pdf, product_a_specs.pdf, product_b_user_guide.pdf, product_b_specs.pdf и т. Д. Поэтому, задавая помощнику ИИ вопрос без указания продукта, он не всегда возвращает правильный результат. Например, если вы задаете вопрос, который относится ко всем четырем продуктам, например, «как мне запустить калибровочный тест», он может вернуть ответ из продукта A, B, C или D. Как я могу сделать моего помощника умнее, чтобы он попросил пользователя «сказать мне, какой продукт вы спрашиваете: A, B, C или D?» Если пользователь не указывает это в своем вопросе? < /p>
Это мой код: < /p>

Код: Выделить всё

bedrock = boto3.client("bedrock-agent-runtime", region_name="us-west-1")

def lambda_handler(event, context):

model_id = "amazon.titan-text-premier-v1:0"
question = event["queryStringParameters"]["question"]
session_id = event["queryStringParameters"]["session_id”] if “session_id" in event["queryStringParameters"] else None

kb_id = os.environ["KNOWLEDGE_BASE_ID"]
region = "us-west-1"
model_arn = f"arn:aws:bedrock:{region}::foundation-model/{model_id}"

# Query the knowledge base
response = queryKB(question, kb_id, model_arn, session_id)

# Extract the generated text and session ID from the response
generated_text = response["output"]["text"].strip()
session_id = response.get("sessionId", "")
citations = response["citations"]

headers = {
"Access-Control-Allow-Origin": "*",
"Access-Control-Allow-Credentials": True
}

# Return the response in the expected format
return {
“statusCode": 200,
"headers": headers,
"body": json.dumps({
"question": question,
"answer": generated_text,
"citations": citations,
"sessionId": session_id
},  ensure_ascii=False)
}

def queryKB(question, kbId, model_arn, sessionId=None):
return bedrock.retrieve_and_generate(
input={
"text": question
},
retrieveAndGenerateConfiguration={
“type": "KNOWLEDGE_BASE",
"knowledgeBaseConfiguration": {
"knowledgeBaseId": kbId,
"modelArn": model_arn
}
},
sessionId=sessionId if sessionId else None
)

Спасибо за вашу помощь.

Подробнее здесь: https://stackoverflow.com/questions/796 ... -is-asking