1656

pending completion

Build # 1656

Build Type

Pull #467

travis-ci-com

Committed by

web-flow

Commit Message

Merge e6053a9fc into 2d0f5eef9

Pull Request Pull Request #467: update kf-server json parse

Run Details

10812 of 23249 branches covered (46.51%)

Branch coverage included in aggregate %.

0 of 1 new or added line in 1 file covered. (0.0%)

20 existing lines in 8 files now uncovered.

25393 of 43938 relevant lines covered (57.79%)

528331.83 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

82.13

/languages/thingtalk/dialogue_acts/coref-questions.ts

// -*- mode: typescript; indent-tabs-mode: nil; js-basic-offset: 4 -*-
//
// This file is part of Genie
//
// Copyright 2020 The Board of Trustees of the Leland Stanford Junior University
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Author: Giovanni Campagna <gcampagn@cs.stanford.edu>


import assert from 'assert';

import { Ast, } from 'thingtalk';

import * as C from '../ast_manip';
import {
    ContextInfo,
    addQuery,
} from '../state_manip';
import {
    queryRefinement,
    refineFilterToAnswerQuestion,
} from './refinement-helpers';
import type { Recommendation } from './recommendation';
import type { ListProposal } from './list-proposal';

export type UserSearchQuestionForList = [Ast.EntityValue|Ast.NumberValue|null, C.ParamSlot[]];

export function userSearchQuestionForListKeyFn([name, questions] : UserSearchQuestionForList) {
    if (questions.length === 0)
        return { idType: null, functionName: null };

    return {
        idType: name ? name.getType() : null,
        functionName: questions[0].schema.qualifiedName
    };
}

function areQuestionsValidForContext(ctx : ContextInfo, questions : C.ParamSlot[]) {
    if (ctx.resultInfo!.isAggregation)
        return null;

    const schema = ctx.currentFunction!;

    // if the function only contains one parameter, do not generate projection for it
    if (Object.keys(schema.out).length === 1)
        return null;

    for (const q of questions) {
        if (!C.isSameFunction(schema, q.schema))
            return false;
        const arg = schema.getArgument(q.name);
        if (!arg || arg.is_input)
            return false;
    }
    return true;
}

function recommendationSearchQuestionReply(ctx : ContextInfo, questions : C.ParamSlot[]) {
    const proposal = ctx.aux as Recommendation;
    const { topResult, info, } = proposal;
    if (!topResult.value.id)
        return null;
    if (info !== null) {
        for (const q of questions) {
            if (info.has(q.name))
                return null;
        }
    }

    if (!areQuestionsValidForContext(ctx, questions))
        return null;

    const currentStmt = ctx.current!.stmt;
    const currentTable = currentStmt.expression;
    const newFilter = new Ast.BooleanExpression.Atom(null, 'id', '==', topResult.value.id);
    const newTable = queryRefinement(currentTable, newFilter, refineFilterToAnswerQuestion,
        questions.map((q) => q.name));
    if (newTable === null)
        return null;
    return addQuery(ctx, 'execute', newTable, 'accepted');
}

function learnMoreSearchQuestionReply(ctx : ContextInfo, questions : C.ParamSlot[]) {
    const topResult = ctx.results![0];
    if (!areQuestionsValidForContext(ctx, questions))
        return null;

    const currentStmt = ctx.current!.stmt;
    const currentTable = currentStmt.expression;
    if (!topResult.value.id)
        return null;
    const newFilter = new Ast.BooleanExpression.Atom(null, 'id', '==', topResult.value.id);
    const newTable = queryRefinement(currentTable, newFilter, refineFilterToAnswerQuestion,
        questions.map((q) => q.name));
    if (newTable === null)
        return null;
    return addQuery(ctx, 'execute', newTable, 'accepted');
}

function displayResultSearchQuestionReply(ctx : ContextInfo, questions : C.ParamSlot[]) {
    if (!areQuestionsValidForContext(ctx, questions))
        return null;

    const currentStmt = ctx.current!.stmt;
    const currentTable = currentStmt.expression;
    const newTable = queryRefinement(currentTable, null, refineFilterToAnswerQuestion,
        questions.map((q) => q.name));
    if (newTable === null)
        return null;
    return addQuery(ctx, 'execute', newTable, 'accepted');
}

