From 13f4ff0e6d8b8ab7b2fe6b1abc921f631b57c5de Mon Sep 17 00:00:00 2001 From: Pao-Sheng Wang Date: Thu, 9 Jan 2025 13:59:06 +0800 Subject: [PATCH] chore: rename test case, and correct pipeline type --- wren-ai-service/src/pipelines/indexing/sql_pairs.py | 2 +- .../tests/pytest/pipelines/indexing/test_sql_pairs.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/wren-ai-service/src/pipelines/indexing/sql_pairs.py b/wren-ai-service/src/pipelines/indexing/sql_pairs.py index c5c322d30..d4bbf9136 100644 --- a/wren-ai-service/src/pipelines/indexing/sql_pairs.py +++ b/wren-ai-service/src/pipelines/indexing/sql_pairs.py @@ -65,7 +65,7 @@ def boilerplates( @observe(capture_input=False) def sql_pairs( - boilerplates: List[str], + boilerplates: Set[str], external_pairs: Dict[str, Any], ) -> List[SqlPair]: return [ diff --git a/wren-ai-service/tests/pytest/pipelines/indexing/test_sql_pairs.py b/wren-ai-service/tests/pytest/pipelines/indexing/test_sql_pairs.py index bbce89413..f658abcf5 100644 --- a/wren-ai-service/tests/pytest/pipelines/indexing/test_sql_pairs.py +++ b/wren-ai-service/tests/pytest/pipelines/indexing/test_sql_pairs.py @@ -35,7 +35,7 @@ async def test_sql_pairs_indexing_saving_to_document_store(): @pytest.mark.asyncio -async def test_sql_pairs_preparation_saving_to_document_store_with_multiple_project_ids(): +async def test_sql_pairs_indexing_saving_to_document_store_with_multiple_project_ids(): pipe_components = generate_components(settings.components) document_store_provider: DocumentStoreProvider = pipe_components[ "sql_pairs_indexing"