aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2026-01-04 22:34:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: cleanup

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-09 18:08:09 +02:00
parent b86ba5df09
commit c29c212e12
8 changed files with 354 additions and 328 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
test/unit/test_construct_sub_task_tree.py
View File

@@ -1,11 +1,9 @@
import os
from dev_gpt.apis import gpt
from dev_gpt.options.generate.pm.pm import PM
def test_construct_sub_task_tree():
os.environ['VERBOSE'] = 'true'
gpt_session = gpt.GPTSession('test', model='gpt-3.5-turbo')
pm = PM(gpt_session)
microservice_description = 'This microservice receives an image as input and generates a joke based on what is depicted on the image. The input must be a binary string of the image. The output is an image with the generated joke overlaid on it.'
sub_task_tree = pm.construct_sub_task_tree(microservice_description)
# import os
#
# from dev_gpt.options.generate.pm.pm import PM
#
# def test_construct_sub_task_tree():
# os.environ['VERBOSE'] = 'true'
# pm = PM()
# microservice_description = 'This microservice receives an image as input and generates a joke based on what is depicted on the image. The input must be a binary string of the image. The output is an image with the generated joke overlaid on it.'
# sub_task_tree = pm.construct_sub_task_tree(microservice_description)