function listProposalSearchQuestionReply(ctx : ContextInfo, [name, questions] : [Ast.Value|null, C.ParamSlot[]]) {
    const proposal = ctx.aux as ListProposal;
    const [results, info] = proposal;

    if (name !== null) {
        let good = false;
        for (const result of results) {
            if (!result.value.id)
                continue;
            if (result.value.id.equals(name)) {
                good = true;
                break;
            }
        }
        if (!good)
            return null;
    }

    if (info !== null) {
        for (const q of questions) {
            if (info.has(q.name))
                return null;
        }
    }

    if (!areQuestionsValidForContext(ctx, questions))
        return null;

    const currentStmt = ctx.current!.stmt;
    const currentTable = currentStmt.expression;
    let newTable;
    if (name !== null) {
        const newFilter = new Ast.BooleanExpression.Atom(null, 'id', '==', name);
        newTable = queryRefinement(currentTable, newFilter, refineFilterToAnswerQuestion,
            questions.map((q) => q.name));
    } else {
        newTable = queryRefinement(currentTable, null, null,
            questions.map((q) => q.name));
    }
    if (newTable === null)
        return null;

    return addQuery(ctx, 'execute', newTable, 'accepted');
}

function corefConstant(ctx : ContextInfo, base : Ast.Expression, param : C.ParamSlot) {
    const previous = ctx.previousDomain;
    assert(previous);
    if (!previous.results || previous.results.results.length === 0)
        return null;
    const ctxStmt = previous.stmt;
    const ctxSchema = ctxStmt.expression.schema!;
    if (!ctxSchema.class) // FIXME not sure how this happens...
        return null;
    if (ctxSchema.class.name !== base.schema!.class!.name)
        return null;
    if (!C.isSameFunction(ctxSchema, param.schema))
        return null;
    const result = previous.results.results[0];
    if (!result.value[param.name])
        return null;
    return result.value[param.name];
}

function booleanQuestion(base : Ast.Expression|null, slot : C.FilterSlot) : C.ParamSlot[]|null {
    if (base !== null) {
        if (!C.isSameFunction(base.schema!, slot.schema))
            return null;
    }

    const ast = slot.ast;
    if (!(ast instanceof Ast.AtomBooleanExpression)) {
        assert(ast instanceof Ast.AndBooleanExpression);
        assert(ast.operands.length === 2);
        const [op1, op2] = [ast.operands[0], ast.operands[1]];
        assert(op1 instanceof Ast.AtomBooleanExpression);
        assert(op2 instanceof Ast.AtomBooleanExpression);
        assert(op1.name === op2.name);
        return [{
            schema: slot.schema,
            type: slot.ptype,
            filterable: slot.schema.getArgument(op1.name)!.getImplementationAnnotation<boolean>('filterable') ?? true,
            name: op1.name,
            ast: new Ast.Value.VarRef(op1.name)
        }];
    }
    if (ast.name === 'id')
        return null;
    return [{
        schema: slot.schema,
        type: slot.ptype,
        filterable: slot.schema.getArgument(ast.name)!.getImplementationAnnotation<boolean>('filterable') ?? true,
        name: ast.name,
        ast: new Ast.Value.VarRef(ast.name),
    }];
}

export {
    recommendationSearchQuestionReply,
    displayResultSearchQuestionReply,
    learnMoreSearchQuestionReply,
    listProposalSearchQuestionReply,
    corefConstant,
    booleanQuestion
};

