aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2025-12-22 16:14:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

🪓 feat: sub task refinement - task refinement works

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-06 19:45:00 +02:00
parent 932d9b44c4
commit 1f86461e1c
1 changed files with 112 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

123
dev_gpt/options/generate/pm.py
View File

@@ -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.
Note: You must not mention any details about algorithms or the technical implementation.'''
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 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.'