Skip to content

Commit 296c972

Browse files
committed
feat: Remove LLAMA2 provider and related imports
fix: Update provider_map to remove LLAMA2 fix: Remove LLAMA2-specific console commands
1 parent 2ef4a8d commit 296c972

File tree

3 files changed

+1
-24
lines changed

3 files changed

+1
-24
lines changed

src/pytgpt/async_providers.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
"phind": AsyncPHIND,
2525
"opengpt": AsyncOPENGPT,
2626
"koboldai": AsyncKOBOLDAI,
27-
# "gpt4free": AsyncGPT4FREE,
2827
"blackboxai": AsyncBLACKBOXAI,
2928
"yepchat": AsyncYEPCHAT,
3029
}

src/pytgpt/auto/main.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from pytgpt.opengpt import OPENGPT, AsyncOPENGPT
33
from pytgpt.koboldai import KOBOLDAI, AsyncKOBOLDAI
44
from pytgpt.phind import PHIND, AsyncPHIND
5-
from pytgpt.llama2 import LLAMA2, AsyncLLAMA2
65
from pytgpt.blackboxai import BLACKBOXAI, AsyncBLACKBOXAI
76
from pytgpt.perplexity import PERPLEXITY
87
from pytgpt.gpt4free import GPT4FREE, AsyncGPT4FREE
@@ -18,13 +17,12 @@
1817

1918

2019
provider_map: dict[
21-
str, Union[OPENGPT, KOBOLDAI, PHIND, LLAMA2, BLACKBOXAI, PERPLEXITY, GPT4FREE]
20+
str, Union[OPENGPT, KOBOLDAI, PHIND, BLACKBOXAI, PERPLEXITY, GPT4FREE]
2221
] = {
2322
"phind": PHIND,
2423
"perplexity": PERPLEXITY,
2524
"opengpt": OPENGPT,
2625
"koboldai": KOBOLDAI,
27-
"llama2": LLAMA2,
2826
"blackboxai": BLACKBOXAI,
2927
"gpt4free": GPT4FREE,
3028
}

src/pytgpt/console.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -555,26 +555,6 @@ def __init__(
555555
quiet=quiet,
556556
)
557557

558-
elif provider == "llama2":
559-
import pytgpt.llama2.main as llama2
560-
561-
self.bot = llama2.LLAMA2(
562-
is_conversation=disable_conversation,
563-
max_tokens=max_tokens,
564-
temperature=temperature,
565-
presence_penalty=top_p,
566-
frequency_penalty=top_k,
567-
top_p=top_p,
568-
model=getOr(model, llama2.default_model),
569-
timeout=timeout,
570-
intro=intro,
571-
filepath=filepath,
572-
update_file=update_file,
573-
proxies=proxies,
574-
history_offset=history_offset,
575-
act=awesome_prompt,
576-
)
577-
578558
elif provider == "blackboxai":
579559

580560
from pytgpt.blackboxai import BLACKBOXAI

0 commit comments

Comments
 (0)