File: sample_chat_async.py

package info (click to toggle)
python-azure 20251118%2Bgit-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 783,356 kB
  • sloc: python: 6,474,533; ansic: 804; javascript: 287; sh: 205; makefile: 198; xml: 109
file content (94 lines) | stat: -rw-r--r-- 3,244 bytes parent folder | download
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
# coding=utf-8
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------

"""
FILE: sample_chat_async.py

DESCRIPTION:
    Async follow-up (contextual) question with answer_context.

USAGE:
    python sample_chat_async.py
"""

from __future__ import annotations
import asyncio


async def sample_chit_chat():
    # [START chit_chat_async]
    import os
    from azure.core.credentials import AzureKeyCredential
    from azure.ai.language.questionanswering.aio import QuestionAnsweringClient
    from azure.ai.language.questionanswering.models import (
        AnswersOptions,
        ShortAnswerOptions,
        KnowledgeBaseAnswerContext,
    )

    endpoint = os.environ["AZURE_QUESTIONANSWERING_ENDPOINT"]
    key = os.environ["AZURE_QUESTIONANSWERING_KEY"]
    project = os.environ["AZURE_QUESTIONANSWERING_PROJECT"]
    deployment = os.environ.get("AZURE_QUESTIONANSWERING_DEPLOYMENT", "production")

    client = QuestionAnsweringClient(endpoint, AzureKeyCredential(key))
    async with client:
        first_question = "How long should my Surface battery last?"
        first_options = AnswersOptions(
            question=first_question,
            top=3,
            confidence_threshold=0.2,
            include_unstructured_sources=True,
            short_answer_options=ShortAnswerOptions(enable=True, confidence_threshold=0.2, top=1),
        )
        first_output = await client.get_answers(
            first_options,
            project_name=project,
            deployment_name=deployment,
        )
        best_candidate = next(
            (a for a in (first_output.answers or []) if a.confidence and a.confidence > 0.7),
            None,
        )
        if not best_candidate:
            print(f"No answers for '{first_question}'")
            return

        print(f"Q: {first_question}")
        print(f"A: {best_candidate.answer}")

        if best_candidate.qna_id:
            followup_question = "How long does it take to charge a Surface?"
            followup_options = AnswersOptions(
                question=followup_question,
                top=3,
                confidence_threshold=0.2,
                answer_context=KnowledgeBaseAnswerContext(
                    previous_question=first_question,
                    previous_qna_id=best_candidate.qna_id,
                ),
                short_answer_options=ShortAnswerOptions(enable=True, confidence_threshold=0.2, top=1),
                include_unstructured_sources=True,
            )
            follow_output = await client.get_answers(
                followup_options,
                project_name=project,
                deployment_name=deployment,
            )
            follow_best = next(
                (a for a in (follow_output.answers or []) if a.confidence and a.confidence > 0.2),
                None,
            )
            if follow_best:
                print(f"Q: {followup_question}")
                print(f"A: {follow_best.answer}")
            else:
                print(f"No answers for follow-up '{followup_question}'")
    # [END chit_chat_async]


if __name__ == "__main__":
    asyncio.run(sample_chit_chat())