Skip to content

Commit 56d8e06

Browse files
committed
Added type hints into tests/unit/utils/test_transcripts.py
1 parent ab4b92f commit 56d8e06

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

tests/unit/utils/test_transcripts.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from pytest_mock import MockerFixture
55

66
from configuration import AppConfig
7-
from models.requests import QueryRequest
7+
from models.requests import Attachment, QueryRequest
88

99
from utils.transcripts import (
1010
construct_transcripts_path,
@@ -13,7 +13,7 @@
1313
from utils.types import ToolCallSummary, TurnSummary
1414

1515

16-
def test_construct_transcripts_path(mocker: MockerFixture):
16+
def test_construct_transcripts_path(mocker: MockerFixture) -> None:
1717
"""Test the construct_transcripts_path function."""
1818

1919
config_dict = {
@@ -52,7 +52,7 @@ def test_construct_transcripts_path(mocker: MockerFixture):
5252
), "Path should be constructed correctly"
5353

5454

55-
def test_store_transcript(mocker: MockerFixture):
55+
def test_store_transcript(mocker: MockerFixture) -> None:
5656
"""Test the store_transcript function."""
5757

5858
mocker.patch("builtins.open", mocker.mock_open())
@@ -83,9 +83,9 @@ def test_store_transcript(mocker: MockerFixture):
8383
],
8484
)
8585
query_is_valid = True
86-
rag_chunks = []
86+
rag_chunks: list[dict] = []
8787
truncated = False
88-
attachments = []
88+
attachments: list[Attachment] = []
8989

9090
store_transcript(
9191
user_id,

0 commit comments

Comments
 (0)