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

💾 refactor: remove system definition example

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-04 19:09:28 +02:00
parent 3204ba2328
commit 7f6e06914f
2 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dev_gpt/apis/gpt.py
View File

@@ -145,10 +145,7 @@ class _GPTConversation:
return response.content return response.content
@staticmethod @staticmethod
def _create_system_message(task_description, test_description, system_definition_examples: List[str] = []) -> SystemMessage: def _create_system_message(task_description, test_description) -> SystemMessage:
if system_definition_examples is None:
return None
system_message = PromptTemplate.from_template(template_system_message_base).format( system_message = PromptTemplate.from_template(template_system_message_base).format(
task_description=task_description, task_description=task_description,
test_description=test_description, test_description=test_description,

11
dev_gpt/options/generate/generator.py
View File

@@ -94,7 +94,6 @@ metas:
destination_folder: str, destination_folder: str,
file_name_s: List[str] = None, file_name_s: List[str] = None,
parse_result_fn: Callable = None, parse_result_fn: Callable = None,
system_definition_examples: List[str] = [],
**template_kwargs **template_kwargs
): ):
"""This function generates file(s) using the given template and persists it/them in the given destination folder. """This function generates file(s) using the given template and persists it/them in the given destination folder.
@@ -108,16 +107,16 @@ metas:
parse_result_fn (Callable, optional): A function that parses the generated content and returns a dictionary parse_result_fn (Callable, optional): A function that parses the generated content and returns a dictionary
mapping file_name to its content. If no content could be extract, it returns an empty dictionary. mapping file_name to its content. If no content could be extract, it returns an empty dictionary.
Defaults to None. If None, default parsing is used which uses the file_name to extract from the generated content. Defaults to None. If None, default parsing is used which uses the file_name to extract from the generated content.
system_definition_examples (List[str], optional): The system definition examples to be used for the conversation. Defaults to [].
**template_kwargs: The keyword arguments to be passed to the template. **template_kwargs: The keyword arguments to be passed to the template.
""" """
if parse_result_fn is None: if parse_result_fn is None:
parse_result_fn = self.get_default_parse_result_fn(file_name_s) parse_result_fn = self.get_default_parse_result_fn(file_name_s)
print_colored('', f'\n\n############# {section_title} #############', 'blue') print_colored('', f'\n\n############# {section_title} #############', 'blue')
system_introduction_message = _GPTConversation._create_system_message(self.microservice_specification.task, system_introduction_message = _GPTConversation._create_system_message(
self.microservice_specification.test, self.microservice_specification.task,
system_definition_examples) self.microservice_specification.test
)
conversation = self.gpt_session.get_conversation(messages=[system_introduction_message]) conversation = self.gpt_session.get_conversation(messages=[system_introduction_message])
template_kwargs = {k: v for k, v in template_kwargs.items() if k in template.input_variables} template_kwargs = {k: v for k, v in template_kwargs.items() if k in template.input_variables}
if 'file_name' in template.input_variables and len(file_name_s) == 1: if 'file_name' in template.input_variables and len(file_name_s) == 1:
@@ -345,7 +344,6 @@ pytest
destination_folder=next_microservice_path, destination_folder=next_microservice_path,
file_name_s=['apt-get-packages.json'], file_name_s=['apt-get-packages.json'],
parse_result_fn=self.parse_result_fn_dockerfile, parse_result_fn=self.parse_result_fn_dockerfile,
system_definition_examples=[],
summarized_error=summarized_error, summarized_error=summarized_error,
all_files_string=dock_req_string, all_files_string=dock_req_string,
) )
@@ -417,7 +415,6 @@ pytest
template=template_generate_possible_packages, template=template_generate_possible_packages,
destination_folder=self.microservice_root_path, destination_folder=self.microservice_root_path,
file_name_s=['strategies.json'], file_name_s=['strategies.json'],
system_definition_examples=[],
description=self.microservice_specification.task description=self.microservice_specification.task
)['strategies.json'] )['strategies.json']
packages_list = [[pkg.strip().lower() for pkg in packages] for packages in json.loads(packages_json_string)] packages_list = [[pkg.strip().lower() for pkg in packages] for packages in json.loads(packages_json_string)]