DocsGPT/tests/llm/test_openai.py

12 lines
288 B
Python
Raw Normal View History

2023-09-27 16:54:57 +00:00
import unittest
2023-09-27 17:10:26 +00:00
from application.llm.openai import OpenAILLM
2023-09-27 16:54:57 +00:00
class TestOpenAILLM(unittest.TestCase):
def setUp(self):
self.api_key = "test_api_key"
self.llm = OpenAILLM(self.api_key)
def test_init(self):
self.assertEqual(self.llm.api_key, self.api_key)