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

docs: implement feedback

Browse Source
This commit is contained in:
Florian Hönicke
2023-04-19 10:44:14 +02:00
parent f513277a05
commit 0af51e4bd8
3 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/apis/gpt.py
View File

@@ -69,7 +69,7 @@ class _GPTConversation:
model_name=model, model_name=model,
streaming=True, streaming=True,
callback_manager=CallbackManager([AssistantStreamingStdOutCallbackHandler()]), callback_manager=CallbackManager([AssistantStreamingStdOutCallbackHandler()]),
verbose=os.environ['VERBOSE'].lower() == 'true', verbose=True,
temperature=0, temperature=0,
) )
self.messages: List[BaseMessage] = [] self.messages: List[BaseMessage] = []

6
src/cli.py
View File

@@ -65,10 +65,8 @@ def generate(
from src.options.generate.generator import Generator from src.options.generate.generator import Generator
generator = Generator(description, test, model=model) generator = Generator(description, test, model=model)
with get_openai_callback() as cb: generator.generate(path)
generator.generate(path)
# print(f"Prompt/Completion/Total Tokens: {cb.prompt_tokens}/{cb.completion_tokens}/{cb.total_tokens}")
# print(f"Total Cost on OpenAI (USD): ${cb.total_cost}")
@main.command() @main.command()
@path_param @path_param

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

@@ -53,14 +53,14 @@ metas:
return all_microservice_files_string.strip() return all_microservice_files_string.strip()
def generate_and_persist_file(self, section_title, template, destination_folder, file_name, **kwargs): def generate_and_persist_file(self, section_title, template, destination_folder, file_name, **template_kwargs):
print_colored('', f'\n############# {section_title} #############', 'blue') print_colored('', f'\n############# {section_title} #############', 'blue')
conversation = self.gpt_session.get_conversation() conversation = self.gpt_session.get_conversation()
kwargs = {k: v for k, v in 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}
content_raw = conversation.chat( content_raw = conversation.chat(
template.format( template.format(
file_name=file_name, file_name=file_name,
**kwargs **template_kwargs
) )
) )
content = self.extract_content_from_result(content_raw, file_name, match_single_block=True) content = self.extract_content_from_result(content_raw, file_name, match_single_block=True)