Skip to content

Commit 1d30eba

Browse files
committed
Removing quantisation logic
1 parent f80fdbb commit 1d30eba

File tree

6 files changed

+8
-26
lines changed

6 files changed

+8
-26
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
2+
/*
3+
* Dropping old virtual VSS table(s).
4+
*/
5+
vector_quantize_cleanup('ml_training_snippets', 'embeddings');

backend/files/system/magic/ml_training_snippets-vss.get.hl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ data.connect:[generic|magic]
7474
.sql:@"
7575
select vss.distance, ts.id, ts.created, ts.type, ts.pushed, ts.uri, ts.prompt, ts.completion, ts.filename, ts.cached, ts.meta, ts.embeddings as embedding_vss
7676
from ml_training_snippets ts
77-
inner join vector_quantize_scan_stream('ml_training_snippets', 'embeddings', vector_as_f32(@embedding)) as vss on vss.rowid = ts.id
77+
inner join vector_full_scan_stream('ml_training_snippets', 'embeddings', vector_as_f32(@embedding)) as vss on vss.rowid = ts.id
7878
where 1"
7979

8080
// Further parametrising invocation if we should.

backend/files/system/openai/magic.startup/magic.ai.get-context.hl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ slots.create:magic.ai.get-context
164164
strings.concat
165165
.:@"
166166
select vss.distance, vss.rowid as id, ts.prompt, ts.completion, ts.uri, ts.cached
167-
from vector_quantize_scan_stream('ml_training_snippets', 'embeddings', vector_as_f32(@embedding)) as vss
167+
from vector_full_scan_stream('ml_training_snippets', 'embeddings', vector_as_f32(@embedding)) as vss
168168
inner join ml_training_snippets ts on ts.id = vss.rowid
169169
where
170170
ts.type = @type and

backend/files/system/openai/magic.startup/magic.ai.search.hl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ slots.create:magic.ai.search
8787
strings.concat
8888
.:@"
8989
select vss.distance, ts.prompt, ts.uri
90-
from vector_quantize_scan_stream('ml_training_snippets', 'embeddings', vector_as_f32(@embedding)) as vss
90+
from vector_full_scan_stream('ml_training_snippets', 'embeddings', vector_as_f32(@embedding)) as vss
9191
inner join ml_training_snippets ts on ts.id = vss.rowid
9292
where
9393
ts.type = @type and "

backend/files/system/openai/magic.startup/magic.ai.vectorise.hl

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -228,25 +228,6 @@ select id, prompt, completion
228228
message:Done creating embeddings for your model
229229
type:success
230230

231-
// Signaling frontend about quantisation process.
232-
sockets.signal:x:@.arguments/*/feedback-channel
233-
args
234-
message:Quantizing your type for faster retrieval
235-
type:info
236-
237-
// Quantizing type.
238-
data.connect:magic
239-
data.execute:@"
240-
select vector_init('ml_training_snippets', 'embeddings', 'dimension=1536,type=FLOAT32,distance=cosine');
241-
select vector_quantize('ml_training_snippets', 'embeddings', 'max_memory=100MB');
242-
select vector_quantize_preload('ml_training_snippets', 'embeddings')"
243-
244-
// Signaling frontend about quantisation process.
245-
sockets.signal:x:@.arguments/*/feedback-channel
246-
args
247-
message:Done quantizing your type
248-
type:info
249-
250231
// Signaling frontend.
251232
strings.concat
252233
.:"Done vectorising model '"

backend/files/system/openai/vectorise-snippet.post.hl

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,6 @@ data.connect:[generic|magic]
8080
data.execute:update ml_training_snippets set embeddings = vector_as_f32(@embeddings) where id = @id
8181
@embeddings:x:@strings.concat
8282
@id:x:@.arguments/*/id
83-
data.execute:@"
84-
select vector_init('ml_training_snippets', 'embeddings', 'dimension=1536,type=FLOAT32,distance=cosine');
85-
select vector_quantize('ml_training_snippets', 'embeddings', 'max_memory=100MB');
86-
select vector_quantize_preload('ml_training_snippets', 'embeddings')"
8783

8884
// Returning success to caller.
8985
return

0 commit comments

Comments
 (0)