订阅以接收新文章的通知:

Vectorize:将 AI 驱动应用程序快速交付生产的矢量数据库

2023-09-27

7 分钟阅读时间
这篇博文也有 EnglishDeutsch日本語한국어EspañolFrançais版本。

Vectorize 是我们全新推出的矢量数据库产品,旨在让您完全在 Cloudflare 的全球网络上构建 AI 加持的全栈应用程序;现已可用。Vectorize 目前处于公测阶段,对任何使用 Cloudflare Workers的开发人员可用。

Vectorize: a vector database for shipping AI-powered applications to production, fast

可将 Vectorize 配合 Workers AI使用,以直接通过 Workers 进行语义搜索、分类、推荐和异常检测等用例,提高 LLM(大型语言模型)的答案准确性和上下文,以及使用来自流行平台(包括 OpenAI 和 Cohere)的自定义嵌入。

欢迎访问 Vectorize 文档 以开始使用,或继续阅读,以进一步了解矢量数据库的作用以及 Vectorize 的不同之处。

为什么我需要一个向量数据库?

机器学习模型无法记住任何事情:只能记住它们训练过的内容。

向量数据库旨在解决这个问题,它捕捉机器学习模型表示数据的方式,包括结构化和非结构化的文本、图像和音频,并以一种可以与_未来_输入进行比较的方式存储数据。这使我们能够利用现有机器学习模型和大型语言模型(LLM)处理它们未经过训练的内容:考虑到训练模型的巨大成本,这将是非常强大的。

为了更好地说明为什么像 Vectorize 这样的矢量数据库是有用的,让我们假设它们不存在,并看看向机器学习模型或大预言模型为语义搜索或推荐任务提供上下文是多么痛苦的事情。我们的目标是理解与我们的查询相似的内容并返回它:基于我们自己的数据集。

  1. 收到我们的用户查询输入:他们正在搜索“如何从 Cloudflare Workers 写入 R2”。

  2. 我们加载整个文档数据集——幸好是一个“小”数据集,大约有 65000 个句子,2.1 GB——并将其与用户的查询一起提供。这使得模型可以根据我们的数据获取所需的上下文。

  3. 我们等待。

  4. (很长时间)

  5. 我们获得与用户查询最相似的句子的相似度分数,然后通过将它们映射回 URL 来返回我们的搜索结果。

……然后收到另一个查询,我们不得不重新开始这个过程。

在实践中,这其实是不可能的:我们无法在 API 调用(提示)中传递如此多的上下文给大多数机器学习模型,即使我们可以,也需要巨大的内存和时间来反复处理我们的数据集。

通过使用向量数据库,我们无需重复第 2 步:我们只需执行一次,或在数据集更新时执行,并使用向量数据库为我们的机器学习模型提供一种形式的长期记忆。我们的工作流程看起来更像是这样:

  1. 我们加载整个文档数据集,将其通过我们的模型运行,并将生成的向量嵌入存储在我们的向量数据库中(仅一次)。

  2. 对于每个用户查询(且仅该查询),我们向同一模型提出请求并检索一个向量表示。

  3. 我们使用该查询向量查询我们的向量数据库,该数据库返回与查询向量最接近的向量。

如果我们将这两个流程并排放在一起看,我们可以很快地看出,如果没有向量数据库,使用将我们的数据集与现有模型一起使用是多么低效和不实用:

使用向量数据库来帮助机器学习模型记住东西。

从这个简单的示例中,您可能开始明白了一些道理:但是您可能也想知道为什么需要矢量数据库而不是常规数据库。

向量是模型对输入的表示:它如何将输入映射到其内部结构,即“特征”。广义上说,向量越相似,模型就认为这些输入越相似,这是基于模型从输入中提取特征的方式。

只看几个维度的示例向量时,这似乎很容易。但是在实际输出中,跨越 10000 到 250000 个向量进行搜索,每个向量可能有 1536 个维度,那可不是小任务。这就是向量数据库发挥作用的地方了:为了使搜索在规模上起作用,向量数据库使用一种特定类别的算法,例如 k 最近邻(kNN) 或其他近似最近邻 (ANN) 算法,来确定向量的相似度。

