wdplx commited on
Commit
f1c7954
1 Parent(s): e24fbee

removed debugger

Browse files
Files changed (2) hide show
  1. app.py +0 -4
  2. sotopia_pi_generate.py +0 -3
app.py CHANGED
@@ -102,7 +102,6 @@ def create_bot_agent_dropdown(environment_id, user_agent_id):
102
  environment, user_agent = environment_dict[environment_id], agent_dict[user_agent_id]
103
 
104
  bot_agent_list = []
105
- # import pdb; pdb.set_trace()
106
  for neighbor_id in relationship_dict[environment.relationship][user_agent.agent_id]:
107
  bot_agent_list.append((agent_dict[neighbor_id].name, neighbor_id))
108
 
@@ -122,7 +121,6 @@ def create_user_info(user_agent_dropdown):
122
 
123
  def create_bot_info(bot_agent_dropdown):
124
  _, _, agent_dict, _ = get_sotopia_profiles()
125
- # import pdb; pdb.set_trace()
126
  bot_agent = agent_dict[bot_agent_dropdown]
127
  text = f"{bot_agent.background} {bot_agent.personality}"
128
  return gr.Textbox(label="Bot Agent Profile", lines=4, value=text)
@@ -215,12 +213,10 @@ def chat_tab():
215
  user_agent = agent_dict[user_agent_dropdown]
216
  bot_agent = agent_dict[bot_agent_dropdown]
217
 
218
- # import pdb; pdb.set_trace()
219
  context = get_context_prompt(bot_agent, user_agent, environment)
220
  dialogue_history, next_turn_idx = dialogue_history_prompt(message, history, user_agent, bot_agent)
221
  prompt_history = f"{context}\n\n{dialogue_history}"
222
  agent_action = generate_action(model_selection, prompt_history, next_turn_idx, ACTION_TYPES, bot_agent.name, TEMPERATURE)
223
- # import pdb; pdb.set_trace()
224
  return agent_action.to_natural_language()
225
 
226
  with gr.Column():
 
102
  environment, user_agent = environment_dict[environment_id], agent_dict[user_agent_id]
103
 
104
  bot_agent_list = []
 
105
  for neighbor_id in relationship_dict[environment.relationship][user_agent.agent_id]:
106
  bot_agent_list.append((agent_dict[neighbor_id].name, neighbor_id))
107
 
 
121
 
122
  def create_bot_info(bot_agent_dropdown):
123
  _, _, agent_dict, _ = get_sotopia_profiles()
 
124
  bot_agent = agent_dict[bot_agent_dropdown]
125
  text = f"{bot_agent.background} {bot_agent.personality}"
126
  return gr.Textbox(label="Bot Agent Profile", lines=4, value=text)
 
213
  user_agent = agent_dict[user_agent_dropdown]
214
  bot_agent = agent_dict[bot_agent_dropdown]
215
 
 
216
  context = get_context_prompt(bot_agent, user_agent, environment)
217
  dialogue_history, next_turn_idx = dialogue_history_prompt(message, history, user_agent, bot_agent)
218
  prompt_history = f"{context}\n\n{dialogue_history}"
219
  agent_action = generate_action(model_selection, prompt_history, next_turn_idx, ACTION_TYPES, bot_agent.name, TEMPERATURE)
 
220
  return agent_action.to_natural_language()
221
 
222
  with gr.Column():
sotopia_pi_generate.py CHANGED
@@ -159,7 +159,6 @@ def generate(
159
  output_parser: BaseOutputParser[OutputType],
160
  temperature: float = 0.7,
161
  ) -> OutputType:
162
- # import pdb; pdb.set_trace()
163
  input_variables = re.findall(r"{(.*?)}", template)
164
  assert (
165
  set(input_variables) == set(list(input_values.keys()) + ["format_instructions"])
@@ -172,7 +171,6 @@ def generate(
172
  input_values["format_instructions"] = output_parser.get_format_instructions()
173
  result = chain.predict([logging_handler], **input_values)
174
  prompt = logging_handler.retrive_prompt()
175
- import pdb; pdb.set_trace()
176
  try:
177
  parsed_result = output_parser.parse(result)
178
  except KeyboardInterrupt:
@@ -182,7 +180,6 @@ def generate(
182
  f"[red] Failed to parse result: {result}\nEncounter Exception {e}\nstart to reparse",
183
  extra={"markup": True},
184
  )
185
- import pdb; pdb.set_trace()
186
  reformat_parsed_result = format_bad_output(
187
  result, format_instructions=output_parser.get_format_instructions()
188
  )
 
159
  output_parser: BaseOutputParser[OutputType],
160
  temperature: float = 0.7,
161
  ) -> OutputType:
 
162
  input_variables = re.findall(r"{(.*?)}", template)
163
  assert (
164
  set(input_variables) == set(list(input_values.keys()) + ["format_instructions"])
 
171
  input_values["format_instructions"] = output_parser.get_format_instructions()
172
  result = chain.predict([logging_handler], **input_values)
173
  prompt = logging_handler.retrive_prompt()
 
174
  try:
175
  parsed_result = output_parser.parse(result)
176
  except KeyboardInterrupt:
 
180
  f"[red] Failed to parse result: {result}\nEncounter Exception {e}\nstart to reparse",
181
  extra={"markup": True},
182
  )
 
183
  reformat_parsed_result = format_bad_output(
184
  result, format_instructions=output_parser.get_format_instructions()
185
  )