def start_training(bot: str, user: str, token: str = None, reload=True):
"""
prevents training of the bot,
if the training session is in progress otherwise start training
:param reload: whether to reload model in the cache
:param bot: bot id
:param token: JWT token for remote model reload
:param user: user id
:return: model path
"""
exception = None
model_file = None
training_status = None
if Utility.environment.get('model') and Utility.environment['model'][
'train'].get('event_url'):
Utility.train_model_event(bot, user, token)
else:
try:
apm_client = Utility.initiate_apm_client()
if apm_client:
elasticapm.instrument()
apm_client.begin_transaction(transaction_type="script")
model_file = train_model_for_bot(bot)
training_status = MODEL_TRAINING_STATUS.DONE.value
agent_url = Utility.environment['model']['train'].get('agent_url')
if agent_url:
if token:
Utility.http_request(
'get', urljoin(agent_url, "/api/bot/model/reload"),
token, user)
else:
if reload:
AgentProcessor.reload(bot)
except Exception as e:
logging.exception(e)
training_status = MODEL_TRAINING_STATUS.FAIL.value
exception = str(e)
finally:
if apm_client:
apm_client.end_transaction(name=__name__, result="success")
ModelProcessor.set_training_status(
bot=bot,
user=user,
status=training_status,
model_path=model_file,
exception=exception,
)
return model_file
async def chat(
request_data: TextData, current_user: User = Depends(auth.get_current_user)
):
"""
Fetches a bot response for a given text/query.
It is basically used to test the chat functionality of the agent
"""
if Utility.environment.get('model') and Utility.environment['model']['train'].get('agent_url'):
agent_url = Utility.environment['model']['train'].get('agent_url')
token = auth.create_access_token(data={"sub": current_user.email})
response = Utility.http_request('post', urljoin(agent_url, "/api/bot/chat"), token.decode('utf8'), current_user.get_user(), json={'data': request_data.data})
else:
model = AgentProcessor.get_agent(current_user.get_bot())
response = await model.handle_text(
request_data.data, sender_id=current_user.get_user()
)
response = {"data": {"response": response}}
return response
async def trigger_data_importer(bot: Text,
user: Text,
save_data: bool,
overwrite: bool = True):
"""
Triggers data importer event which validates and imports data into kairon.
@param bot: bot id.
@param user: kairon username.
@param save_data: Flag to import data into kairon. If set to false, then only validation is run.
Otherwise, both validation and import is done.
@param overwrite: Overwrite existing data(if set to true) or append (if set to false).
@return:
"""
validation_status = 'Failure'
path = None
try:
if Utility.get_event_url("DATA_IMPORTER"):
import_flag = '--import-data' if save_data else ''
overwrite_flag = '--overwrite' if overwrite else ''
env_var = {
'BOT': bot,
'USER': user,
"IMPORT_DATA": import_flag,
"OVERWRITE": overwrite_flag
}
event_request = Utility.build_event_request(env_var)
Utility.http_request(
"POST", Utility.environment['model']['data_importer'].get(
'event_url'), None, user, event_request)
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.TASKSPAWNED.value)
else:
path = Utility.get_latest_file(
os.path.join('training_data', bot))
files_received = DataImporterLogProcessor.get_files_received_for_latest_event(
bot)
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.PARSE.value)
data_importer = DataImporter(path, bot, user, files_received,
save_data, overwrite)
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.VALIDATING.value)
summary, component_count = await data_importer.validate()
is_data_valid = all(
[not summary[key] for key in summary.keys()])
validation_status = 'Success' if is_data_valid else 'Failure'
DataImporterLogProcessor.update_summary(
bot,
user,
component_count,
summary,
status=validation_status,
event_status=EVENT_STATUS.SAVE.value)
if is_data_valid:
data_importer.import_data()
DataImporterLogProcessor.add_log(
bot, user, event_status=EVENT_STATUS.COMPLETED.value)
except exceptions.ConnectionError as e:
logger.error(str(e))
DataImporterLogProcessor.add_log(
bot,
user,
exception='Failed to trigger the event.',
status=validation_status,
event_status=EVENT_STATUS.FAIL.value)
except Exception as e:
logger.error(str(e))
DataImporterLogProcessor.add_log(
bot,
user,
exception=str(e),
status=validation_status,
event_status=EVENT_STATUS.FAIL.value)
if path:
Utility.delete_directory(path)
