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

Merge pull request #105 from jina-ai/refactor_simplify_name

Browse Source 
🤏 refactor: simplify name
This commit is contained in:
Florian Hönicke
2023-05-23 21:46:20 +02:00
committed by GitHub
parent 9a53e4514b 553ccaf479
commit 7414018285
3 changed files with 17 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dev_gpt/apis/gpt.py
View File

@@ -17,6 +17,7 @@ from urllib3.exceptions import InvalidChunkLength
from dev_gpt.constants import PRICING_GPT4_PROMPT, PRICING_GPT4_GENERATION, PRICING_GPT3_5_TURBO_PROMPT, \
PRICING_GPT3_5_TURBO_GENERATION, CHARS_PER_TOKEN
from dev_gpt.options.generate.conversation_logger import ConversationLogger
from dev_gpt.options.generate.parser import identity_parser
from dev_gpt.options.generate.templates_system import template_system_message_base
from dev_gpt.utils.string_tools import print_colored, get_template_parameters
@@ -170,7 +171,7 @@ class _GPTConversation:
return SystemMessage(content=system_message)
def ask_gpt(prompt_template, parser, **kwargs):
def ask_gpt(prompt_template: str, parser=identity_parser, **kwargs):
template_parameters = get_template_parameters(prompt_template)
if set(template_parameters) != set(kwargs.keys()):
raise ValueError(