Modelos Amazon Titan Text - Amazon Bedrock

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Modelos Amazon Titan Text

Os modelos Amazon Titan Text são compatíveis com os parâmetros de inferência a seguir.

Para obter mais informações sobre as diretrizes de engenharia de prompts do Titan Text, consulte o documento Titan Text Prompt Engineering Guidelines.

Para obter mais informações sobre modelos do Titan, consulte Visão geral dos modelos do Amazon Titan.

Solicitação e reposta

O corpo da solicitação é passado no body campo de uma InvokeModelWithResponseStreamsolicitação InvokeModelor.

Request
{ "inputText": string, "textGenerationConfig": { "temperature": float, "topP": float, "maxTokenCount": int, "stopSequences": [string] } }

Os seguintes parâmetros são obrigatórios:

  • inputText: o prompt a ser fornecido ao modelo para gerar uma resposta. Para gerar respostas em um estilo conversacional, envie a solicitação usando o seguinte formato:

    "inputText": "User: <theUserPrompt>\nBot:"

    Esse formato indica ao modelo que ele deve responder em uma nova linha após o usuário fornecer uma solicitação.

A textGenerationConfig é opcional. É possível usá-la para configurar os seguintes parâmetros de inferência:

  • temperature: use um valor menor para reduzir a aleatoriedade nas respostas.

    Padrão Mínimo Máximo
    0.7 0.0 1,0
  • topP: use um valor menor para ignorar opções menos prováveis e reduzir a diversidade de respostas.

    Padrão Mínimo Máximo
    0.9 0.0 1,0
  • maxTokenCount— Especifique o número máximo de tokens a serem gerados na resposta. Os limites máximos de tokens são rigorosamente aplicados.

    Modelo Padrão Mínimo Máximo
    Titan Text Lite 512 0 4.096
    Titan Text Express 512 0 8,192
    Titan Text Premier 512 0 3.072
  • stopSequences: especifique uma sequência de caracteres para indicar onde o modelo deve parar.

InvokeModel Response
{ "inputTextTokenCount": int, "results": [{ "tokenCount": int, "outputText": "\n<response>\n", "completionReason": "string" }] }

O corpo da resposta contém os seguintes campos:

  • inputTextTokenContagem — O número de tokens no prompt.

  • results: uma matriz de um item, um objeto que contém os seguintes campos:

    • tokenCount: o número de tokens na resposta.

    • outputText: o texto na resposta.

    • completionReason: o motivo pelo qual a resposta terminou de ser gerada. Os seguintes motivos são possíveis:

      • FINISHED: a resposta foi totalmente gerada.

      • LENGTH: a resposta foi truncada devido ao tamanho da resposta que você definiu.

      • STOP_CRITERIA_MET: a resposta foi truncada porque o critério de parada foi atingido.

      • RAG_QUERY_WHEN_RAG_DISABLED: o recurso está desabilitado e não pode concluir a consulta.

      • CONTENT_FILTERED: o conteúdo foi filtrado ou removido pelo filtro de conteúdo aplicado.

InvokeModelWithResponseStream Response

Cada fragmento de texto no corpo do fluxo da resposta tem o formato a seguir. Você deve decodificar o campo bytes (consulte Envie uma única solicitação com InvokeModel para obter um exemplo).

{ "chunk": { "bytes": b'{ "index": int, "inputTextTokenCount": int, "totalOutputTextTokenCount": int, "outputText": "<response-chunk>", "completionReason": "string" }' } }
  • index: o índice do fragmento na resposta de streaming.

  • inputTextTokenContagem — O número de tokens no prompt.

  • totalOutputTextTokenCount: o número de tokens na resposta.

  • outputText: o texto na resposta.

  • completionReason: o motivo pelo qual a resposta terminou de ser gerada. Os seguintes motivos são possíveis.

    • FINISHED: a resposta foi totalmente gerada.

    • LENGTH: a resposta foi truncada devido ao tamanho da resposta que você definiu.

    • STOP_CRITERIA_MET: a resposta foi truncada porque o critério de parada foi atingido.

    • RAG_QUERY_WHEN_RAG_DISABLED: o recurso está desabilitado e não pode concluir a consulta.

    • CONTENT_FILTERED: o conteúdo foi filtrado ou removido pelo filtro aplicado.

Exemplos de código

