title: ai-prompt-decorator keywords:

  • Apache APISIX
  • API Gateway
  • Plugin
  • ai-prompt-decorator description: This document contains information about the Apache APISIX ai-prompt-decorator Plugin.

Description

The ai-prompt-decorator plugin simplifies access to LLM providers, such as OpenAI and Anthropic, and their models by appending or prepending prompts into the request.

Plugin Attributes

FieldRequiredTypeDescription
prependConditionally*ArrayAn array of prompt objects to be prepended
prepend.roleYesStringRole of the message (system, user, assistant)
prepend.contentYesStringContent of the message. Minimum length: 1
appendConditionally*ArrayAn array of prompt objects to be appended
append.roleYesStringRole of the message (system, user, assistant)
append.contentYesStringContent of the message. Minimum length: 1

* Conditionally Required: At least one of prepend or append must be provided.

Example usage

Create a route with the ai-prompt-decorator plugin like so:

curl "http://127.0.0.1:9180/apisix/admin/routes/1" -X PUT \
  -H "X-API-KEY: ${ADMIN_API_KEY}" \
  -d '{
    "uri": "/v1/chat/completions",
    "plugins": {
      "ai-prompt-decorator": {
        "prepend":[
          {
            "role": "system",
            "content": "I have exams tomorrow so explain conceptually and briefly"
          }
        ],
        "append":[
          {
            "role": "system",
            "content": "End the response with an analogy."
          }
        ]
      }
    },
    "upstream": {
      "type": "roundrobin",
      "nodes": {
        "api.openai.com:443": 1
      },
      "pass_host": "node",
      "scheme": "https"
    }
  }'

Now send a request:

curl http://127.0.0.1:9080/v1/chat/completions -i -XPOST  -H 'Content-Type: application/json' -d '{
  "model": "gpt-4",
  "messages": [{ "role": "user", "content": "What is TLS Handshake?" }]
}' -H "Authorization: Bearer <your token here>"

Then the request body will be modified to something like this:

{
  "model": "gpt-4",
  "messages": [
    {
      "role": "system",
      "content": "I have exams tomorrow so explain conceptually and briefly"
    },
    { "role": "user", "content": "What is TLS Handshake?" },
    {
      "role": "system",
      "content": "End the response with an analogy."
    }
  ]
}