feat: Remove LLAMA2 provider and related imports

fix: Update provider_map to remove LLAMA2 fix: Remove LLAMA2-specific console commands
Simatwa · Nov 10, 2024 · 296c972 · 296c972
1 parent 2ef4a8d
commit 296c972
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 24 deletions.
diff --git a/src/pytgpt/async_providers.py b/src/pytgpt/async_providers.py
@@ -24,7 +24,6 @@
     "phind": AsyncPHIND,
     "opengpt": AsyncOPENGPT,
     "koboldai": AsyncKOBOLDAI,
-    # "gpt4free": AsyncGPT4FREE,
     "blackboxai": AsyncBLACKBOXAI,
     "yepchat": AsyncYEPCHAT,
 }
diff --git a/src/pytgpt/auto/main.py b/src/pytgpt/auto/main.py
@@ -2,7 +2,6 @@
 from pytgpt.opengpt import OPENGPT, AsyncOPENGPT
 from pytgpt.koboldai import KOBOLDAI, AsyncKOBOLDAI
 from pytgpt.phind import PHIND, AsyncPHIND
-from pytgpt.llama2 import LLAMA2, AsyncLLAMA2
 from pytgpt.blackboxai import BLACKBOXAI, AsyncBLACKBOXAI
 from pytgpt.perplexity import PERPLEXITY
 from pytgpt.gpt4free import GPT4FREE, AsyncGPT4FREE
@@ -18,13 +17,12 @@
 
 
 provider_map: dict[
-    str, Union[OPENGPT, KOBOLDAI, PHIND, LLAMA2, BLACKBOXAI, PERPLEXITY, GPT4FREE]
+    str, Union[OPENGPT, KOBOLDAI, PHIND, BLACKBOXAI, PERPLEXITY, GPT4FREE]
 ] = {
     "phind": PHIND,
     "perplexity": PERPLEXITY,
     "opengpt": OPENGPT,
     "koboldai": KOBOLDAI,
-    "llama2": LLAMA2,
     "blackboxai": BLACKBOXAI,
     "gpt4free": GPT4FREE,
 }

diff --git a/src/pytgpt/console.py b/src/pytgpt/console.py
@@ -555,26 +555,6 @@ def __init__(
                     quiet=quiet,
                 )
 
-            elif provider == "llama2":
-                import pytgpt.llama2.main as llama2
-
-                self.bot = llama2.LLAMA2(
-                    is_conversation=disable_conversation,
-                    max_tokens=max_tokens,
-                    temperature=temperature,
-                    presence_penalty=top_p,
-                    frequency_penalty=top_k,
-                    top_p=top_p,
-                    model=getOr(model, llama2.default_model),
-                    timeout=timeout,
-                    intro=intro,
-                    filepath=filepath,
-                    update_file=update_file,
-                    proxies=proxies,
-                    history_offset=history_offset,
-                    act=awesome_prompt,
-                )
-
             elif provider == "blackboxai":
 
                 from pytgpt.blackboxai import BLACKBOXAI