Update topology_generation.py

topology_generation.py

@@ -17,6 +17,7 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_openai import ChatOpenAI
 
 import resinet_llm_prompt
+import data.scenarios.scenarios as sc
 from collections import Counter
 
 from models_configuration import get_model_instance, model_configurations
@@ -253,7 +254,7 @@ def main(args):
     output_format_prompt_stage_3 = resinet_llm_prompt.OUTPUT_FORMAT_PROMPT_STAGE_3.strip()
     output_format_prompt_stage_4 = resinet_llm_prompt.OUTPUT_FORMAT_PROMPT_STAGE_4.strip()
 
-    user_requirements = getattr(resinet_llm_prompt, f"USER_REQUIREMENTS_{args.use_case}").strip()
+    user_requirements = getattr(sc, f"SCENARIO_{args.use_case}").strip()
     topology_ok_parser = PydanticOutputParser(pydantic_object=TopologyData)
     all_devices = []
     all_results = []