dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 17:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

roughed in openai 1.x

Browse source
This commit is contained in:
Paul Gauthier 2023-12-05 07:37:05 -08:00
parent fd34766aa9
commit 6ebc142377
15 changed files with 136 additions and 110 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_models.py
View file

@ -24,12 +24,13 @@ class TestModels(unittest.TestCase):
model = Model.create("gpt-4-32k-2123")
self.assertEqual(model.max_context_tokens, 32 * 1024)
@patch("openai.Model.list")
@patch("openai.resources.Models.list")
def test_openrouter_model_properties(self, mock_model_list):
import openai
# import openai
old_base = openai.api_base
openai.api_base = "https://openrouter.ai/api/v1"
# old_base = openai.api_base
# TODO: fixme
# openai.api_base = "https://openrouter.ai/api/v1"
mock_model_list.return_value = {
"data": [
{
@ -49,7 +50,8 @@ class TestModels(unittest.TestCase):
self.assertEqual(model.max_context_tokens, 8192)
self.assertEqual(model.prompt_price, 0.06)
self.assertEqual(model.completion_price, 0.12)
openai.api_base = old_base
# TODO: fixme
# openai.api_base = old_base
if __name__ == "__main__":