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"