Fix the max_tokens, temperature, and model setup.

This commit is contained in:
Oleksandr Kozachuk
2023-08-12 12:20:49 +02:00
parent bc5e6228a6
commit e4d055b900
2 changed files with 23 additions and 14 deletions
+5 -6
View File
@@ -1,8 +1,7 @@
import shutil
from pprint import PrettyPrinter
from typing import List, Dict, Union
ConfigType = Dict[str, Union[str, Dict[str, Union[str, int]]]]
ConfigType = dict[str, str | dict[str, str | int | float]]
def terminal_width() -> int:
@@ -31,15 +30,15 @@ def print_tag_args(tags: list[str], extags: list[str], otags: list[str]) -> None
print()
def append_message(chat: List[Dict[str, str]],
def append_message(chat: list[dict[str, str]],
role: str,
content: str
) -> None:
chat.append({'role': role, 'content': content.replace("''", "'")})
def message_to_chat(message: Dict[str, str],
chat: List[Dict[str, str]],
def message_to_chat(message: dict[str, str],
chat: list[dict[str, str]],
with_tags: bool = False,
with_file: bool = False
) -> None:
@@ -80,6 +79,6 @@ def print_chat_hist(chat, dump=False, source_code=False) -> None:
print(f"{message['role'].upper()}: {message['content']}")
def print_tags_frequency(tags: List[str]) -> None:
def print_tags_frequency(tags: list[str]) -> None:
for tag in sorted(set(tags)):
print(f"- {tag}: {tags.count(tag)}")