尽管向量数据库在构建 AI 和机器学习应用程序时非常有用,但它们_不仅仅_在这些用例中有用:它们可以用于多种分类和异常检测任务。了解查询输入与其他输入是否相似或潜在不相似可以用于内容审核(是否与已知的不良内容匹配?)和安全警报(我之前见过这个吗?)等任务。

使用向量搜索构建一个推荐引擎

我们将 Vectorize 打造成 Workers AI 的强大伙伴: 让您能够尽可能地在靠近用户的位置运行向量搜索任务,而无需考虑如何为生产环境而扩展。

我们将举一个真实世界的例子——为一个电子商务商店构建一个(产品)推荐引擎——并简化一些事情。

我们的目标是在每个产品列表页面上显示一个“相关产品”的列表:这是向量搜索的一个完美用例。在这个示例中,我们的输入向量是占位符,但在实际应用中,我们可以基于产品描述和/或购物车数据,将这些数据通过一个句子相似性模型(如 Worker AI 文本嵌入模型)来生成它们。

每个向量代表商店中的一个产品,并将产品的 URL 与之关联。我们还可以将每个向量的 ID 设置为产品ID:这两种方法都是有效的。我们的查询 —— 向量搜索 —— 代表用户当前正在查看的产品的产品描述和内容。让我们分步看看这在代码中的样子:本例是直接从我们的 开发人员文档提取出来的:

上面的代码故意保持简单,但是它展示了向量搜索的核心:我们将向量插入到数据库中,并查询与我们的查询向量距离最小的向量。

以下是结果,以及包含的数值,以便我们直观地观察到我们的查询向量[54.8, 5.5, 3.1] 与我们从搜索返回、得分最高的匹配项: [58.799, 6.699, 3.400] 相似。这个索引使用 余弦 相似度来计算向量之间的距离,这意味着得分越接近 1,匹配项与我们的查询向量越相似。

export interface Env {
	// This makes our vector index methods available on env.MY_VECTOR_INDEX.*
	// e.g. env.MY_VECTOR_INDEX.insert() or .query()
	TUTORIAL_INDEX: VectorizeIndex;
}

// Sample vectors: 3 dimensions wide.
//
// Vectors from a machine-learning model are typically ~100 to 1536 dimensions
// wide (or wider still).
const sampleVectors: Array<VectorizeVector> = [
	{ id: '1', values: [32.4, 74.1, 3.2], metadata: { url: '/products/sku/13913913' } },
	{ id: '2', values: [15.1, 19.2, 15.8], metadata: { url: '/products/sku/10148191' } },
	{ id: '3', values: [0.16, 1.2, 3.8], metadata: { url: '/products/sku/97913813' } },
	{ id: '4', values: [75.1, 67.1, 29.9], metadata: { url: '/products/sku/418313' } },
	{ id: '5', values: [58.8, 6.7, 3.4], metadata: { url: '/products/sku/55519183' } },
];

export default {
	async fetch(request: Request, env: Env, ctx: ExecutionContext): Promise<Response> {
		if (new URL(request.url).pathname !== '/') {
			return new Response('', { status: 404 });
		}
		// Insert some sample vectors into our index
		// In a real application, these vectors would be the output of a machine learning (ML) model,
		// such as Workers AI, OpenAI, or Cohere.
		let inserted = await env.TUTORIAL_INDEX.insert(sampleVectors);

		// Log the number of IDs we successfully inserted
		console.info(`inserted ${inserted.count} vectors into the index`);

		// In a real application, we would take a user query - e.g. "durable
		// objects" - and transform it into a vector emebedding first.
		//
		// In our example, we're going to construct a simple vector that should
		// match vector id #5
		let queryVector: Array<number> = [54.8, 5.5, 3.1];

		// Query our index and return the three (topK = 3) most similar vector
		// IDs with their similarity score.
		//
		// By default, vector values are not returned, as in many cases the
		// vectorId and scores are sufficient to map the vector back to the
		// original content it represents.
		let matches = await env.TUTORIAL_INDEX.query(queryVector, { topK: 3, returnVectors: true });

		// We map over our results to find the most similar vector result.
		//
		// Since our index uses the 'cosine' distance metric, scores will range
		// from 1 to -1.  A value of '1' means the vector is the same; the
		// closer to 1, the more similar. Values of -1 (least similar) and 0 (no
		// match).
		// let closestScore = 0;
		// let mostSimilarId = '';
		// matches.matches.map((match) => {
		// 	if (match.score > closestScore) {
		// 		closestScore = match.score;
		// 		mostSimilarId = match.vectorId;
		// 	}
		// });

		return Response.json({
			// This will return the closest vectors: we'll see that the vector
			// with id = 5 has the highest score (closest to 1.0) as the
			// distance between it and our query vector is the smallest.
			// Return the full set of matches so we can see the possible scores.
			matches: matches,
		});
	},
};

