REGULAR_COLLECTION_NAME = os.getenv('DISCOVERY_REGULAR_COLLECTION_NAME',
'knowledge_base_regular')
ENRICHED_COLLECTION_NAME = os.getenv('DISCOVERY_ENRICHED_COLLECTION_NAME',
'knowledge_base_enriched')
# please provide the collection desired
DESIRED_COLLECTION_NAME = REGULAR_COLLECTION_NAME
# retrieve the following:
# {
# environment_id,
# collection_id_regular,
# collection_id_enriched
# }
discovery_constants = get_constants(discovery,
regular_name=REGULAR_COLLECTION_NAME,
enriched_name=ENRICHED_COLLECTION_NAME)
COLLECTION_ID = discovery_constants['collection_id_regular']
if DESIRED_COLLECTION_NAME == ENRICHED_COLLECTION_NAME:
COLLECTION_ID = discovery_constants['collection_id_enriched']
qopts = {'natural_language_query': QUESTION, 'count': MAX_DOCUMENTS}
# Invoke search using Discovery query API
output = discovery.query(environment_id=discovery_constants['environment_id'],
collection_id=COLLECTION_ID,
query_options=qopts)
try:
print("No. of rows : %d" % output.get('matching_results'))
print("Results : %s" % json.dumps(output['results'], indent=4))
except:
file_object.seek(0)
docs_uploaded += 1
done_percent = write_progress(docs_uploaded, total_documents,
done_percent)
logging.info("Finished uploading %d files", total_files)
print("\nFinished uploading %d files" % total_files)
print('Retrieving environment and collection constants...')
"""
retrieve the following:
{
environment_id: env_id,
collection_id: {
regular: regular_id,
trained: trained_id
}
}
"""
discovery_constants = get_constants(
discovery,
regular_name=os.getenv('DISCOVERY_REGULAR_COLLECTION_NAME',
'knowledge_base_regular'),
trained_name=os.getenv('DISCOVERY_TRAINED_COLLECTION_NAME',
'knowledge_base_trained'))
print('Constants retrieved!')
print(discovery_constants)
print("Log file located at: %s" % LOG_FILE_PATH)
upload_documents(DOCS_DIRECTORY)
file_object.name, collection_id)
upload_document(file_object,
discovery_constants['environment_id'],
collection_id)
file_object.seek(0)
docs_uploaded += 1
done_percent = write_progress(docs_uploaded, total_documents,
done_percent)
logging.info("Finished uploading %d files", total_files)
print("\nFinished uploading %d files" % total_files)
print('Retrieving environment and collection constants...')
# retrieve the following:
# {
# environment_id,
# collection_id_regular,
# collection_id_enriched
# }
discovery_constants = get_constants(
discovery,
regular_name=os.getenv('DISCOVERY_REGULAR_COLLECTION_NAME',
'knowledge_base_regular'),
enriched_name=os.getenv('DISCOVERY_ENRICHED_COLLECTION_NAME',
'knowledge_base_enriched'))
print('Constants retrieved!')
print(discovery_constants)
print("Log file located at: %s" % LOG_FILE_PATH)
upload_documents(DOCS_DIRECTORY)
# please provide the collection desired
DESIRED_COLLECTION_NAME = REGULAR_COLLECTION_NAME
"""
retrieve the following:
{
environment_id: env_id,
collection_id: {
regular: regular_id,
trained: trained_id
}
}
"""
discovery_constants = get_constants(
discovery,
regular_name=REGULAR_COLLECTION_NAME,
trained_name=TRAINED_COLLECTION_NAME,
)
COLLECTION_ID = discovery_constants['collection_id']['regular']
if DESIRED_COLLECTION_NAME == TRAINED_COLLECTION_NAME:
COLLECTION_ID = discovery_constants['collection_id']['trained']
qopts = {'natural_language_query': QUESTION, 'count': MAX_DOCUMENTS}
# Invoke search using Discovery query API
output = discovery.query(environment_id=discovery_constants['environment_id'],
collection_id=COLLECTION_ID,
query_options=qopts)
try:
print ("No. of rows : %d" % output.get('matching_results'))
print ("Results : %s" % json.dumps(output['results'], indent=4))