Chilkat Online Tools

QueryAssistant Python Example

Amazon Wisdom

import sys
import chilkat

# This example requires the Chilkat API to have been previously unlocked.
# See Global Unlock Sample for sample code.

rest = chilkat.CkRest()

authAws = chilkat.CkAuthAws()
authAws.put_AccessKey("AWS_ACCESS_KEY")
authAws.put_SecretKey("AWS_SECRET_KEY")

# Don't forget to change the region to your particular region. (Also make the same change in the call to Connect below.)
authAws.put_Region("us-west-2")
authAws.put_ServiceName("wisdom")
# SetAuthAws causes Chilkat to automatically add the following headers: Authorization, X-Amz-Date
rest.SetAuthAws(authAws)

# URL: https://wisdom.us-west-2.amazonaws.com/
# Use the same region as specified above.
success = rest.Connect("wisdom.us-west-2.amazonaws.com",443,True,True)
if (success != True):
    print("ConnectFailReason: " + str(rest.get_ConnectFailReason()))
    print(rest.lastErrorText())
    sys.exit()

# The following code creates the JSON request body.
# The JSON created by this code is shown below.

# Use this online tool to generate code from sample JSON:
# Generate Code to Create JSON

json = chilkat.CkJsonObject()
json.UpdateInt("maxResults",123)
json.UpdateString("nextToken","string")
json.UpdateString("queryText","string")

# The JSON request body created by the above code:

# {
#   "maxResults": number,
#   "nextToken": "string",
#   "queryText": "string"
# }

rest.AddHeader("Content-Type","application/x-amz-json-1.1")
rest.AddHeader("X-Amz-Target","QueryAssistant")

sbRequestBody = chilkat.CkStringBuilder()
json.EmitSb(sbRequestBody)
sbResponseBody = chilkat.CkStringBuilder()
success = rest.FullRequestSb("POST","/assistants/{assistantId}/query",sbRequestBody,sbResponseBody)
if (success != True):
    print(rest.lastErrorText())
    sys.exit()

respStatusCode = rest.get_ResponseStatusCode()
print("response status code = " + str(respStatusCode))
if (respStatusCode != 200):
    print("Response Header:")
    print(rest.responseHeader())
    print("Response Body:")
    print(sbResponseBody.getAsString())
    sys.exit()

jResp = chilkat.CkJsonObject()
jResp.LoadSb(sbResponseBody)

# The following code parses the JSON response.
# A sample JSON response is shown below the sample code.

# Use this online tool to generate parsing code from sample JSON:
# Generate Parsing Code from JSON

nextToken = jResp.stringOf("nextToken")
i = 0
count_i = jResp.SizeOfArray("results")
while i < count_i :
    jResp.put_I(i)
    ContentArn = jResp.stringOf("results[i].document.contentReference.contentArn")
    ContentId = jResp.stringOf("results[i].document.contentReference.contentId")
    KnowledgeBaseArn = jResp.stringOf("results[i].document.contentReference.knowledgeBaseArn")
    KnowledgeBaseId = jResp.stringOf("results[i].document.contentReference.knowledgeBaseId")
    Text = jResp.stringOf("results[i].document.excerpt.text")
    TitleText = jResp.stringOf("results[i].document.title.text")
    relevanceScore = jResp.IntOf("results[i].relevanceScore")
    resultId = jResp.stringOf("results[i].resultId")
    j = 0
    count_j = jResp.SizeOfArray("results[i].document.excerpt.highlights")
    while j < count_j :
        jResp.put_J(j)
        beginOffsetInclusive = jResp.IntOf("results[i].document.excerpt.highlights[j].beginOffsetInclusive")
        endOffsetExclusive = jResp.IntOf("results[i].document.excerpt.highlights[j].endOffsetExclusive")
        j = j + 1

    j = 0
    count_j = jResp.SizeOfArray("results[i].document.title.highlights")
    while j < count_j :
        jResp.put_J(j)
        beginOffsetInclusive = jResp.IntOf("results[i].document.title.highlights[j].beginOffsetInclusive")
        endOffsetExclusive = jResp.IntOf("results[i].document.title.highlights[j].endOffsetExclusive")
        j = j + 1

    i = i + 1

# A sample JSON response body parsed by the above code:

# {
#   "nextToken": "string",
#   "results": [
#     {
#       "document": {
#         "contentReference": {
#           "contentArn": "string",
#           "contentId": "string",
#           "knowledgeBaseArn": "string",
#           "knowledgeBaseId": "string"
#         },
#         "excerpt": {
#           "highlights": [
#             {
#               "beginOffsetInclusive": number,
#               "endOffsetExclusive": number
#             }
#           ],
#           "text": "string"
#         },
#         "title": {
#           "highlights": [
#             {
#               "beginOffsetInclusive": number,
#               "endOffsetExclusive": number
#             }
#           ],
#           "text": "string"
#         }
#       },
#       "relevanceScore": number,
#       "resultId": "string"
#     }
#   ]
# }