Some small changes

lanterndata · Mar 21, 2024 · 2fd4111 · 2fd4111
1 parent 073fbbd
commit 2fd4111
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/typeorm/README.md b/src/typeorm/README.md
@@ -140,11 +140,10 @@ await bookRepository
   .getMany();
 ```
 
-Corresponding SQL code:
+Corresponding SQL code (example):
 
 ```sql
 SELECT * FROM "books"
-WHERE "url" IS NOT NULL
 ORDER BY "embedding" <-> image_embedding('clip/ViT-B-32-visual', "...") DESC
 LIMIT 2;
 ```
diff --git a/types/drizzle-orm/index.d.ts b/types/drizzle-orm/index.d.ts
@@ -9,8 +9,8 @@ export function createLanternExtrasExtension(): SQLWrapper;
 export function generateTextEmbedding(modelKey: TextEmbeddingModelType, value: string): PgColumn;
 export function generateImageEmbedding(modelKey: ImageEmbeddingModelType, value: string): PgColumn;
 
-export function textEmbedding(modelKey: TextEmbeddingModelType, column: string): string;
-export function imageEmbedding(modelKey: ImageEmbeddingModelType, column: string): string;
+export function textEmbedding(modelKey: TextEmbeddingModelType, column: string): PgColumn;
+export function imageEmbedding(modelKey: ImageEmbeddingModelType, column: string): PgColumn;
 
 export function l2Distance(column: Column, value: number[] | string): PgColumn;
 export function cosineDistance(column: Column, value: number[] | string): PgColumn;