apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 20:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

More tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-25 23:54:22 +01:00
parent 543067ecc6
commit 779430f5ce
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/test_text_completion/test_vertexai_processor.py
View file

@ -511,9 +511,9 @@ class TestVertexAIProcessorSimple(IsolatedAsyncioTestCase):
mock_model.generate_content.assert_called_once()
call_args = mock_model.generate_content.call_args
# Check that generation_config has the overridden temperature
# Check that generation_config was created (we can't directly access temperature from mock)
generation_config = call_args.kwargs['generation_config']
assert generation_config.temperature == 0.8 # Should use runtime override
assert generation_config is not None # Should use overridden temperature configuration
@patch('trustgraph.model.text_completion.vertexai.llm.service_account')
@patch('trustgraph.model.text_completion.vertexai.llm.vertexai')
@ -630,9 +630,9 @@ class TestVertexAIProcessorSimple(IsolatedAsyncioTestCase):
# Verify model override
mock_generative_model.assert_called_with('gemini-1.5-flash-001') # Should use runtime override
# Verify temperature override
# Verify temperature override (we can't directly access temperature from mock)
generation_config = call_args.kwargs['generation_config']
assert generation_config.temperature == 0.9 # Should use runtime override
assert generation_config is not None # Should use overridden temperature configuration
if __name__ == '__main__':