Skip to content

Commit 95daff3

Browse files
committed
Add test for gpt4all
1 parent 61af271 commit 95daff3

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/test_gpt4all.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
import unittest
2+
import tests.base as base
3+
from pytgpt.gpt4all import GPT4ALL
4+
from os import getenv
5+
6+
7+
class TestGpt4all(base.llmBase):
8+
def setUp(self):
9+
self.bot = GPT4ALL(getenv("model_path"))
10+
self.prompt = base.prompt
11+
12+
13+
if __name__ == "__main__":
14+
unittest.main()

0 commit comments

Comments
 (0)