在实际应用中,我们现在可以基于最相似的产品快速返回产品推荐的 URL,按它们的分数(从高到低)进行排序,并且如果我们想显示更多,可以增加 topK 值。与每个向量一起存储的元数据嵌入一个 R2 对象的路径, D1 数据库中一行的 UUID,或来自 Workers KV 的一个键值对。

Workers AI + Vectorize:Cloudflare 上的全栈向量搜索

{
  "matches": {
    "count": 3,
    "matches": [
      {
        "score": 0.999909,
        "vectorId": "5",
        "vector": {
          "id": "5",
          "values": [
            58.79999923706055,
            6.699999809265137,
            3.4000000953674316
          ],
          "metadata": {
            "url": "/products/sku/55519183"
          }
        }
      },
      {
        "score": 0.789848,
        "vectorId": "4",
        "vector": {
          "id": "4",
          "values": [
            75.0999984741211,
            67.0999984741211,
            29.899999618530273
          ],
          "metadata": {
            "url": "/products/sku/418313"
          }
        }
      },
      {
        "score": 0.611976,
        "vectorId": "2",
        "vector": {
          "id": "2",
          "values": [
            15.100000381469727,
            19.200000762939453,
            15.800000190734863
          ],
          "metadata": {
            "url": "/products/sku/10148191"
          }
        }
      }
    ]
  }
}

在实际应用中,我们需要一个机器学习模型,它既可以从我们的原始数据集中生成向量嵌入(用于初始化我们的数据库),也可以_快速_将用户查询转换为向量嵌入。这些向量嵌入需要来自同一个模型,因为每个模型以不同的方式表示特征。

这是一个在 Cloudflare 上构建完整端到端向量搜索流程的简洁示例:

上面的代码做了四件事情:

  1. 它将这三个句子传递给 Workers AI 文本嵌入模型 (@cf/baai/bge-base-en-v1.5) 并检索它们的向量嵌入。

  2. 它将这些向量插入到我们的 Vectorize 索引中。

  3. 通过相同的 Workers AI 模型,将用户查询转换为向量嵌入。

  4. 查询我们的 Vectorize 索引以获取匹配项。

import { Ai } from '@cloudflare/ai';
export interface Env {
	TEXT_EMBEDDINGS: VectorizeIndex;
	AI: any;
}
interface EmbeddingResponse {
	shape: number[];
	data: number[][];
}

export default {
	async fetch(request: Request, env: Env, ctx: ExecutionContext): Promise<Response> {
		const ai = new Ai(env.AI);
		let path = new URL(request.url).pathname;
		if (path.startsWith('/favicon')) {
			return new Response('', { status: 404 });
		}

		// We only need to generate vector embeddings just the once (or as our
		// data changes), not on every request
		if (path === '/insert') {
			// In a real-world application, we could read in content from R2 or
			// a SQL database (like D1) and pass it to Workers AI
			const stories = ['This is a story about an orange cloud', 'This is a story about a llama', 'This is a story about a hugging emoji'];
			const modelResp: EmbeddingResponse = await ai.run('@cf/baai/bge-base-en-v1.5', {
				text: stories,
			});

			// We need to convert the vector embeddings into a format Vectorize can accept.
			// Each vector needs an id, a value (the vector) and optional metadata.
			// In a real app, our ID would typicaly be bound to the ID of the source
			// document.
			let vectors: VectorizeVector[] = [];
			let id = 1;
			modelResp.data.forEach((vector) => {
				vectors.push({ id: `${id}`, values: vector });
				id++;
			});

			await env.TEXT_EMBEDDINGS.upsert(vectors);
		}

		// Our query: we expect this to match vector id: 1 in this simple example
		let userQuery = 'orange cloud';
		const queryVector: EmbeddingResponse = await ai.run('@cf/baai/bge-base-en-v1.5', {
			text: [userQuery],
		});

		let matches = await env.TEXT_EMBEDDINGS.query(queryVector.data[0], { topK: 1 });
		return Response.json({
			// We expect vector id: 1 to be our top match with a score of
			// ~0.896888444
			// We are using a cosine distance metric, where the closer to one,
			// the more similar.
			matches: matches,
		});
	},
};

