Skip to content

Commit 1f6aa1a

Browse files
authored
[ML] Enable DeepSeek Completion (#124665)
Enable streaming Completion as well as Chat Completion
1 parent 9e754ec commit 1f6aa1a

File tree

1 file changed

+2
-1
lines changed
  • x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/deepseek

1 file changed

+2
-1
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/deepseek/DeepSeekService.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ public class DeepSeekService extends SenderService {
5757
TaskType.COMPLETION,
5858
TaskType.CHAT_COMPLETION
5959
);
60+
private static final EnumSet<TaskType> SUPPORTED_TASK_TYPES_FOR_STREAMING = EnumSet.of(TaskType.COMPLETION, TaskType.CHAT_COMPLETION);
6061

6162
public DeepSeekService(HttpRequestSender.Factory factory, ServiceComponents serviceComponents) {
6263
super(factory, serviceComponents);
@@ -169,7 +170,7 @@ public TransportVersion getMinimalSupportedVersion() {
169170

170171
@Override
171172
public Set<TaskType> supportedStreamingTasks() {
172-
return EnumSet.of(TaskType.CHAT_COMPLETION);
173+
return SUPPORTED_TASK_TYPES_FOR_STREAMING;
173174
}
174175

175176
@Override

0 commit comments

Comments
 (0)