Skip to content

Commit d1c0601

Browse files
committed
fix: update Pinecone SDK rerank call to single-object API and format code
1 parent 2d4f136 commit d1c0601

2 files changed

Lines changed: 10 additions & 16 deletions

File tree

src/pinecone-client.ts

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -328,23 +328,21 @@ export class PineconeClient {
328328
const pc = this.ensureClient();
329329

330330
try {
331-
const rerankResult = await pc.inference.rerank(
332-
this.rerankModel,
331+
const rerankResult = await pc.inference.rerank({
332+
model: this.rerankModel,
333333
query,
334334
// The Pinecone SDK types constrain document values to `Record<string, string>`,
335335
// but the underlying HTTP API accepts any JSON value. We pass MergedHit objects
336336
// (metadata may contain number/boolean/string[]) and only `chunk_text` — which is
337337
// always a string — is accessed via rankFields. The double cast via `as unknown`
338338
// is intentional: it bypasses the SDK's over-narrow type without stringifying
339339
// metadata values that we need to read back from the returned documents.
340-
results as unknown as (string | Record<string, string>)[],
341-
{
342-
topN,
343-
rankFields: ['chunk_text'],
344-
returnDocuments: true,
345-
parameters: { truncate: 'END' },
346-
}
347-
);
340+
documents: results as unknown as (string | Record<string, string>)[],
341+
topN,
342+
rankFields: ['chunk_text'],
343+
returnDocuments: true,
344+
parameters: { truncate: 'END' },
345+
});
348346

349347
const reranked: SearchResult[] = [];
350348
for (const item of rerankResult.data || []) {

src/server/url-generation.test.ts

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,7 @@ describe('generateUrlForNamespace', () => {
3636
list_name: 'boost-users',
3737
doc_id: '12345',
3838
});
39-
expect(r.url).toBe(
40-
'https://lists.boost.org/archives/list/boost-users/message/12345/'
41-
);
39+
expect(r.url).toBe('https://lists.boost.org/archives/list/boost-users/message/12345/');
4240
expect(r.method).toBe('generated.mailing');
4341
});
4442

@@ -47,9 +45,7 @@ describe('generateUrlForNamespace', () => {
4745
list_name: 'boost-announce',
4846
msg_id: '67890',
4947
});
50-
expect(r.url).toBe(
51-
'https://lists.boost.org/archives/list/boost-announce/message/67890/'
52-
);
48+
expect(r.url).toBe('https://lists.boost.org/archives/list/boost-announce/message/67890/');
5349
expect(r.method).toBe('generated.mailing');
5450
});
5551

0 commit comments

Comments
 (0)