modified: lazydev/develop.py

	modified:   lazydev/modules/developer.py
	modified:   lazydev/modules/prompts.py

Author: thecodacus

lazydev/develop.py

@@ -3,7 +3,6 @@
 import os
 from dotenv import load_dotenv
 
-from lazydev.modules.prompts import PrompBook
 from .modules.developer import Developer
 
 

lazydev/modules/developer.py

@@ -1,7 +1,20 @@
 
 from typing import List
-from .prompts import PrompBook
+from .prompts import PromptBook
 from langchain.llms import OpenAI
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts.chat import (
+    ChatPromptTemplate,
+    SystemMessagePromptTemplate,
+    AIMessagePromptTemplate,
+    HumanMessagePromptTemplate,
+)
+from langchain.schema import (
+    AIMessage,
+    HumanMessage,
+    SystemMessage
+)
+
 import json
 
 from .utils import Utilities
@@ -11,16 +24,24 @@ def __init__(self, requirement:str,root_dir:str, openai_api_key:str):
         self.requirement=requirement
         self.root_dir=root_dir
         self.api_key=openai_api_key
-        self.brain = OpenAI(
-            model_name="gpt-3.5-turbo",
+        self.brain = ChatOpenAI(
+            model="gpt-3.5-turbo",
             openai_api_key=self.api_key,
             temperature=0,
             streaming=False
             )
-
+        
+    def brain_storm(self,prompt:str)->str:
+        messages = [
+            SystemMessage(content="you are a senior software developer"),
+            HumanMessage(content=prompt)
+        ]
+        aiMessage:AIMessage= self.brain(messages=messages)
+        return aiMessage.content
+    
     def clear_doubts(self):
-        prompt=PrompBook.expand_requirements(self.requirement)
-        doubts:str=self.brain(prompt)
+        prompt=PromptBook.expand_requirements(self.requirement)
+        doubts=self.brain_storm(prompt)
         doubt_list:List[str]=doubts.split("\n")
         doubt_list=[doubt.strip() for doubt in doubt_list if doubt.strip()!=""]
         print("""
@@ -45,21 +66,20 @@ def clear_doubts(self):
         return doubt_list,answer_list
 
     def plan_project(self):
-        prompt=PrompBook.plan_project(self.requirement,self.clarifications)
-        plannings:str=self.brain(prompt)
+        prompt=PromptBook.plan_project(self.requirement,self.clarifications)
+        plannings:str=self.brain_storm(prompt)
         return plannings
 
     def generate_folder_structure(self):
-        prompt=PrompBook.design_folder_structure(
+        prompt=PromptBook.design_folder_structure(
             question=self.requirement,
             plan=self.plannings,
             clarifications=self.clarifications
             )
-        folder_tree_str:str=self.brain(prompt)
+        folder_tree_str:str=self.brain_storm(prompt)
         folder_tree:dict=json.loads(folder_tree_str)
         Utilities.generate_files_and_folders(structure=folder_tree,root_dir=self.root_dir)
 
-
     def develop(self):
         # clearing all doubts
         doubts,answers=self.clear_doubts()

lazydev/modules/prompts.py

@@ -1,5 +1,8 @@
 
-class PrompBook:
+from typing import List
+
+
+class PromptBook:
     @staticmethod
     def expand_requirements(question:str)->str:
         return f"""
@@ -76,4 +79,20 @@ def design_folder_structure(question:str,plan:str,clarifications:str)->str:
 * do not write the contents of the files that will be done by the next developer
 Begin!
 """
+    @staticmethod
+    def prioritise_file_list(filelist:List[str])->List[str]:
+        list_string='\n'.join(filelist)
+        return f"""
+you are a senior programmer. you going to write a api service.
+below are the file list that you are going to complete in future.
+
+reorder them in a way that most suitable based on how one will be dependent on other
+
+{list_string}
+
+As your response will go to an automated parser, things to keep in mind all the time:
+* only write file list in order nothing else
+* do not write any explanation
+Begin!
+        """