Skip to content

Commit 7d23007

Browse files
committed
Add rerank task type specification
1 parent 379de4f commit 7d23007

File tree

2 files changed

+51
-38
lines changed

2 files changed

+51
-38
lines changed

output/schema/schema.json

Lines changed: 37 additions & 37 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

specification/inference/_types/CommonTypes.ts

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -700,13 +700,26 @@ export class EisServiceSettings {
700700
}
701701

702702
export enum EisTaskType {
703-
chat_completion
703+
chat_completion,
704+
rerank
704705
}
705706

706707
export enum EisServiceType {
707708
elastic
708709
}
709710

711+
export class EisTaskSettings {
712+
/**
713+
* For a `rerank` task, return doc text within the results.
714+
*/
715+
return_documents?: boolean
716+
717+
/**
718+
* For a `rerank` task, the number of the top N documents that should be returned.
719+
*/
720+
top_n?: integer
721+
}
722+
710723
export class ElasticsearchServiceSettings {
711724
/**
712725
* Adaptive allocations configuration details.

0 commit comments

Comments
 (0)