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

feat: search fix web search

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-20 01:05:09 +02:00
parent b2f1ce4489
commit 16f1f7b6af
4 changed files with 34 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -257,11 +257,12 @@ metas:
print('\nFirst version of the microservice generated. Start iterating on it to make the tests pass...')
def add_missing_imports_post_process_fn(self, content_raw: str):
def add_missing_imports_post_process_fn(self, content_dict: dict):
for indicator, import_statement in INDICATOR_TO_IMPORT_STATEMENT.items():
if indicator in content_raw and import_statement not in content_raw:
content_raw = f'{import_statement}\n{content_raw}'
return content_raw
for file_name, file_content in content_dict.items():
if indicator in file_content and import_statement not in file_content:
content_dict[file_name] = f'{import_statement}\n{file_content}'
return content_dict
@staticmethod