1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# pylint: disable=line-too-long,useless-suppression
# coding=utf-8
# -------------------------------------------------------------------------
# Inference async tests: text records querying (authoring removed)
# -------------------------------------------------------------------------
import pytest
from azure.ai.language.questionanswering.aio import QuestionAnsweringClient
from azure.ai.language.questionanswering.models import AnswersFromTextOptions, TextDocument
from azure.core.credentials import AzureKeyCredential
from testcase import QuestionAnsweringTestCase
class TestQueryTextAsync(QuestionAnsweringTestCase):
@pytest.mark.asyncio
async def test_query_text_basic(self, recorded_test, qna_creds):
client = QuestionAnsweringClient(qna_creds["qna_endpoint"], AzureKeyCredential(qna_creds["qna_key"]))
params = AnswersFromTextOptions(
question="What is the meaning of life?",
text_documents=[
TextDocument(text="abc Graphics Surprise, surprise -- our 4K ", id="doc1"),
TextDocument(text="Short text about battery life and charging speed.", id="doc2"),
],
language="en",
)
async with client:
output = await client.get_answers_from_text(params)
assert output.answers
for answer in output.answers:
assert answer.answer
assert answer.confidence is not None
assert answer.id is not None
@pytest.mark.asyncio
async def test_query_text_with_str_records(self, recorded_test, qna_creds):
client = QuestionAnsweringClient(qna_creds["qna_endpoint"], AzureKeyCredential(qna_creds["qna_key"]))
params = {
"question": "How long it takes to charge surface?",
# Use object form matching TextDocument wire schema (records list of objects with text/id)
"records": [
{
"text": "Power and charging. It takes two to four hours to charge the Surface Pro 4 battery fully.",
"id": "r1",
},
{
"text": "You can use the USB port on your Surface Pro 4 power supply to charge other devices.",
"id": "r2",
},
],
"language": "en",
}
async with client:
output = await client.get_answers_from_text(params)
assert output.answers
@pytest.mark.asyncio
async def test_query_text_overload_errors(self): # negative parameter validation
async with QuestionAnsweringClient("http://fake.com", AzureKeyCredential("123")) as client:
with pytest.raises(TypeError):
await client.get_answers_from_text("positional_one", "positional_two") # type: ignore[arg-type]
with pytest.raises(TypeError):
await client.get_answers_from_text("positional_options_bag", options="options bag by name") # type: ignore[arg-type]
params = AnswersFromTextOptions(
question="Meaning?",
text_documents=[TextDocument(text="foo", id="doc1"), TextDocument(text="bar", id="doc2")],
)
with pytest.raises(TypeError):
await client.get_answers_from_text(options=params) # type: ignore[arg-type]
with pytest.raises(TypeError):
await client.get_answers_from_text(question="why?", text_documents=["foo", "bar"], options=params) # type: ignore[arg-type]
with pytest.raises(TypeError):
await client.get_answers_from_text(params, question="Why?") # type: ignore[arg-type]
@pytest.mark.asyncio
async def test_query_text_default_lang_override(self, recorded_test, qna_creds):
client = QuestionAnsweringClient(
qna_creds["qna_endpoint"], AzureKeyCredential(qna_creds["qna_key"]), default_language="es"
)
async with client:
output = await client.get_answers_from_text(
{
"question": "How long it takes to charge surface?",
# explicit language to assert value (client default_language does not auto inject into body)
"language": "es",
# normalize field name for clarity
"records": [
{
"text": "Power and charging. It takes two to four hours to charge the Surface Pro 4 battery fully.",
"id": "doc1",
}
],
},
raw_response_hook=lambda r: _assert_request_language(r, "es"),
)
output = await client.get_answers_from_text(
{
"question": "How long it takes to charge surface?",
"language": "en",
"records": [
{
"text": "Power and charging. It takes two to four hours to charge the Surface Pro 4 battery fully.",
"id": "doc1",
}
],
},
raw_response_hook=lambda r: _assert_request_language(r, "en"),
)
assert output is not None
def _assert_request_language(response, expected):
import json
body = json.loads(response.http_request.content)
assert body.get("language") == expected
|
