Updated tests

This commit is contained in:
Cyber MacGeddon 2025-07-11 16:43:27 +01:00
parent 4ff85cd6bd
commit d25c2f4d10
40 changed files with 301 additions and 485 deletions

View file

@ -4,8 +4,8 @@ Pytest configuration and fixtures for text completion tests
import pytest
from unittest.mock import MagicMock, AsyncMock
from trustgraph.base.types import TextCompletionRequest, TextCompletionResponse, LlmResult
from trustgraph.schema import TextCompletionRequest, TextCompletionResponse
from trustgraph.base import LlmResult
@pytest.fixture
def mock_vertexai_credentials():
@ -156,4 +156,4 @@ def mock_async_context_manager():
async def __aexit__(self, exc_type, exc_val, exc_tb):
pass
return MockAsyncContextManager
return MockAsyncContextManager