这个例子看起来可能“太”简单了,但在实际的应用中,我们只需要改变两个地方:只需一次性插入我们的向量(或通过 Cron Triggers定期进行) 并将我们的三个示例句子替换为存储在R2、D1 数据库或其他存储提供商中的真实数据。

实际上,这与我们运行 Cursor 的方式非常相似,这是一个可以回答 Cloudflare Worker 相关问题的 AI 助手:我们已将 Cursor 迁移到 Workers AI 和 Vectorize 上运行。我们使用内置的文本嵌入模型从我们的开发人员文档中生成文本嵌入,将它们插入到 Vectorize 索引中,并通过相同的模型实时转换用户查询。

从您最喜欢的 AI API 导入自有嵌入

Vectorize 并不局限于 Workers AI,它是一个完全成熟、独立的向量数据库。

如果您已经在使用 OpenAI 的嵌入 API,Cohere 的 多语言模型,或任何其他嵌入 API,那么您都可以轻松地将自己的向量带到 Vectorize 中。

它以完成相同的方式工作:生成您的嵌入,将它们插入到 Vectorize 中,在查询索引之前通过模型传递您的查询。Vectorize 包含一些最流行的嵌入模型的快捷方式。

如果您已经围绕现有的嵌入 API 建立了一个工作流程,并且/或者已经验证了一个特定的多模态或多语言嵌入模型以适应您的用例,那么这将特别有用。

使 AI 的成本可预测

# Vectorize has ready-to-go presets that set the dimensions and distance metric for popular embeddings models
$ wrangler vectorize create openai-index-example --preset=openai-text-embedding-ada-002

对于 AI 和 ML,人们都非常兴奋,但也有一个重大关注点:它的实验成本太高,而且难以预测规模化的成本。

通过 Vectorize,我们希望为矢量数据库引入一种更简单的定价模型。工作中需要进行概念验证?我们的免费级限制应该适合。扩展和优化嵌入维度,以便在性能和准确性之间取得平衡?这应该不会带来高额成本。

重要的是,Vectorize 旨在提供可预测性:您无需估计 CPU 和内存消耗,这在刚开始时可能很困难,而在生产环境中为全新的用例计划高峰和低峰时段则更加困难。相反,您的费用是根据您存储的向量维度总数以及每个月对其进行的查询次数来计算的。我们的任务是处理扩展以满足您的查询模式。

以下是 Vectorize 的定价——而且,如果您现在已经使用 Workers 付费计划,则可完全免费使用 Vectorize 至 2024 年:

.tg {border-collapse:collapse;border-color:#ccc;border-spacing:0;} .tg td{background-color:#fff;border-color:#ccc;border-style:solid;border-width:1px;color:#333; font-family:Arial, sans-serif;font-size:14px;overflow:hidden;padding:10px 5px;word-break:normal;} .tg th{background-color:#f0f0f0;border-color:#ccc;border-style:solid;border-width:1px;color:#333; font-family:Arial, sans-serif;font-size:14px;font-weight:normal;overflow:hidden;padding:10px 5px;word-break:normal;} .tg .tg-1wig{font-weight:bold;text-align:left;vertical-align:top} .tg .tg-baqh{text-align:center;vertical-align:top} .tg .tg-0lax{text-align:left;vertical-align:top} .tg .tg-amwm{font-weight:bold;text-align:center;vertical-align:top}

Workers 免费 (即将推出)

Workers Free (coming soon) Workers Paid ($5/month)
Queried vector dimensions included 30M total queried dimensions / month 50M total queried dimensions / month
Stored vector dimensions included 5M stored dimensions / month 10M stored dimensions / month
Additional cost $0.04 / 1M vector dimensions queried or stored $0.04 / 1M vector dimensions queried or stored

Workers 付费 (5 美元/月)

包含的查询向量维度

30M 总查询维度/月

50M 总查询维度/月

包含的存储向量维度

5M 存储维度/月

10M 存储维度/月

额外费用

$0.04 美元 / 1M 查询或存储的向量维度

$0.04 美元 / 1M 查询或存储的向量维度

定价完全基于您存储和查询的内容:(查询和存储的总向量维度)* 每个向量的维度 * 价格。需要更多查询?很容易预测。优化到每个向量较小的维度以提高速度和降低总延迟?成本降低。有一些索引用于原型设计或实验新用例?我们不按索引收费。

您可以根据需要创建尽可能多的索引,用于根据新想法的原型设计和/或将生产环境从开发环境分离。

举个例子:如果您加载了 10000 个 Workers AI 向量(每个向量有 384 个维度),并每天对您的索引进行 5000 次查询,那么总共查询的向量维度将达到 4900 万。这_仍然_适用于我们的 Workers 付费计划(每月 5 美元)。更好的是:我们不会因为不活动而删除您的索引。

请注意,尽管这个定价还不是最终版本,但我们预计日后只有很少的调整。我们希望避免出现令人意外的情况:在您投入时间编写代码、测试和学习一项技术的细节之_后_,发现定价无法接受,这是非常糟糕的。

Vectorize!

每个使用 Workers 付费计划的开发人员都可以立即开始使用 Vectorize:公测版现已可用,欢迎 访问我们的开发人员文档以开始使用

对于 Cloudflare 的向量数据库而言,这只是一个开端。在接下来的几周和几个月里,我们计划推出一个新的查询引擎,以进一步提高查询性能,支持更大的索引,引入子索引过滤功能,增加元数据限制,并提供按索引的分析功能。

如果你正在寻找关于构建的灵感, 请参考这个语义搜索教程,它结合使用 Workers AI 和 Vectorize 来进行文档搜索,并完全在 Cloudflare 上运行。也可以参考这个 例子,它结合使用 OpenAI 和 Vectorize,为 LLM 提供更多上下文,并显著提高其答案的准确性。

如果您需要向我们的产品和工程团队询问如何使用 Vectorize,或者只是想与其他在 Workers AI 上构建的开发人员交流想法,请在我们的 Developer Discord上加入 #vectorize 和 #workers-ai 频道。

我们保护整个企业网络,帮助客户高效构建互联网规模的应用程序,加速任何网站或互联网应用程序抵御 DDoS 攻击,防止黑客入侵,并能协助您实现 Zero Trust 的过程

从任何设备访问 1.1.1.1,以开始使用我们的免费应用程序,帮助您更快、更安全地访问互联网。要进一步了解我们帮助构建更美好互联网的使命,请从这里开始。如果您正在寻找新的职业方向,请查看我们的空缺职位
Birthday WeekVectorizeCloudflare WorkersDeveloper PlatformAIDatabase产品新闻开发人员

在 X 上关注

Matt Silverlock|@elithrar
Jérôme Schneider|@jeromeschneider
Cloudflare|@cloudflare

相关帖子

2024年10月09日 13:00

Improving platform resilience at Cloudflare through automation

We realized that we need a way to automatically heal our platform from an operations perspective, and designed and built a workflow orchestration platform to provide these self-healing capabilities across our global network. We explore how this has helped us to reduce the impact on our customers due to operational issues, and the rich variety of similar problems it has empowered us to solve....

2024年10月08日 13:00

Cloudflare acquires Kivera to add simple, preventive cloud security to Cloudflare One

The acquisition and integration of Kivera broadens the scope of Cloudflare’s SASE platform beyond just apps, incorporating increased cloud security through proactive configuration management of cloud services. ...

2024年9月27日 13:00

Network trends and natural language: Cloudflare Radar’s new Data Explorer & AI Assistant

The Cloudflare Radar Data Explorer provides a simple Web-based interface to build more complex API queries, including comparisons and filters, and visualize the results. The accompanying AI Assistant translates a user’s natural language statements or questions into the appropriate Radar API calls....