SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 20:12:55 +01:00
Code Issues Projects Releases Packages Activity

fix: mypy/black

Browse source
This commit is contained in:
Javier Martinez 2024-09-26 09:40:24 +02:00
parent 0cd016a732
commit 4b5c3362f1
No known key found for this signature in database
4 changed files with 20 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

5
private_gpt/components/llm/prompt_helper.py
View file

@ -286,8 +286,9 @@ class ChatMLPromptStyle(AbstractPromptStyle):
def get_prompt_style(
prompt_style: Literal["default", "llama2", "llama3", "tag", "mistral", "chatml"]
| None
prompt_style: (
Literal["default", "llama2", "llama3", "tag", "mistral", "chatml"] | None
)
) -> AbstractPromptStyle:
"""Get the prompt style to use from the given string.