apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 04:31:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Some tests broken

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-26 00:56:25 +01:00
parent c2d0d22549
commit cfc92918ff
1 changed files with 19 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

42
tests/unit/test_base/test_llm_service_parameters.py
View file

@ -12,26 +12,26 @@ from trustgraph.base import ParameterSpec, ConsumerSpec, ProducerSpec
from trustgraph.schema import TextCompletionRequest, TextCompletionResponse from trustgraph.schema import TextCompletionRequest, TextCompletionResponse
class MockAsyncProcessor:
def __init__(self, **params):
self.config_handlers = []
self.id = params.get('id', 'test-service')
self.specifications = []
class TestLlmServiceParameters(IsolatedAsyncioTestCase): class TestLlmServiceParameters(IsolatedAsyncioTestCase):
"""Test LLM service parameter specification functionality""" """Test LLM service parameter specification functionality"""
@patch('trustgraph.base.async_processor.AsyncProcessor.__init__') @patch('trustgraph.base.async_processor.AsyncProcessor', MockAsyncProcessor)
def test_parameter_specs_registration(self, mock_async_init): def test_parameter_specs_registration(self):
"""Test that LLM service registers model and temperature parameter specs""" """Test that LLM service registers model and temperature parameter specs"""
# Mock AsyncProcessor.__init__ to set required attributes
def mock_init(self, **params):
self.config_handlers = []
self.id = params.get('id', 'test-service')
self.specifications = []
mock_async_init.side_effect = mock_init
# Arrange # Arrange
config = { config = {
'id': 'test-llm-service', 'id': 'test-llm-service',
'concurrency': 1 'concurrency': 1,
'taskgroup': AsyncMock() # Add required taskgroup
} }
# Act # Act
@ -45,20 +45,18 @@ class TestLlmServiceParameters(IsolatedAsyncioTestCase):
assert "temperature" in param_specs assert "temperature" in param_specs
assert len(param_specs) >= 2 # May have other parameter specs assert len(param_specs) >= 2 # May have other parameter specs
@patch('trustgraph.base.async_processor.AsyncProcessor.__init__', return_value=None) @patch('trustgraph.base.async_processor.AsyncProcessor', MockAsyncProcessor)
def test_model_parameter_spec_properties(self, mock_async_init): def test_model_parameter_spec_properties(self):
"""Test that model parameter spec has correct properties""" """Test that model parameter spec has correct properties"""
# Arrange # Arrange
config = { config = {
'id': 'test-llm-service', 'id': 'test-llm-service',
'concurrency': 1 'concurrency': 1,
'taskgroup': AsyncMock()
} }
# Act # Act
service = LlmService(**config) service = LlmService(**config)
# Manually set required attributes that would be set by AsyncProcessor
service.config_handlers = []
service.id = config['id']
# Assert # Assert
model_spec = None model_spec = None
@ -70,20 +68,18 @@ class TestLlmServiceParameters(IsolatedAsyncioTestCase):
assert model_spec is not None assert model_spec is not None
assert model_spec.name == "model" assert model_spec.name == "model"
@patch('trustgraph.base.async_processor.AsyncProcessor.__init__', return_value=None) @patch('trustgraph.base.async_processor.AsyncProcessor', MockAsyncProcessor)
def test_temperature_parameter_spec_properties(self, mock_async_init): def test_temperature_parameter_spec_properties(self):
"""Test that temperature parameter spec has correct properties""" """Test that temperature parameter spec has correct properties"""
# Arrange # Arrange
config = { config = {
'id': 'test-llm-service', 'id': 'test-llm-service',
'concurrency': 1 'concurrency': 1,
'taskgroup': AsyncMock()
} }
# Act # Act
service = LlmService(**config) service = LlmService(**config)
# Manually set required attributes that would be set by AsyncProcessor
service.config_handlers = []
service.id = config['id']
# Assert # Assert
temperature_spec = None temperature_spec = None