INNER CODE UNIT · JavaScript

updateParams

serverless/aws-ai-stack · ai-chat-api/handler.js:253

      const updateParams = (pk) => ({
        TableName: USAGE_TABLE_NAME,
        Key: {
          PK: pk,
          SK: `MODEL#${process.env.MODEL_ID}`,
        },
        UpdateExpression:
          "ADD invocationCount :inc, inputTokens :in, outputTokens :out, totalTokens :tot",
        ExpressionAttributeValues: {
          ":inc": 1,
          ":in": usage.inputTokens || 0,
          ":out": usage.outputTokens || 0,
          ":tot": usage.totalTokens || 0,
        },
      });
      const userUsageUpdateCommand = new UpdateCommand(
        updateParams(userUsageKey)
      );

View source record →