mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-22 03:31:02 +02:00
Updated tests
This commit is contained in:
parent
4ff85cd6bd
commit
d25c2f4d10
40 changed files with 301 additions and 485 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue