File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed 
x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change 1616import  org .elasticsearch .inference .SecretSettings ;
1717import  org .elasticsearch .inference .ServiceSettings ;
1818import  org .elasticsearch .inference .TaskSettings ;
19+ import  org .elasticsearch .inference .UnifiedCompletionRequest ;
1920import  org .elasticsearch .xpack .core .inference .results .ChatCompletionResults ;
2021import  org .elasticsearch .xpack .core .inference .results .ErrorChunkedInferenceResults ;
2122import  org .elasticsearch .xpack .core .inference .results .InferenceChunkedSparseEmbeddingResults ;
@@ -137,11 +138,18 @@ public static List<NamedWriteableRegistry.Entry> getNamedWriteables() {
137138        addEisNamedWriteables (namedWriteables );
138139        addAlibabaCloudSearchNamedWriteables (namedWriteables );
139140
141+         addUnifiedNamedWriteables (namedWriteables );
142+ 
140143        namedWriteables .addAll (StreamingTaskManager .namedWriteables ());
141144
142145        return  namedWriteables ;
143146    }
144147
148+     private  static  void  addUnifiedNamedWriteables (List <NamedWriteableRegistry .Entry > namedWriteables ) {
149+         var  writeables  = UnifiedCompletionRequest .getNamedWriteables ();
150+         namedWriteables .addAll (writeables );
151+     }
152+ 
145153    private  static  void  addAmazonBedrockNamedWriteables (List <NamedWriteableRegistry .Entry > namedWriteables ) {
146154        namedWriteables .add (
147155            new  NamedWriteableRegistry .Entry (
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments