|
9 | 9 | from ..informational_agent.informational_prompts import \
|
10 | 10 | informational_role_prompt, conv_pref_prompt, update_conv_pref_prompt, summary_prompt, update_summary_prompt
|
11 | 11 | except ImportError:
|
12 |
| - from agents.utils.parse_json_context_to_prompt import parse_json_to_prompt |
| 12 | + from src.agents.utils.parse_json_context_to_prompt import parse_json_to_prompt |
13 | 13 | from src.agents.base_agent.base_agent import invoke_base_agent
|
14 | 14 | from src.agents.informational_agent.informational_agent import InformationalAgent, invoke_informational_agent
|
15 | 15 | from src.agents.socratic_agent.socratic_agent import invoke_socratic_agent
|
|
18 | 18 |
|
19 | 19 | # File path for the input text
|
20 | 20 | path = "src/agents/utils/"
|
21 |
| -input_file = path + "example_inputs/" + "example_input_1.json" |
| 21 | +input_file = path + "example_inputs/" + "example_input_6.json" |
22 | 22 |
|
23 | 23 | """
|
24 | 24 | STEP 1: Read the USER INFO from the WEB client from a file
|
@@ -127,10 +127,10 @@ def testbench_agents(message, remove_index, agent_type = "informational", inform
|
127 | 127 | f.write("message\t response\t prompt_missing\t prompt\n")
|
128 | 128 |
|
129 | 129 | # NOTE: #### This is the testing message!! #####
|
130 |
| - message = "How to tackle part a?" |
| 130 | + message = "What is the question?" |
131 | 131 | # NOTE: ########################################
|
132 | 132 |
|
133 |
| - index_count = 5 # Number of prompts to be removed for testing (-1 if no removal) |
| 133 | + index_count = -1 # Number of prompts to be removed for testing (-1 if no removal) |
134 | 134 | indices = range(0, index_count) if index_count > -1 else [-1]
|
135 | 135 | for i in indices:
|
136 | 136 | # if i == 16:
|
|
0 commit comments