Merge pull request #75 from weaviate/add-autocut-support

Add support for autocut

Author: antas-marcin

src/graphql/getter.test.ts

@@ -75,6 +75,27 @@ test('a simple query with a group', () => {
   expect(mockClient.query).toHaveBeenCalledWith(expectedQuery);
 });
 
+test('a simple query with autocut', () => {
+  const mockClient: any = {
+    query: jest.fn(),
+  };
+
+  const expectedQuery = `{Get{Person(where:{operator:Equal,valueText:"hawaii",path:["name"]},autocut:10){name}}}`;
+
+  new Getter(mockClient)
+    .withClassName('Person')
+    .withAutocut(10)
+    .withWhere({
+      operator: 'Equal',
+      valueText: 'hawaii',
+      path: ['name'],
+    })
+    .withFields('name')
+    .do();
+
+  expect(mockClient.query).toHaveBeenCalledWith(expectedQuery);
+});
+
 describe('query with consistency level', () => {
   test('One', () => {
     const mockClient: any = {

src/graphql/getter.ts

@@ -37,6 +37,7 @@ export default class GraphQLGetter extends CommandBase {
   private consistencyLevel?: ConsistencyLevel;
   private groupByString?: string;
   private tenant?: string;
+  private autocut?: number;
 
   constructor(client: Connection) {
     super(client);
@@ -172,6 +173,11 @@ export default class GraphQLGetter extends CommandBase {
     return this;
   };
 
+  withAutocut = (autocut: number) => {
+    this.autocut = autocut;
+    return this;
+  };
+
   withSort = (args: SortArgs[]) => {
     this.sortString = new Sort(args).toString();
     return this;
@@ -265,6 +271,10 @@ export default class GraphQLGetter extends CommandBase {
       args = [...args, `offset:${this.offset}`];
     }
 
+    if (this.autocut) {
+      args = [...args, `autocut:${this.autocut}`];
+    }
+
     if (this.sortString) {
       args = [...args, `sort:[${this.sortString}]`];
     }

src/graphql/journey.test.ts

@@ -311,6 +311,22 @@ describe('the graphql journey', () => {
       });
   });
 
+  test('graphql get nearText with autocut', () => {
+    return client.graphql
+      .get()
+      .withClassName('Article')
+      .withNearText({ concepts: ['Article'] })
+      .withAutocut(3)
+      .withFields('_additional { id }')
+      .do()
+      .then((res: any) => {
+        expect(res.data.Get.Article.length).toBe(3);
+      })
+      .catch((e: any) => {
+        throw new Error('it should not have errord' + e);
+      });
+  });
+
   test('graphql get hybrid with query (no vector, alpha 0)', () => {
     return client.graphql
       .get()