Skip to content

Commit b1e8ae2

Browse files
authored
Add simulator_id to configuration (#147)
Fixes #140
1 parent 07b4d33 commit b1e8ae2

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

config/default/config_default.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
agent_id: "IAI MovieBot"
2+
simulator_id: "User Simulator"
23
output_name: "moviebot"
34
# By default, the agent has an HTTP API.
45
agent_uri: "http://127.0.0.1:5001"

usersimcrs/run_simulation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def main(config: confuse.Configuration, agent: Agent) -> None:
113113
nlg = ConditionalNLG(template)
114114

115115
simulator = AgendaBasedSimulator(
116-
"TEST03",
116+
config["simulator_id"].get(),
117117
preference_model,
118118
interaction_model,
119119
nlu,
@@ -358,9 +358,9 @@ def load_rasa_diet_classifier(
358358
# See usage example in README for more details.
359359
agent_uri = config["agent_uri"].get()
360360
try:
361-
response = requests.get(agent_uri)
361+
response = requests.get(agent_uri, timeout=60)
362362
assert response.status_code == 200
363-
agent = MovieBotAgent(agent_id="MovieBotTester", uri=agent_uri)
363+
agent = MovieBotAgent(agent_id=config["agent_id"].get(), uri=agent_uri)
364364
except requests.exceptions.RequestException:
365365
raise RuntimeError(
366366
f"Connection refused to {agent_uri}. Please check that "

0 commit comments

Comments
 (0)