Update Prompts_and_Chains.py

Prompts_and_Chains.py

@@ -26,9 +26,20 @@ class PromptTemplates:
         self.summary_prompt = PromptTemplate(
             input_variables=["project_name", "rfp_details"], template=summary_template
         )
-        
+
         self.estimations_prompt_template = PromptTemplate(
-            input_variables=["project_summary", "user_stories","tech_leads", "senior_developers","junior_developers"], template=estimations_template
+            input_variables=[
+                "project_summary",
+                "user_stories",
+                "tech_leads",
+                "senior_developers",
+                "junior_developers",
+            ],
+            template=estimations_template,
+        )
+
+        self.roadmap_prompt = PromptTemplate(
+            input_variables=["project_input"], template=roadmap_template
         )
 
 
@@ -66,9 +77,15 @@ class LLMChains:
             prompt=obj.json_prompt_template,
             verbose="true",
         )
-        
+
         self.estimations_chain = LLMChain(
             llm=OpenAI(model_name="gpt-3.5-turbo-16k", temperature=0.7),
             prompt=obj.estimations_prompt_template,
             verbose="true",
         )
+
+        self.roadmap_chain = LLMChain(
+            llm=OpenAI(model_name="gpt-3.5-turbo-16k", temperature=0.7),
+            prompt=obj.roadmap_prompt,
+            verbose=True,
+        )