fix: fix bring back the test description for func generation

This commit is contained in:
Florian Hönicke
2023-05-26 00:32:07 +02:00
parent c148e8305d
commit 3e25d89c57
2 changed files with 3 additions and 12 deletions

View File

@@ -165,8 +165,8 @@ class _GPTConversation:
@staticmethod @staticmethod
def _create_system_message(task_description, test_description) -> SystemMessage: def _create_system_message(task_description, test_description) -> SystemMessage:
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,
) )
return SystemMessage(content=system_message) return SystemMessage(content=system_message)

View File

@@ -2,17 +2,8 @@ from dev_gpt.options.generate.templates_user import not_allowed_docker_string, n
template_system_message_base = f'''It is September 2021. template_system_message_base = f'''It is September 2021.
You are a principal engineer working at Jina - an open source company. You are a helpful assistant.
You accurately satisfy all of the user's requirements. You accurately satisfy all of the user's requirements.
To be more specific, you help the user to build a microservice with the following requirements:
```
{{task_description}}
```
and the following test scenario:
```
{{test_description}}
```
You must obey the following rules: You must obey the following rules:
{not_allowed_function_string} {not_allowed_function_string}
{not_allowed_docker_string}''' {not_allowed_docker_string}'''