🪓 feat: sub task refinement - task refinement works

dev_gpt/options/generate/pm.py

@@ -10,7 +10,7 @@ class PM:
         self.gpt_session = gpt_session
 
     def refine(self, microservice_description):
-        microservice_description = self.refine_description(microservice_description)
+        # microservice_description = self.refine_description(microservice_description)
         sub_task_tree = self.construct_sub_task_tree(microservice_description)
 
     def construct_sub_task_tree(self, microservice_description):
@@ -83,17 +83,30 @@ class PM:
             }
             ]
         """
-        sub_task_tree = self.ask_gpt(construct_sub_task_tree_prompt, json_parser, microservice_description=microservice_description)
+        microservice_description = self.refine_description(microservice_description)
+        sub_task_tree = self.ask_gpt(construct_sub_task_tree_prompt, json_parser,
+                                     microservice_description=microservice_description)
+        # reflections = self.ask_gpt(sub_task_tree_reflections_prompt, identity_parser, microservice_description=microservice_description, sub_task_tree=sub_task_tree)
+        # solutions = self.ask_gpt(sub_task_tree_solutions_prompt, identity_parser, microservice_description=microservice_description, sub_task_tree=sub_task_tree, reflections=reflections)
+        # sub_task_tree_updated = self.ask_gpt(sub_task_tree_update_prompt, json_parser, microservice_description=microservice_description, sub_task_tree=sub_task_tree, solutions=solutions)
+        # return sub_task_tree_updated
         return sub_task_tree
 
+
     def refine_description(self, microservice_description):
+        microservice_description = self.ask_gpt(better_description_prompt, identity_parser, microservice_description=microservice_description)
         request_schema = self.ask_gpt(generate_request_schema_prompt, identity_parser,
                                       microservice_description=microservice_description)
         response_schema = self.ask_gpt(generate_output_schema_prompt, identity_parser,
                                        microservice_description=microservice_description, request_schema=request_schema)
+        # additional_specifications = self.add_additional_specifications(microservice_description, request_schema,
+        #                                                                response_schema)
         microservice_description = self.ask_gpt(summarize_description_and_schemas_prompt, identity_parser,
                                                 microservice_description=microservice_description,
-                                                request_schema=request_schema, response_schema=response_schema)
+                                                request_schema=request_schema,
+                                                response_schema=response_schema,
+                                                # additional_specifications=additional_specifications
+                                                )
 
         while (user_feedback := self.get_user_feedback(microservice_description)):
             microservice_description = self.ask_gpt(add_feedback_prompt, identity_parser,
@@ -101,9 +114,25 @@ class PM:
                                                     user_feedback=user_feedback)
         return microservice_description
 
+    def add_additional_specifications(self, microservice_description, request_schema, response_schema):
+        questions = self.ask_gpt(
+            ask_questions_prompt, identity_parser,
+            microservice_description=microservice_description,
+            request_schema=request_schema, response_schema=response_schema)
+        additional_specifications = self.ask_gpt(
+            answer_questions_prompt,
+            identity_parser,
+            microservice_description=microservice_description,
+            request_schema=request_schema,
+            response_schema=response_schema,
+            questions=questions
+        )
+        return additional_specifications
+
     def get_user_feedback(self, microservice_description):
         while True:
-            user_feedback = input(f'I suggest that we implement the following microservice:\n{microservice_description}\nDo you agree? [y/n]')
+            user_feedback = input(
+                f'I suggest that we implement the following microservice:\n{microservice_description}\nDo you agree? [y/n]')
             if user_feedback.lower() in ['y', 'yes', 'yeah', 'yep', 'yup', 'sure', 'ok', 'okay']:
                 print('Great! I will hand this over to the developers!')
                 return None
@@ -133,26 +162,38 @@ class PM:
     #             print('Sorry, I can not handle this feedback. Please formulate it more precisely.')
 
 
-
-
 def identity_parser(x):
     return x
 
+
 def boolean_parser(x):
     return 'yes' in x.lower()
 
+
 def json_parser(x):
     if '```' in x:
         pattern = r'```(.+)```'
         x = re.findall(pattern, x, re.DOTALL)[-1]
     return json.loads(x)
 
+
 client_description = '''\
 Microservice description:
 ```
 {microservice_description}
 ```'''
 
+better_description_prompt = client_description + '''
+Update the description of the Microservice to make it more precise without adding or removing information.
+Note: the output must be a list of tasks the Microservice has to perform.
+Example for the description: "return the average temperature of the 5 days weather forecast for a given location."
+1. get the 5 days weather forcast from the https://openweathermap.org/ API
+2. extract the temperature from the response
+3. calculate the average temperature'''
+
+# better_description_prompt = client_description + '''
+# Update the description of the Microservice to make it more precise without adding or removing information.'''
+
 generate_request_schema_prompt = client_description + '''
 Generate the lean request json schema of the Microservice.
 Note: If you are not sure about the details, the come up with the minimal number of parameters possible.'''
@@ -177,16 +218,18 @@ Note: If you are not sure about the details, the come up with the minimal number
 
 
 summarize_description_and_schemas_prompt = client_description + '''
-request json schema:
+Request json schema:
 ```
 {request_schema}
 ```
-response json schema:
+Response json schema:
 ```
 {response_schema}
 ```
-Update the microservice description by incorporating information about the request and response parameters in a concise way without losing any information.
+Write an updated microservice description by incorporating information about the request and response parameters in a concise way without losing any information.
-Note: You must not mention any details about algorithms or the technical implementation.'''
+Note: You must not mention any details about algorithms or the technical implementation.
+Note: You must not mention that there is a request and response JSON schema
+Note: You must not use any formatting like triple backticks.'''
 
 add_feedback_prompt = client_description + '''
 User feedback:
@@ -203,7 +246,6 @@ Note: You must ignore facts that are not relevant.
 Note: You must ignore facts that are unknown.
 Note: You must ignore facts that are unclear.'''
 
-
 construct_sub_task_tree_prompt = client_description + '''\
 Recursively constructs a tree of sub-tasks that need to be done to implement the microservice
 Example1:
@@ -234,6 +276,65 @@ Output:
 Note: you must only output the json string - nothing else.
 Note: you must pretty print the json string.'''
 
+sub_task_tree_reflections_prompt = client_description + '''\
+Sub task tree:
+```
+{sub_task_tree}
+```
+Reflect on the sub task tree and write up to 10 constructive criticisms (5 words) about it.'''
+
+sub_task_tree_solutions_prompt = client_description + '''\
+Sub task tree:
+```
+{sub_task_tree}
+```
+Reflections:
+```
+{reflections}
+```
+For each constructive criticism, write a solution (5 words) that address the criticism.'''
+
+sub_task_tree_update_prompt = client_description + '''\
+Sub task tree:
+```
+{sub_task_tree}
+```
+Solutions:
+```
+{solutions}
+```
+Update the sub task tree by applying the solutions. (pritty print the json string)'''
+
+
+ask_questions_prompt = client_description + '''
+Request json schema:
+```
+{request_schema}
+```
+Response json schema:
+```
+{response_schema}
+```
+Ask the user up to 5 unique detailed questions (5 words) about the microservice description that are not yet answered.
+'''
+
+answer_questions_prompt = client_description + '''
+Request json schema:
+```
+{request_schema}
+```
+Response json schema:
+```
+{response_schema}
+```
+Questions:
+```
+{questions}
+```
+Answer all questions where you can think of a plausible answer.
+Note: You must not answer questions with something like "...is not specified", "I don't know" or "Unknown".
+'''
+
 if __name__ == '__main__':
     gpt_session = gpt.GPTSession(None, 'GPT-3.5-turbo')
     first_question = 'Please specify your microservice.'