1	// -- mode: typescript; indent-tabs-mode: nil; js-basic-offset: 4 --
2	//
3	// This file is part of Genie
4	//
5	// Copyright 2020 The Board of Trustees of the Leland Stanford Junior University
6	//
7	// Licensed under the Apache License, Version 2.0 (the "License");
8	// you may not use this file except in compliance with the License.
9	// You may obtain a copy of the License at
10	//
11	// http://www.apache.org/licenses/LICENSE-2.0
12	//
13	// Unless required by applicable law or agreed to in writing, software
14	// distributed under the License is distributed on an "AS IS" BASIS,
15	// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16	// See the License for the specific language governing permissions and
17	// limitations under the License.
18	//
19	// Author: Giovanni Campagna <gcampagn@cs.stanford.edu>
20
21
22	import assert from 'assert';	2✔
23
24	import { Ast, } from 'thingtalk';	2✔
25
26	import * as C from '../ast_manip';	2✔
27	import {	2✔
28	ContextInfo,
29	addQuery,
30	} from '../state_manip';
31	import {	2✔
32	queryRefinement,
33	refineFilterToAnswerQuestion,
34	} from './refinement-helpers';
35	import type { Recommendation } from './recommendation';
36	import type { ListProposal } from './list-proposal';
37
38	export type UserSearchQuestionForList = [Ast.EntityValue\|Ast.NumberValue\|null, C.ParamSlot[]];
39
40	export function userSearchQuestionForListKeyFn([name, questions] : UserSearchQuestionForList) {	2✔
41	if (questions.length === 0)	23,896!
42	return { idType: null, functionName: null };	×
43
44	return {	23,896✔
45	idType: name ? name.getType() : null,	23,896✔
46	functionName: questions[0].schema.qualifiedName
47	};
48	}
49
50	function areQuestionsValidForContext(ctx : ContextInfo, questions : C.ParamSlot[]) {
51	if (ctx.resultInfo!.isAggregation)	37,126!
52	return null;	×
53
54	const schema = ctx.currentFunction!;	37,126✔
55
56	// if the function only contains one parameter, do not generate projection for it
57	if (Object.keys(schema.out).length === 1)	37,126✔
58	return null;	131✔
59
60	for (const q of questions) {	36,995✔
61	if (!C.isSameFunction(schema, q.schema))	44,926!
62	return false;	×
63	const arg = schema.getArgument(q.name);	44,926✔
64	if (!arg \|\| arg.is_input)	44,926!
65	return false;	×
66	}
67	return true;	36,995✔
68	}
69
70	function recommendationSearchQuestionReply(ctx : ContextInfo, questions : C.ParamSlot[]) {
71	const proposal = ctx.aux as Recommendation;	42,217✔
72	const { topResult, info, } = proposal;	42,217✔
73	if (!topResult.value.id)	42,217!
UNCOV 74	return null;	×
75	if (info !== null) {	42,217✔
76	for (const q of questions) {	39,383✔
77	if (info.has(q.name))	41,359✔
78	return null;	29,154✔
79	}
80	}
81
82	if (!areQuestionsValidForContext(ctx, questions))	13,063!
83	return null;	×
84
85	const currentStmt = ctx.current!.stmt;	13,063✔
86	const currentTable = currentStmt.expression;	13,063✔
87	const newFilter = new Ast.BooleanExpression.Atom(null, 'id', '==', topResult.value.id);	13,063✔
88	const newTable = queryRefinement(currentTable, newFilter, refineFilterToAnswerQuestion,	13,063✔
89	questions.map((q) => q.name));	15,542✔
90	if (newTable === null)	13,063!
91	return null;	×
92	return addQuery(ctx, 'execute', newTable, 'accepted');	13,063✔
93	}
94
95	function learnMoreSearchQuestionReply(ctx : ContextInfo, questions : C.ParamSlot[]) {
96	const topResult = ctx.results![0];	6,812✔
97	if (!areQuestionsValidForContext(ctx, questions))	6,812!
98	return null;	×
99
100	const currentStmt = ctx.current!.stmt;	6,812✔
101	const currentTable = currentStmt.expression;	6,812✔
102	if (!topResult.value.id)	6,812!
103	return null;	×
104	const newFilter = new Ast.BooleanExpression.Atom(null, 'id', '==', topResult.value.id);	6,812✔
105	const newTable = queryRefinement(currentTable, newFilter, refineFilterToAnswerQuestion,	6,812✔
106	questions.map((q) => q.name));	7,488✔
107	if (newTable === null)	6,812!
108	return null;	×
109	return addQuery(ctx, 'execute', newTable, 'accepted');	6,812✔
110	}
111
112	function displayResultSearchQuestionReply(ctx : ContextInfo, questions : C.ParamSlot[]) {
113	if (!areQuestionsValidForContext(ctx, questions))	2,487✔
114	return null;	131✔
115
116	const currentStmt = ctx.current!.stmt;	2,356✔
117	const currentTable = currentStmt.expression;	2,356✔
118	const newTable = queryRefinement(currentTable, null, refineFilterToAnswerQuestion,	2,356✔
119	questions.map((q) => q.name));	4,783✔
120	if (newTable === null)	2,356!
121	return null;	×
122	return addQuery(ctx, 'execute', newTable, 'accepted');	2,356✔
123	}
124
125	function listProposalSearchQuestionReply(ctx : ContextInfo, [name, questions] : [Ast.Value\|null, C.ParamSlot[]]) {
126	const proposal = ctx.aux as ListProposal;	28,215✔
127	const [results, info] = proposal;	28,215✔
128
129	if (name !== null) {	28,215✔
130	let good = false;	15,723✔
131	for (const result of results) {	15,723✔
132	if (!result.value.id)	37,813!
133	continue;	×
134	if (result.value.id.equals(name)) {	37,813✔
135	good = true;	2,686✔
136	break;	2,686✔
137	}
138	}
139	if (!good)	15,723✔
140	return null;	13,037✔
141	}
142
143	if (info !== null) {	15,178✔
144	for (const q of questions) {	4,569✔
145	if (info.has(q.name))	4,828✔
146	return null;	414✔
147	}
148	}
149
150	if (!areQuestionsValidForContext(ctx, questions))	14,764!
151	return null;	×
152
153	const currentStmt = ctx.current!.stmt;	14,764✔
154	const currentTable = currentStmt.expression;	14,764✔
155	let newTable;
156	if (name !== null) {	14,764✔
157	const newFilter = new Ast.BooleanExpression.Atom(null, 'id', '==', name);	2,641✔
158	newTable = queryRefinement(currentTable, newFilter, refineFilterToAnswerQuestion,	2,641✔
159	questions.map((q) => q.name));	3,440✔
160	} else {
161	newTable = queryRefinement(currentTable, null, null,	12,123✔
162	questions.map((q) => q.name));	13,673✔
163	}
164	if (newTable === null)	14,764!
165	return null;	×
166
167	return addQuery(ctx, 'execute', newTable, 'accepted');	14,764✔
168	}
169
170	function corefConstant(ctx : ContextInfo, base : Ast.Expression, param : C.ParamSlot) {
171	const previous = ctx.previousDomain;	4,851✔
172	assert(previous);	4,851✔
173	if (!previous.results \|\| previous.results.results.length === 0)	4,851✔
174	return null;	845✔
175	const ctxStmt = previous.stmt;	4,006✔
176	const ctxSchema = ctxStmt.expression.schema!;	4,006✔
177	if (!ctxSchema.class) // FIXME not sure how this happens...	4,006!
178	return null;	×
179	if (ctxSchema.class.name !== base.schema!.class!.name)	4,006✔
180	return null;	3,303✔
181	if (!C.isSameFunction(ctxSchema, param.schema))	703✔
182	return null;	553✔
183	const result = previous.results.results[0];	150✔
184	if (!result.value[param.name])	150!
185	return null;	×
186	return result.value[param.name];	150✔
187	}
188
189	function booleanQuestion(base : Ast.Expression\|null, slot : C.FilterSlot) : C.ParamSlot[]\|null {
190	if (base !== null) {	169,030✔
191	if (!C.isSameFunction(base.schema!, slot.schema))	109,184!
192	return null;	×
193	}
194
195	const ast = slot.ast;	169,030✔
196	if (!(ast instanceof Ast.AtomBooleanExpression)) {	169,030✔
197	assert(ast instanceof Ast.AndBooleanExpression);	94,171✔
198	assert(ast.operands.length === 2);	94,171✔
199	const [op1, op2] = [ast.operands[0], ast.operands[1]];	94,171✔
200	assert(op1 instanceof Ast.AtomBooleanExpression);	94,171✔
201	assert(op2 instanceof Ast.AtomBooleanExpression);	94,171✔
202	assert(op1.name === op2.name);	94,171✔
203	return [{	94,171✔
204	schema: slot.schema,
205	type: slot.ptype,
206	filterable: slot.schema.getArgument(op1.name)!.getImplementationAnnotation<boolean>('filterable') ?? true,	282,513!
207	name: op1.name,
208	ast: new Ast.Value.VarRef(op1.name)
209	}];
210	}
211	if (ast.name === 'id')	74,859!
212	return null;	×
213	return [{	74,859✔
214	schema: slot.schema,
215	type: slot.ptype,
216	filterable: slot.schema.getArgument(ast.name)!.getImplementationAnnotation<boolean>('filterable') ?? true,	224,577✔
217	name: ast.name,
218	ast: new Ast.Value.VarRef(ast.name),
219	}];
220	}
221
222	export {
223	recommendationSearchQuestionReply,	2✔
224	displayResultSearchQuestionReply,	2✔
225	learnMoreSearchQuestionReply,	2✔
226	listProposalSearchQuestionReply,	2✔
227	corefConstant,	2✔
228	booleanQuestion	2✔
229	};

stanford-oval / genie-toolkit / 1656

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous