Skip to content

Commit b36eaf3

Browse files
authored
Merge pull request #49 from modelstudioai/feat/knowledge-api-key
fix(core): 修复 Rerank 字段类型用于请求体中
2 parents abe29d1 + d20eea5 commit b36eaf3

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

packages/cli/src/commands/knowledge/retrieve.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ async function runWithAkSk(
220220
};
221221
if (flags.rerankMode) rerank.RerankMode = flags.rerankMode as string;
222222
if (flags.rerankInstruct) rerank.RerankInstruct = flags.rerankInstruct as string;
223-
body.Rerank = rerank;
223+
body.Rerank = [rerank];
224224
}
225225

226226
const pathname = `/${workspaceId}/index/retrieve`;

packages/core/src/types/api.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -363,11 +363,11 @@ export interface KnowledgeRetrieveRequest {
363363
EnableRewrite?: boolean;
364364
RerankTopN?: number;
365365
TopK?: number;
366-
Rerank?: {
366+
Rerank?: Array<{
367367
ModelName?: string;
368368
RerankMode?: string;
369369
RerankInstruct?: string;
370-
};
370+
}>;
371371
RerankTopN_legacy?: number;
372372
SearchFilters?: Array<{
373373
Key: string;

0 commit comments

Comments
 (0)