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

feat: add test for --model-metadata-file option

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-15 14:45:13 -07:00
parent e0bd9881ea
commit ec836b99cc
1 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/basic/test_main.py
View file

@ -1,6 +1,7 @@
import os import os
import subprocess import subprocess
import tempfile import tempfile
import json
from io import StringIO from io import StringIO
from pathlib import Path from pathlib import Path
from unittest import TestCase from unittest import TestCase
@ -489,3 +490,22 @@ class TestMain(TestCase):
self.assertIn(real_external_file_path, coder.abs_read_only_fnames) self.assertIn(real_external_file_path, coder.abs_read_only_fnames)
finally: finally:
os.unlink(external_file_path) os.unlink(external_file_path)
def test_model_metadata_file(self):
with GitTemporaryDirectory():
metadata_file = Path(".aider.model.metadata.json")
metadata_content = {
"deepseek-chat": {
"max_input_tokens": 1234
}
}
metadata_file.write_text(json.dumps(metadata_content))
coder = main(
["--model", "deepseek-chat", "--model-metadata-file", str(metadata_file), "--exit", "--yes"],
input=DummyInput(),
output=DummyOutput(),
return_coder=True,
)
self.assertEqual(coder.main_model.info["max_input_tokens"], 1234)