@@ -26,7 +26,7 @@ public class Embeddings {
26
26
public interface Remote {
27
27
@ GET ("search" )
28
28
Call <List <SearchResult >> search (@ Query ("query" ) String query , @ Query ("limit" ) int limit ,
29
- @ Query ("weights" ) float weights , @ Query ("index" ) String index );
29
+ @ Query ("weights" ) Float weights , @ Query ("index" ) String index );
30
30
31
31
@ POST ("batchsearch" )
32
32
Call <List <List <SearchResult >>> batchsearch (@ Body HashMap params );
@@ -116,7 +116,7 @@ public Embeddings(String url) {
116
116
* @param index index name, if applicable
117
117
* @return list of {id: value, score: value}
118
118
*/
119
- public List <SearchResult > search (String query , int limit , float weights , String index ) throws IOException {
119
+ public List <SearchResult > search (String query , int limit , Float weights , String index ) throws IOException {
120
120
return this .api .search (query , limit , weights , index ).execute ().body ();
121
121
}
122
122
@@ -131,7 +131,7 @@ public List<SearchResult> search(String query, int limit, float weights, String
131
131
* @param index index name, if applicable
132
132
* @return list of {id: value, score: value} per query
133
133
*/
134
- public List <List <SearchResult >> batchsearch (List <String > queries , int limit , float weights , String index ) throws IOException {
134
+ public List <List <SearchResult >> batchsearch (List <String > queries , int limit , Float weights , String index ) throws IOException {
135
135
// Post parameters
136
136
HashMap <String , Object > params = new HashMap <String , Object >();
137
137
params .put ("queries" , queries );
0 commit comments