vector.ts — drizzle-orm Source File

Architecture documentation for vector.ts, a typescript file in the drizzle-orm codebase. 7 imports, 1 dependents.

File typescript DrizzleORM DatabaseDrivers 7 imports 1 dependents 1 functions 2 classes

Entity Profile

DrizzleORM→ DatabaseDrivers→ vector.ts — drizzle-orm Source File

Dependency Diagram

graph LR
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18["vector.ts"]
  f4f42b4c_8610_03dd_fe01_232098668127["common.ts"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> f4f42b4c_8610_03dd_fe01_232098668127
  099d7c6d_2fc4_7b20_40e3_1b8af4138fe6["PgColumn"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> 099d7c6d_2fc4_7b20_40e3_1b8af4138fe6
  2dc784c6_95e9_4e28_ec81_7caf4acbd426["column-builder.ts"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> 2dc784c6_95e9_4e28_ec81_7caf4acbd426
  05f0a280_d0c9_693a_a4bf_83cc671012d2["column.ts"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> 05f0a280_d0c9_693a_a4bf_83cc671012d2
  27705a9d_afe9_57dd_8c97_e52d8a67d426["entity.ts"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> 27705a9d_afe9_57dd_8c97_e52d8a67d426
  17418917_ff81_c4e5_90f4_f37d557e5d51["table.ts"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> 17418917_ff81_c4e5_90f4_f37d557e5d51
  ecce3253_1e75_a87f_27b3_ca87e81a3024["utils.ts"]
  9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 --> ecce3253_1e75_a87f_27b3_ca87e81a3024
  b38cf654_48d6_41dc_6c0f_203697984d98["all.ts"]
  b38cf654_48d6_41dc_6c0f_203697984d98 --> 9ce71e3e_4724_d5ff_7d18_e5e4ce850b18
  style 9ce71e3e_4724_d5ff_7d18_e5e4ce850b18 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

import type { ColumnBuilderBaseConfig, ColumnBuilderRuntimeConfig, MakeColumnConfig } from '~/column-builder.ts';
import type { ColumnBaseConfig } from '~/column.ts';
import { entityKind } from '~/entity.ts';
import type { AnyPgTable } from '~/pg-core/table.ts';
import { getColumnNameAndConfig } from '~/utils.ts';
import { PgColumn, PgColumnBuilder } from '../common.ts';

export type PgVectorBuilderInitial<TName extends string, TDimensions extends number> = PgVectorBuilder<{
	name: TName;
	dataType: 'array';
	columnType: 'PgVector';
	data: number[];
	driverParam: string;
	enumValues: undefined;
	dimensions: TDimensions;
}>;

export class PgVectorBuilder<T extends ColumnBuilderBaseConfig<'array', 'PgVector'> & { dimensions: number }>
	extends PgColumnBuilder<
		T,
		{ dimensions: T['dimensions'] },
		{ dimensions: T['dimensions'] }
	>
{
	static override readonly [entityKind]: string = 'PgVectorBuilder';

	constructor(name: string, config: PgVectorConfig<T['dimensions']>) {
		super(name, 'array', 'PgVector');
		this.config.dimensions = config.dimensions;
	}

	/** @internal */
	override build<TTableName extends string>(
		table: AnyPgTable<{ name: TTableName }>,
	): PgVector<MakeColumnConfig<T, TTableName> & { dimensions: T['dimensions'] }> {
		return new PgVector<MakeColumnConfig<T, TTableName> & { dimensions: T['dimensions'] }>(
			table,
			this.config as ColumnBuilderRuntimeConfig<any, any>,
		);
	}
}

export class PgVector<T extends ColumnBaseConfig<'array', 'PgVector'> & { dimensions: number | undefined }>
	extends PgColumn<T, { dimensions: T['dimensions'] }, { dimensions: T['dimensions'] }>
{
	static override readonly [entityKind]: string = 'PgVector';

	readonly dimensions: T['dimensions'] = this.config.dimensions;

	getSQLType(): string {
		return `vector(${this.dimensions})`;
	}

	override mapToDriverValue(value: unknown): unknown {
		return JSON.stringify(value);
	}

	override mapFromDriverValue(value: string): unknown {
		return value
			.slice(1, -1)
			.split(',')
			.map((v) => Number.parseFloat(v));
	}
}

export interface PgVectorConfig<TDimensions extends number = number> {
	dimensions: TDimensions;
}

export function vector<D extends number>(
	config: PgVectorConfig<D>,
): PgVectorBuilderInitial<'', D>;
export function vector<TName extends string, D extends number>(
	name: TName,
	config: PgVectorConfig<D>,
): PgVectorBuilderInitial<TName, D>;
export function vector(a: string | PgVectorConfig, b?: PgVectorConfig) {
	const { name, config } = getColumnNameAndConfig<PgVectorConfig>(a, b);
	return new PgVectorBuilder(name, config);
}

Domain

DrizzleORM

Subdomains

DatabaseDrivers

Functions

vector()

Classes

Types

Dependencies

PgColumn
column-builder.ts
column.ts
common.ts
entity.ts
table.ts
utils.ts

Imported By

drizzle-orm/src/pg-core/columns/all.ts

Source

View on GitHub

Frequently Asked Questions

What does vector.ts do?

vector.ts is a source file in the drizzle-orm codebase, written in typescript. It belongs to the DrizzleORM domain, DatabaseDrivers subdomain.

What functions are defined in vector.ts?

vector.ts defines 1 function(s): vector.

What does vector.ts depend on?

vector.ts imports 7 module(s): PgColumn, column-builder.ts, column.ts, common.ts, entity.ts, table.ts, utils.ts.

What files import vector.ts?

vector.ts is imported by 1 file(s): all.ts.

Where is vector.ts in the architecture?

vector.ts is located at drizzle-orm/src/pg-core/columns/vector_extension/vector.ts (domain: DrizzleORM, subdomain: DatabaseDrivers, directory: drizzle-orm/src/pg-core/columns/vector_extension).

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free