RAG injection with OpenAI and Valkey
Configure the AI RAG Injector policy to use Valkey as a vector database and the OpenAI text-embedding-3-large model for embedding.
Valkey is automatically detected when using the redis vectordb strategy. AI Gateway queries the server and uses the Valkey-specific driver when it detects a Valkey backend.
This example uses
text-embedding-3-large, which supports up to3072dimensions. If you usetext-embedding-ada-002instead, you must setdimensionsto exactly1536.
Prerequisites
-
An existing AI Model.
-
You have an OpenAI account.
-
A Valkey 8.x instance.
-
Port
6379, or your custom Valkey port, is open and reachable from AI Gateway.
Environment variables
-
OPENAI_API_KEY: The API key to use to connect to OpenAI. -
VALKEY_HOST: The Valkey server’s host
ai_gateway_policies:
- ref: rag-injector-openai-valkey
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
name: rag-injector-openai-valkey
display_name: RAG injection with OpenAI and Valkey
type: ai-rag-injector
enabled: true
config:
inject_template: |
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
embeddings:
auth:
header_name: Authorization
header_value: Bearer $OPENAI_API_KEY
model:
provider: openai
name: text-embedding-3-large
vectordb:
strategy: redis
redis:
host: !env VALKEY_HOST
port: 6379
distance_metric: cosine
dimensions: 3072
global: falseMake sure to replace the following placeholders with your own values:
-
AI_GATEWAY_ID: Theidof your AI Gateway. -
!env OPENAI_API_KEY: The API key to use to connect to OpenAI. -
!env VALKEY_HOST: The Valkey server’s host
curl -X POST https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/policies \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"name": "rag-injector-openai-valkey",
"display_name": "RAG injection with OpenAI and Valkey",
"type": "ai-rag-injector",
"enabled": true,
"config": {
"inject_template": "Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.\n<RAG><CONTEXT></RAG>\nUser's question: <PROMPT>\n",
"embeddings": {
"auth": {
"header_name": "Authorization",
"header_value": "Bearer '$OPENAI_API_KEY'"
},
"model": {
"provider": "openai",
"name": "text-embedding-3-large"
}
},
"vectordb": {
"strategy": "redis",
"redis": {
"host": "'$VALKEY_HOST'",
"port": 6379
},
"distance_metric": "cosine",
"dimensions": 3072
}
},
"global": false
}
'Make sure to replace the following placeholders with your own values:
-
region: Geographic region where your Kong Konnect is hosted and operates. -
KONNECT_TOKEN: Your Personal Access Token (PAT) associated with your Konnect account. -
AIGatewayId: Theidof the AI Gateway.
See the Konnect AI Gateway API reference to learn about region-specific URLs and personal access tokens.
Prerequisite: Configure your Personal Access Token
terraform {
required_providers {
konnect = {
source = "kong/konnect"
}
}
}
provider "konnect" {
personal_access_token = "$KONNECT_TOKEN"
server_url = "https://us.api.konghq.com/"
}Add the following to your Terraform configuration:
resource "konnect_ai_gateway_policy" "my_policy" {
name = "rag-injector-openai-valkey"
display_name = "RAG injection with OpenAI and Valkey"
type = "ai-rag-injector"
enabled = true
global = false
config = jsonencode({
inject_template = <<EOF
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
EOF
embeddings = {
auth = {
header_name = "Authorization"
header_value = "Bearer var.openai_api_key"
}
model = {
provider = "openai"
name = "text-embedding-3-large"
}
}
vectordb = {
strategy = "redis"
redis = {
host = var.valkey_host
port = 6379
}
distance_metric = "cosine"
dimensions = 3072
}
})
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}ai_gateway_policies:
- ref: rag-injector-openai-valkey
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
name: rag-injector-openai-valkey
display_name: RAG injection with OpenAI and Valkey
type: ai-rag-injector
enabled: true
config:
inject_template: |
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
embeddings:
auth:
header_name: Authorization
header_value: Bearer $OPENAI_API_KEY
model:
provider: openai
name: text-embedding-3-large
vectordb:
strategy: redis
redis:
host: !env VALKEY_HOST
port: 6379
distance_metric: cosine
dimensions: 3072
global: falseMake sure to replace the following placeholders with your own values:
-
AI_GATEWAY_ID: Theidof your AI Gateway. -
!env OPENAI_API_KEY: The API key to use to connect to OpenAI. -
!env VALKEY_HOST: The Valkey server’s host
curl -X POST https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/policies \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"name": "rag-injector-openai-valkey",
"display_name": "RAG injection with OpenAI and Valkey",
"type": "ai-rag-injector",
"enabled": true,
"config": {
"inject_template": "Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.\n<RAG><CONTEXT></RAG>\nUser's question: <PROMPT>\n",
"embeddings": {
"auth": {
"header_name": "Authorization",
"header_value": "Bearer '$OPENAI_API_KEY'"
},
"model": {
"provider": "openai",
"name": "text-embedding-3-large"
}
},
"vectordb": {
"strategy": "redis",
"redis": {
"host": "'$VALKEY_HOST'",
"port": 6379
},
"distance_metric": "cosine",
"dimensions": 3072
}
},
"global": false
}
'Make sure to replace the following placeholders with your own values:
-
region: Geographic region where your Kong Konnect is hosted and operates. -
KONNECT_TOKEN: Your Personal Access Token (PAT) associated with your Konnect account. -
AIGatewayId: Theidof the AI Gateway.
See the Konnect AI Gateway API reference to learn about region-specific URLs and personal access tokens.
Prerequisite: Configure your Personal Access Token
terraform {
required_providers {
konnect = {
source = "kong/konnect"
}
}
}
provider "konnect" {
personal_access_token = "$KONNECT_TOKEN"
server_url = "https://us.api.konghq.com/"
}Add the following to your Terraform configuration:
resource "konnect_ai_gateway_policy" "my_policy" {
name = "rag-injector-openai-valkey"
display_name = "RAG injection with OpenAI and Valkey"
type = "ai-rag-injector"
enabled = true
global = false
config = jsonencode({
inject_template = <<EOF
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
EOF
embeddings = {
auth = {
header_name = "Authorization"
header_value = "Bearer var.openai_api_key"
}
model = {
provider = "openai"
name = "text-embedding-3-large"
}
}
vectordb = {
strategy = "redis"
redis = {
host = var.valkey_host
port = 6379
}
distance_metric = "cosine"
dimensions = 3072
}
})
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}ai_gateway_policies:
- ref: rag-injector-openai-valkey
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
name: rag-injector-openai-valkey
display_name: RAG injection with OpenAI and Valkey
type: ai-rag-injector
enabled: true
config:
inject_template: |
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
embeddings:
auth:
header_name: Authorization
header_value: Bearer $OPENAI_API_KEY
model:
provider: openai
name: text-embedding-3-large
vectordb:
strategy: redis
redis:
host: !env VALKEY_HOST
port: 6379
distance_metric: cosine
dimensions: 3072
global: falseMake sure to replace the following placeholders with your own values:
-
AI_GATEWAY_ID: Theidof your AI Gateway. -
!env OPENAI_API_KEY: The API key to use to connect to OpenAI. -
!env VALKEY_HOST: The Valkey server’s host
curl -X POST https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/policies \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"name": "rag-injector-openai-valkey",
"display_name": "RAG injection with OpenAI and Valkey",
"type": "ai-rag-injector",
"enabled": true,
"config": {
"inject_template": "Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.\n<RAG><CONTEXT></RAG>\nUser's question: <PROMPT>\n",
"embeddings": {
"auth": {
"header_name": "Authorization",
"header_value": "Bearer '$OPENAI_API_KEY'"
},
"model": {
"provider": "openai",
"name": "text-embedding-3-large"
}
},
"vectordb": {
"strategy": "redis",
"redis": {
"host": "'$VALKEY_HOST'",
"port": 6379
},
"distance_metric": "cosine",
"dimensions": 3072
}
},
"global": false
}
'Make sure to replace the following placeholders with your own values:
-
region: Geographic region where your Kong Konnect is hosted and operates. -
KONNECT_TOKEN: Your Personal Access Token (PAT) associated with your Konnect account. -
AIGatewayId: Theidof the AI Gateway.
See the Konnect AI Gateway API reference to learn about region-specific URLs and personal access tokens.
Prerequisite: Configure your Personal Access Token
terraform {
required_providers {
konnect = {
source = "kong/konnect"
}
}
}
provider "konnect" {
personal_access_token = "$KONNECT_TOKEN"
server_url = "https://us.api.konghq.com/"
}Add the following to your Terraform configuration:
resource "konnect_ai_gateway_policy" "my_policy" {
name = "rag-injector-openai-valkey"
display_name = "RAG injection with OpenAI and Valkey"
type = "ai-rag-injector"
enabled = true
global = false
config = jsonencode({
inject_template = <<EOF
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
EOF
embeddings = {
auth = {
header_name = "Authorization"
header_value = "Bearer var.openai_api_key"
}
model = {
provider = "openai"
name = "text-embedding-3-large"
}
}
vectordb = {
strategy = "redis"
redis = {
host = var.valkey_host
port = 6379
}
distance_metric = "cosine"
dimensions = 3072
}
})
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}ai_gateway_policies:
- ref: rag-injector-openai-valkey
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
name: rag-injector-openai-valkey
display_name: RAG injection with OpenAI and Valkey
type: ai-rag-injector
enabled: true
config:
inject_template: |
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
embeddings:
auth:
header_name: Authorization
header_value: Bearer $OPENAI_API_KEY
model:
provider: openai
name: text-embedding-3-large
vectordb:
strategy: redis
redis:
host: !env VALKEY_HOST
port: 6379
distance_metric: cosine
dimensions: 3072
global: trueMake sure to replace the following placeholders with your own values:
-
AI_GATEWAY_ID: Theidof your AI Gateway. -
!env OPENAI_API_KEY: The API key to use to connect to OpenAI. -
!env VALKEY_HOST: The Valkey server’s host
curl -X POST https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/policies \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"name": "rag-injector-openai-valkey",
"display_name": "RAG injection with OpenAI and Valkey",
"type": "ai-rag-injector",
"enabled": true,
"config": {
"inject_template": "Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.\n<RAG><CONTEXT></RAG>\nUser's question: <PROMPT>\n",
"embeddings": {
"auth": {
"header_name": "Authorization",
"header_value": "Bearer '$OPENAI_API_KEY'"
},
"model": {
"provider": "openai",
"name": "text-embedding-3-large"
}
},
"vectordb": {
"strategy": "redis",
"redis": {
"host": "'$VALKEY_HOST'",
"port": 6379
},
"distance_metric": "cosine",
"dimensions": 3072
}
},
"global": true
}
'Make sure to replace the following placeholders with your own values:
-
region: Geographic region where your Kong Konnect is hosted and operates. -
KONNECT_TOKEN: Your Personal Access Token (PAT) associated with your Konnect account. -
AIGatewayId: Theidof the AI Gateway.
See the Konnect AI Gateway API reference to learn about region-specific URLs and personal access tokens.
Prerequisite: Configure your Personal Access Token
terraform {
required_providers {
konnect = {
source = "kong/konnect"
}
}
}
provider "konnect" {
personal_access_token = "$KONNECT_TOKEN"
server_url = "https://us.api.konghq.com/"
}Add the following to your Terraform configuration:
resource "konnect_ai_gateway_policy" "my_policy" {
name = "rag-injector-openai-valkey"
display_name = "RAG injection with OpenAI and Valkey"
type = "ai-rag-injector"
enabled = true
global = true
config = jsonencode({
inject_template = <<EOF
Only use the following information surrounded by <RAG></RAG> and your existing knowledge to provide the best possible answer to the user.
<RAG><CONTEXT></RAG>
User's question: <PROMPT>
EOF
embeddings = {
auth = {
header_name = "Authorization"
header_value = "Bearer var.openai_api_key"
}
model = {
provider = "openai"
name = "text-embedding-3-large"
}
}
vectordb = {
strategy = "redis"
redis = {
host = var.valkey_host
port = 6379
}
distance_metric = "cosine"
dimensions = 3072
}
})
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}After creating the policy, add rag-injector-openai-valkey to the policies array of your AI Model. The tool examples update an existing AI Model with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Model.
ai_gateway_models:
- ref: my-gpt-4o
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
display_name: my-gpt-4o
name: my-gpt-4o
type: model
capabilities:
- generate
formats:
- type: openai
targets:
- name: gpt-4o
provider: my-openai-account
config:
type: openai
config:
route:
paths:
- "/v1"
model:
body_param: model
values:
- my-gpt-4o
logging:
payloads: false
policies:
- !ref rag-injector-openai-valkey#nameAfter creating the policy, add rag-injector-openai-valkey to the policies array of your AI Model. The tool examples update an existing AI Model with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Model.
curl -X PUT https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/models/my-gpt-4o \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"display_name": "my-gpt-4o",
"name": "my-gpt-4o",
"type": "model",
"capabilities": [
"generate"
],
"formats": [
{
"type": "openai"
}
],
"policies": [
"rag-injector-openai-valkey"
],
"targets": [
{
"name": "gpt-4o",
"provider": "my-openai-account",
"config": {
"type": "openai"
}
}
],
"config": {
"route": {
"paths": [
"/v1"
],
"model": {
"body_param": "model",
"values": [
"my-gpt-4o"
]
}
},
"logging": {
"payloads": false
}
}
}
'After creating the policy, add rag-injector-openai-valkey to the policies array of your AI Model. The tool examples update an existing AI Model with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Model.
resource "konnect_ai_gateway_model" "my_model" {
display_name = "my-gpt-4o"
name = "my-gpt-4o"
type = "model"
capabilities = ["generate"]
formats = [
{
type = "openai"
} ]
targets = [
{
name = "gpt-4o"
provider = "my-openai-account"
config = {
type = "openai"
}
} ]
config = jsonencode({
route = {
paths = ["/v1"]
model = {
body_param = "model"
values = ["my-gpt-4o"]
}
}
logging = {
payloads = false
}
})
policies = [konnect_ai_gateway_policy.my_policy.name]
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}After creating the policy, add rag-injector-openai-valkey to the policies array of your AI Consumer. The tool examples update an existing AI Consumer with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Consumer.
ai_gateway_consumers:
- ref: mobile-app-production
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
display_name: Mobile App - Production
name: mobile-app-production
type: api-key
policies:
- !ref rag-injector-openai-valkey#nameAfter creating the policy, add rag-injector-openai-valkey to the policies array of your AI Consumer. The tool examples update an existing AI Consumer with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Consumer.
curl -X PUT https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/consumers/mobile-app-production \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"display_name": "Mobile App - Production",
"name": "mobile-app-production",
"type": "api-key",
"policies": [
"rag-injector-openai-valkey"
]
}
'After creating the policy, add rag-injector-openai-valkey to the policies array of your AI Consumer. The tool examples update an existing AI Consumer with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Consumer.
resource "konnect_ai_gateway_consumer" "my_consumer" {
display_name = "Mobile App - Production"
name = "mobile-app-production"
type = "api-key"
policies = [konnect_ai_gateway_policy.my_policy.name]
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}After creating the policy, add rag-injector-openai-valkey to the policies array of your AI Consumer Group. The tool examples update an existing AI Consumer Group with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Consumer Group.
ai_gateway_consumer_groups:
- ref: internal-teams
ai_gateway: !lookup {id: !env AI_GATEWAY_ID}
display_name: Internal Teams
name: internal-teams
policies:
- !ref rag-injector-openai-valkey#nameAfter creating the policy, add rag-injector-openai-valkey to the policies array of your AI Consumer Group. The tool examples update an existing AI Consumer Group with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Consumer Group.
curl -X PUT https://{region}.api.konghq.com/v1/ai-gateways/{AIGatewayId}/consumer-groups/internal-teams \
--header "accept: application/json" \
--header "Content-Type: application/json" \
--header "Authorization: Bearer $KONNECT_TOKEN" \
--data '
{
"display_name": "Internal Teams",
"name": "internal-teams",
"policies": [
"rag-injector-openai-valkey"
]
}
'After creating the policy, add rag-injector-openai-valkey to the policies array of your AI Consumer Group. The tool examples update an existing AI Consumer Group with the policy attached: replace the entity configuration in the example with your own, as the update replaces the entire AI Consumer Group.
resource "konnect_ai_gateway_consumer_group" "my_consumer_group" {
display_name = "Internal Teams"
name = "internal-teams"
policies = [konnect_ai_gateway_policy.my_policy.name]
gateway_id = konnect_ai_gateway.my_ai_gateway.id
}This example requires the following variables to be added to your manifest. You can specify values at runtime by setting TF_VAR_name=value.
variable "openai_api_key" {
type = string
}
variable "valkey_host" {
type = string
}