/
githubmirror
/
DemoGPT
Обзор
Документация
Войти
/
githubmirror
/
DemoGPT
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
tests/test_llms.py
18 строк
644 B
melih
llm tests are implemented
04 ноя 2024, 23:44
04 ноя 2024, 23:44
ecf2514
Код
Авторство
О чём код?
import unittest from demogpt_agenthub.llms.openai import OpenAIModel, OpenAIChatModel class TestOpenAIModel(unittest.TestCase): def setUp(self): self.model = OpenAIModel(model="gpt-3.5-turbo-instruct") def test_correct_response(self): result = self.model.run("What is the capital of Turkey?") self.assertIn("Ankara", result) class TestOpenAIChatModel(unittest.TestCase): def setUp(self): self.model = OpenAIChatModel(model="gpt-4o-mini") def test_correct_response(self): result = self.model.run("What is the capital of Turkey?") self.assertIn("Ankara", result)