async def test_agent_with_model_server_in_thread(
model_server: TestClient, default_domain: Domain,
unpacked_trained_rasa_model: Text):
model_endpoint_config = EndpointConfig.from_dict({
"url":
model_server.make_url("/model"),
"wait_time_between_pulls":
2
})
agent = Agent()
agent = await rasa.core.agent.load_from_server(
agent, model_server=model_endpoint_config)
await asyncio.sleep(5)
assert agent.fingerprint == "somehash"
assert agent.domain.as_dict() == default_domain.as_dict()
expected_policies = PolicyEnsemble.load_metadata(
str(Path(unpacked_trained_rasa_model, "core")))["policy_names"]
agent_policies = {
rasa.shared.utils.common.module_path_from_instance(p)
for p in agent.policy_ensemble.policies
}
assert agent_policies == set(expected_policies)
assert model_server.app.number_of_model_requests == 1
jobs.kill_scheduler()
def moodbot_metadata(unpacked_trained_moodbot_path):
return PolicyEnsemble.load_metadata(
os.path.join(unpacked_trained_moodbot_path, "core"))
def moodbot_metadata():
return PolicyEnsemble.load_metadata(os.path.join(MOODBOT_MODEL_PATH, "core"))
def moodbot_metadata():
return PolicyEnsemble.load_metadata(MOODBOT_MODEL_PATH)