O exemplo a seguir mostra como executar inferência com o modelo Amazon Titan Text Premier usando o SDK para Python.

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. # SPDX-License-Identifier: Apache-2.0 """ Shows how to create a list of action items from a meeting transcript with the Amazon Titan Text model (on demand). """ import json import logging import boto3 from botocore.exceptions import ClientError class ImageError(Exception): "Custom exception for errors returned by Amazon Titan Text models" def __init__(self, message): self.message = message logger = logging.getLogger(__name__) logging.basicConfig(level=logging.INFO) def generate_text(model_id, body): """ Generate text using Amazon Titan Text models on demand. Args: model_id (str): The model ID to use. body (str) : The request body to use. Returns: response (json): The response from the model. """ logger.info( "Generating text with Amazon Titan Text model %s", model_id) bedrock = boto3.client(service_name='bedrock-runtime') accept = "application/json" content_type = "application/json" response = bedrock.invoke_model( body=body, modelId=model_id, accept=accept, contentType=content_type ) response_body = json.loads(response.get("body").read()) finish_reason = response_body.get("error") if finish_reason is not None: raise ImageError(f"Text generation error. Error is {finish_reason}") logger.info( "Successfully generated text with Amazon Titan Text model %s", model_id) return response_body def main(): """ Entrypoint for Amazon Titan Text model example. """ try: logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s") # You can replace the model_id with any other Titan Text Models # Titan Text Model family model_id is as mentioned below: # amazon.titan-text-premier-v1:0, amazon.titan-text-express-v1, amazon.titan-text-lite-v1 model_id = 'amazon.titan-text-premier-v1:0' prompt = """Meeting transcript: Miguel: Hi Brant, I want to discuss the workstream for our new product launch Brant: Sure Miguel, is there anything in particular you want to discuss? Miguel: Yes, I want to talk about how users enter into the product. Brant: Ok, in that case let me add in Namita. Namita: Hey everyone Brant: Hi Namita, Miguel wants to discuss how users enter into the product. Miguel: its too complicated and we should remove friction. for example, why do I need to fill out additional forms? I also find it difficult to find where to access the product when I first land on the landing page. Brant: I would also add that I think there are too many steps. Namita: Ok, I can work on the landing page to make the product more discoverable but brant can you work on the additonal forms? Brant: Yes but I would need to work with James from another team as he needs to unblock the sign up workflow. Miguel can you document any other concerns so that I can discuss with James only once? Miguel: Sure. From the meeting transcript above, Create a list of action items for each person. """ body = json.dumps({ "inputText": prompt, "textGenerationConfig": { "maxTokenCount": 3072, "stopSequences": [], "temperature": 0.7, "topP": 0.9 } }) response_body = generate_text(model_id, body) print(f"Input token count: {response_body['inputTextTokenCount']}") for result in response_body['results']: print(f"Token count: {result['tokenCount']}") print(f"Output text: {result['outputText']}") print(f"Completion reason: {result['completionReason']}") except ClientError as err: message = err.response["Error"]["Message"] logger.error("A client error occurred: %s", message) print("A client error occured: " + format(message)) except ImageError as err: logger.error(err.message) print(err.message) else: print( f"Finished generating text with the Amazon Titan Text Premier model {model_id}.") if __name__ == "__main__": main()

O exemplo a seguir mostra como executar inferência com o modelo Amazon Titan Text G1 - Express usando o SDK para Python.

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. # SPDX-License-Identifier: Apache-2.0 """ Shows how to create a list of action items from a meeting transcript with the Amazon &titan-text-express; model (on demand). """ import json import logging import boto3 from botocore.exceptions import ClientError class ImageError(Exception): "Custom exception for errors returned by Amazon &titan-text-express; model" def __init__(self, message): self.message = message logger = logging.getLogger(__name__) logging.basicConfig(level=logging.INFO) def generate_text(model_id, body): """ Generate text using Amazon &titan-text-express; model on demand. Args: model_id (str): The model ID to use. body (str) : The request body to use. Returns: response (json): The response from the model. """ logger.info( "Generating text with Amazon &titan-text-express; model %s", model_id) bedrock = boto3.client(service_name='bedrock-runtime') accept = "application/json" content_type = "application/json" response = bedrock.invoke_model( body=body, modelId=model_id, accept=accept, contentType=content_type ) response_body = json.loads(response.get("body").read()) finish_reason = response_body.get("error") if finish_reason is not None: raise ImageError(f"Text generation error. Error is {finish_reason}") logger.info( "Successfully generated text with Amazon &titan-text-express; model %s", model_id) return response_body def main(): """ Entrypoint for Amazon &titan-text-express; example. """ try: logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s") model_id = 'amazon.titan-text-express-v1' prompt = """Meeting transcript: Miguel: Hi Brant, I want to discuss the workstream for our new product launch Brant: Sure Miguel, is there anything in particular you want to discuss? Miguel: Yes, I want to talk about how users enter into the product. Brant: Ok, in that case let me add in Namita. Namita: Hey everyone Brant: Hi Namita, Miguel wants to discuss how users enter into the product. Miguel: its too complicated and we should remove friction. for example, why do I need to fill out additional forms? I also find it difficult to find where to access the product when I first land on the landing page. Brant: I would also add that I think there are too many steps. Namita: Ok, I can work on the landing page to make the product more discoverable but brant can you work on the additonal forms? Brant: Yes but I would need to work with James from another team as he needs to unblock the sign up workflow. Miguel can you document any other concerns so that I can discuss with James only once? Miguel: Sure. From the meeting transcript above, Create a list of action items for each person. """ body = json.dumps({ "inputText": prompt, "textGenerationConfig": { "maxTokenCount": 4096, "stopSequences": [], "temperature": 0, "topP": 1 } }) response_body = generate_text(model_id, body) print(f"Input token count: {response_body['inputTextTokenCount']}") for result in response_body['results']: print(f"Token count: {result['tokenCount']}") print(f"Output text: {result['outputText']}") print(f"Completion reason: {result['completionReason']}") except ClientError as err: message = err.response["Error"]["Message"] logger.error("A client error occurred: %s", message) print("A client error occured: " + format(message)) except ImageError as err: logger.error(err.message) print(err.message) else: print( f"Finished generating text with the Amazon &titan-text-express; model {model_id}.") if __name__ == "__main__": main()