Skip to content

Commit 6231b28

Browse files
committed
fix accuracy tests
1 parent f934235 commit 6231b28

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tests/accuracy/createIndex.test.ts

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ describeAccuracyTests([
4747
],
4848
},
4949
{
50-
prompt: "Create a vector search index on 'mydb.movies' namespace with on the 'plotSummary' field. The index should use 1024 dimensions.",
50+
prompt: "Create a vector search index on 'mydb.movies' namespace on the 'plotSummary' field. The index should use 1024 dimensions.",
5151
expectedToolCalls: [
5252
{
5353
toolName: "create-index",
@@ -64,10 +64,6 @@ describeAccuracyTests([
6464
path: "plotSummary",
6565
numDimensions: 1024,
6666
},
67-
{
68-
type: "filter",
69-
path: "releaseDate",
70-
},
7167
],
7268
},
7369
],
@@ -76,7 +72,7 @@ describeAccuracyTests([
7672
],
7773
},
7874
{
79-
prompt: "Create a vector search index on 'mydb.movies' namespace with on the 'plotSummary' field and 'genre' field. Pick a sensible number of dimensions for a voyage 3.5 model.",
75+
prompt: "Create a vector search index on 'mydb.movies' namespace with on the 'plotSummary' field and 'genre' field, both of which contain vector embeddings. Pick a sensible number of dimensions for a voyage 3.5 model.",
8076
expectedToolCalls: [
8177
{
8278
toolName: "create-index",
@@ -94,13 +90,15 @@ describeAccuracyTests([
9490
numDimensions: Matcher.number(
9591
(value) => value % 2 === 0 && value >= 256 && value <= 8192
9692
),
93+
similarity: Matcher.anyOf(Matcher.undefined, Matcher.string()),
9794
},
9895
{
9996
type: "vector",
10097
path: "genre",
10198
numDimensions: Matcher.number(
10299
(value) => value % 2 === 0 && value >= 256 && value <= 8192
103100
),
101+
similarity: Matcher.anyOf(Matcher.undefined, Matcher.string()),
104102
},
105103
],
106104
},

0 commit comments

Comments
 (0)