apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-28 01:46:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tests are failing (#534)

Browse source 
* Fix tests, update to new model parameter usage
This commit is contained in:
cybermaggedon 2025-09-25 21:32:19 +01:00 • committed by GitHub
parent 6f4f7ce6b4
commit b0a3716b0e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 66 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/integration/test_text_completion_integration.py
View file

@ -60,13 +60,13 @@ class TestTextCompletionIntegration:
"""Create text completion processor with test configuration"""
# Create a minimal processor instance for testing generate_content
processor = MagicMock()
processor.model = processor_config["model"]
processor.default_model = processor_config["model"]
processor.temperature = processor_config["temperature"]
processor.max_output = processor_config["max_output"]
# Add the actual generate_content method from Processor class
processor.generate_content = Processor.generate_content.__get__(processor, Processor)
return processor
@pytest.mark.asyncio
@ -112,11 +112,11 @@ class TestTextCompletionIntegration:
for config in test_configs:
# Arrange - Create minimal processor mock
processor = MagicMock()
processor.model = config['model']
processor.default_model = config['model']
processor.temperature = config['temperature']
processor.max_output = config['max_output']
processor.openai = mock_openai_client
# Add the actual generate_content method
processor.generate_content = Processor.generate_content.__get__(processor, Processor)
@ -242,7 +242,7 @@ class TestTextCompletionIntegration:
processors = []
for i in range(5):
processor = MagicMock()
processor.model = processor_config["model"]
processor.default_model = processor_config["model"]
processor.temperature = processor_config["temperature"]
processor.max_output = processor_config["max_output"]
processor.openai = mock_openai_client
@ -348,7 +348,7 @@ class TestTextCompletionIntegration:
"""Test that model parameters are correctly passed to OpenAI API"""
# Arrange
processor = MagicMock()
processor.model = "gpt-4"
processor.default_model = "gpt-4"
processor.temperature = 0.8
processor.max_output = 2048
processor.openai = mock_openai_client