mirror of
https://github.com/arc53/DocsGPT.git
synced 2025-11-29 08:33:20 +00:00
13 lines
320 B
Python
13 lines
320 B
Python
import unittest
|
|
from unittest.mock import patch
|
|
from application.llm.openai import OpenAILLM
|
|
|
|
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)
|