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

Fix typing, linting and add tests

Browse source
This commit is contained in:
Louis 2023-12-03 16:31:20 +01:00
parent 76faffb269
commit 5bc5054000
3 changed files with 111 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

6
private_gpt/components/llm/llm_component.py
View file

@ -22,12 +22,14 @@ class LLMComponent:
case "local":
from llama_index.llms import LlamaCPP
from private_gpt.components.llm.prompt.prompt_helper import get_prompt_style
from private_gpt.components.llm.prompt.prompt_helper import (
get_prompt_style,
)
prompt_style = get_prompt_style(
prompt_style=settings.local.prompt_style,
template_name=settings.local.template_name,
default_system_prompt=settings.local.default_system_prompt
default_system_prompt=settings.local.default_system_prompt,
)
self.llm = LlamaCPP(