Fix the max_tokens, temperature, and model setup.
This commit is contained in:
@@ -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)}")
|
||||
|
||||
Reference in New Issue
Block a user