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

🧪 test: level 1

Browse Source
This commit is contained in:
Florian Hönicke
2023-04-28 23:17:51 +02:00
parent cf0d4a3264
commit aab91a4077
2 changed files with 28 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/options/generate/generator.py
View File

@@ -37,9 +37,9 @@ class TaskSpecification:
test: Optional[Text]
class Generator:
def __init__(self, task_description, test_description, path, model='gpt-4'):
self.gpt_session = gpt.GPTSession(task_description, test_description, model=model)
self.microservice_specification = TaskSpecification(task=task_description, test=test_description)
def __init__(self, task_description, path, model='gpt-4'):
self.gpt_session = gpt.GPTSession(task_description, model=model)
self.microservice_specification = TaskSpecification(task=task_description)
self.microservice_root_path = path
def extract_content_from_result(self, plain_text, file_name, match_single_block=False, can_contain_code_block=True):

27
test/test_generator.py
View File

@@ -1,7 +1,30 @@
import os
from src.options.generate.generator import Generator
def test_generator(tmpdir):
# The cognitive difficulty level is determined by the number of Requirements the microservice has.
def test_generation_level_0(tmpdir):
"""
Requirements:
pip packages: ❌
environment: ❌
GPT-3.5-turbo: ❌
APIs: ❌
Databases: ❌
"""
os.environ['VERBOSE'] = 'true'
generator = Generator("The microservice is very simple, it does not take anything as input and only outputs the word 'test'", "my test", str(tmpdir) + 'microservice', 'gpt-3.5-turbo')
generator = Generator("The microservice is very simple, it does not take anything as input and only outputs the word 'test'", str(tmpdir) + 'microservice', 'gpt-3.5-turbo')
generator.generate()
def test_generation_level_1(tmpdir):
"""
Requirements:
pip packages: ✅
environment: ❌
GPT-3.5-turbo: ❌
APIs: ❌
Databases: ❌
"""
os.environ['VERBOSE'] = 'true'
generator = Generator("The input is a PDF like https://www.africau.edu/images/default/sample.pdf and the output the parsed text", str(tmpdir) + 'microservice', 'gpt-3.5-turbo')
generator.generate()