Skip to content

Commit ad1f62a

Browse files
bevzzzbevzzz
committed
feat: add nearVector overloads for float[][] input
1 parent 78701bf commit ad1f62a

3 files changed

Lines changed: 196 additions & 0 deletions

File tree

‎src/main/java/io/weaviate/client6/v1/api/collections/aggregate/AbstractAggregateClient.java‎

Lines changed: 71 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -305,6 +305,38 @@ public ResponseT nearVector(float[] vector, Function<NearVector.Builder, ObjectB
305305
return nearVector(NearVector.of(Target.vector(vector), nv), fn);
306306
}
307307

308+
/**
309+
* Aggregate results of a near vector query.
310+
*
311+
* @param vector Query vector.
312+
* @param fn Lambda expression for optional aggregation parameters.
313+
* @return Aggregation result.
314+
*
315+
* @throws WeaviateApiException in case the server returned with an
316+
* error status code.
317+
* @see AggregateResponse
318+
*/
319+
public ResponseT nearVector(float[][] vector, Function<Aggregation.Builder, ObjectBuilder<Aggregation>> fn) {
320+
return nearVector(NearVector.of(Target.vector(vector)), fn);
321+
}
322+
323+
/**
324+
* Aggregate results of a near vector query.
325+
*
326+
* @param vector Query vector.
327+
* @param nv Lambda expression for optional near vector parameters.
328+
* @param fn Lambda expression for optional aggregation parameters.
329+
* @return Aggregation result.
330+
*
331+
* @throws WeaviateApiException in case the server returned with an
332+
* error status code.
333+
* @see AggregateResponse
334+
*/
335+
public ResponseT nearVector(float[][] vector, Function<NearVector.Builder, ObjectBuilder<NearVector>> nv,
336+
Function<Aggregation.Builder, ObjectBuilder<Aggregation>> fn) {
337+
return nearVector(NearVector.of(Target.vector(vector), nv), fn);
338+
}
339+
308340
/**
309341
* Aggregate results of a near vector query.
310342
*
@@ -359,6 +391,45 @@ public GroupedResponseT nearVector(float[] vector, Function<NearVector.Builder,
359391
return nearVector(NearVector.of(Target.vector(vector), nv), fn, groupBy);
360392
}
361393

394+
/**
395+
* Aggregate results of a near vector query.
396+
*
397+
* @param vector Query vector.
398+
* @param fn Lambda expression for optional aggregation parameters.
399+
* @param groupBy GroupBy clause.
400+
* @return Grouped aggregation result.
401+
*
402+
* @throws WeaviateApiException in case the server returned with an
403+
* error status code.
404+
*
405+
* @see GroupBy
406+
* @see AggregateResponseGrouped
407+
*/
408+
public GroupedResponseT nearVector(float[][] vector, Function<Aggregation.Builder, ObjectBuilder<Aggregation>> fn,
409+
GroupBy groupBy) {
410+
return nearVector(NearVector.of(Target.vector(vector)), fn, groupBy);
411+
}
412+
413+
/**
414+
* Aggregate results of a near vector query.
415+
*
416+
* @param vector Query vector.
417+
* @param nv Lambda expression for optional near vector parameters.
418+
* @param fn Lambda expression for optional aggregation parameters.
419+
* @param groupBy GroupBy clause.
420+
* @return Grouped aggregation result.
421+
*
422+
* @throws WeaviateApiException in case the server returned with an
423+
* error status code.
424+
*
425+
* @see GroupBy
426+
* @see AggregateResponseGrouped
427+
*/
428+
public GroupedResponseT nearVector(float[][] vector, Function<NearVector.Builder, ObjectBuilder<NearVector>> nv,
429+
Function<Aggregation.Builder, ObjectBuilder<Aggregation>> fn, GroupBy groupBy) {
430+
return nearVector(NearVector.of(Target.vector(vector), nv), fn, groupBy);
431+
}
432+
362433
/**
363434
* Aggregate results of a near vector query.
364435
*

‎src/main/java/io/weaviate/client6/v1/api/collections/generate/AbstractGenerateClient.java‎

Lines changed: 68 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -419,6 +419,34 @@ public ResponseT nearVector(float[] vector,
419419
return nearVector(Target.vector(vector), fn, generateFn);
420420
}
421421

422+
/**
423+
* Run a generative task on the results of a near vector search.
424+
*
425+
* @param vector Query vector.
426+
* @param generateFn Lambda expression for generative task parameters.
427+
* @throws WeaviateApiException in case the server returned with an
428+
* error status code.
429+
*/
430+
public ResponseT nearVector(float[][] vector,
431+
Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn) {
432+
return nearVector(Target.vector(vector), generateFn);
433+
}
434+
435+
/**
436+
* Run a generative task on the results of a near vector search.
437+
*
438+
* @param vector Query vector.
439+
* @param fn Lambda expression for optional search parameters.
440+
* @param generateFn Lambda expression for generative task parameters.
441+
* @throws WeaviateApiException in case the server returned with an
442+
* error status code.
443+
*/
444+
public ResponseT nearVector(float[][] vector,
445+
Function<NearVector.Builder, ObjectBuilder<NearVector>> fn,
446+
Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn) {
447+
return nearVector(Target.vector(vector), fn, generateFn);
448+
}
449+
422450
/**
423451
* Run a generative task on the results of a near vector search.
424452
*
@@ -499,6 +527,46 @@ public GroupedResponseT nearVector(float[] vector,
499527
return nearVector(Target.vector(vector), fn, generateFn, groupBy);
500528
}
501529

530+
/**
531+
* Run a generative task on the results of a near vector search.
532+
*
533+
* @param vector Query vector.
534+
* @param generateFn Lambda expression for generative task parameters.
535+
* @param groupBy Group-by clause.
536+
* @return Grouped query result.
537+
* @throws WeaviateApiException in case the server returned with an
538+
* error status code.
539+
*
540+
* @see GroupBy
541+
* @see GenerativeResponseGrouped
542+
*/
543+
public GroupedResponseT nearVector(float[][] vector,
544+
Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn,
545+
GroupBy groupBy) {
546+
return nearVector(Target.vector(vector), generateFn, groupBy);
547+
}
548+
549+
/**
550+
* Run a generative task on the results of a near vector search.
551+
*
552+
* @param vector Query vector.
553+
* @param fn Lambda expression for optional search parameters.
554+
* @param generateFn Lambda expression for generative task parameters.
555+
* @param groupBy Group-by clause.
556+
* @return Grouped query result.
557+
* @throws WeaviateApiException in case the server returned with an
558+
* error status code.
559+
*
560+
* @see GroupBy
561+
* @see GenerativeResponseGrouped
562+
*/
563+
public GroupedResponseT nearVector(float[][] vector,
564+
Function<NearVector.Builder, ObjectBuilder<NearVector>> fn,
565+
Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn,
566+
GroupBy groupBy) {
567+
return nearVector(Target.vector(vector), fn, generateFn, groupBy);
568+
}
569+
502570
/**
503571
* Run a generative task on the results of a near vector search.
504572
*

‎src/main/java/io/weaviate/client6/v1/api/collections/query/AbstractQueryClient.java‎

Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -415,6 +415,29 @@ public ResponseT nearVector(float[] vector, Function<NearVector.Builder, ObjectB
415415
return nearVector(Target.vector(vector), fn);
416416
}
417417

418+
/**
419+
* Query collection objects using near vector search.
420+
*
421+
* @param vector Query vector.
422+
* @throws WeaviateApiException in case the server returned with an
423+
* error status code.
424+
*/
425+
public ResponseT nearVector(float[][] vector) {
426+
return nearVector(Target.vector(vector));
427+
}
428+
429+
/**
430+
* Query collection objects using near vector search.
431+
*
432+
* @param vector Query vector.
433+
* @param fn Lambda expression for optional parameters.
434+
* @throws WeaviateApiException in case the server returned with an
435+
* error status code.
436+
*/
437+
public ResponseT nearVector(float[][] vector, Function<NearVector.Builder, ObjectBuilder<NearVector>> fn) {
438+
return nearVector(Target.vector(vector), fn);
439+
}
440+
418441
/**
419442
* Query collection objects using near vector search.
420443
*
@@ -484,6 +507,40 @@ public GroupedResponseT nearVector(float[] vector, Function<NearVector.Builder,
484507
return nearVector(Target.vector(vector), fn, groupBy);
485508
}
486509

510+
/**
511+
* Query collection objects using near vector search.
512+
*
513+
* @param vector Query vector.
514+
* @param groupBy Group-by clause.
515+
* @return Grouped query result.
516+
* @throws WeaviateApiException in case the server returned with an
517+
* error status code.
518+
*
519+
* @see GroupBy
520+
* @see QueryResponseGrouped
521+
*/
522+
public GroupedResponseT nearVector(float[][] vector, GroupBy groupBy) {
523+
return nearVector(Target.vector(vector), groupBy);
524+
}
525+
526+
/**
527+
* Query collection objects using near vector search.
528+
*
529+
* @param vector Query vector.
530+
* @param fn Lambda expression for optional parameters.
531+
* @param groupBy Group-by clause.
532+
* @return Grouped query result.
533+
* @throws WeaviateApiException in case the server returned with an
534+
* error status code.
535+
*
536+
* @see GroupBy
537+
* @see QueryResponseGrouped
538+
*/
539+
public GroupedResponseT nearVector(float[][] vector, Function<NearVector.Builder, ObjectBuilder<NearVector>> fn,
540+
GroupBy groupBy) {
541+
return nearVector(Target.vector(vector), fn, groupBy);
542+
}
543+
487544
/**
488545
* Query collection objects using near vector search.
489546
*

0 commit comments

Comments
 (0)