Skip to content

Commit ffe6c71

Browse files
[autofix.ci] apply automated fixes
1 parent 0292c5b commit ffe6c71

2 files changed

Lines changed: 4 additions & 12 deletions

File tree

src/backend/tests/unit/components/files_and_knowledge/test_retrieval.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -525,9 +525,7 @@ async def test_resolve_provider_variables_empty_provider_vars(self, component_cl
525525

526526
assert result == {}
527527

528-
async def test_resolve_provider_variables_variable_service_returns_none(
529-
self, component_class, default_kwargs
530-
):
528+
async def test_resolve_provider_variables_variable_service_returns_none(self, component_class, default_kwargs):
531529
"""Test _resolve_provider_variables when variable_service is None."""
532530
component = component_class(**default_kwargs)
533531

@@ -784,9 +782,7 @@ def test_build_embeddings_watsonx_api_key_from_provider_vars(
784782
assert result == mock_embeddings
785783

786784
@patch("langchain_ibm.WatsonxEmbeddings")
787-
def test_build_embeddings_watsonx_partial_vars(
788-
self, mock_watsonx_embeddings, component_class, default_kwargs
789-
):
785+
def test_build_embeddings_watsonx_partial_vars(self, mock_watsonx_embeddings, component_class, default_kwargs):
790786
"""Test WatsonX with only apikey, no project_id or url."""
791787
component = component_class(**default_kwargs)
792788

@@ -896,9 +892,7 @@ async def test_retrieve_data_with_search_query(self, component_class, default_kw
896892
result = await component.retrieve_data()
897893

898894
# Verify similarity_search_with_score was used (not similarity_search)
899-
mock_chroma_instance.similarity_search_with_score.assert_called_once_with(
900-
query="find me something", k=5
901-
)
895+
mock_chroma_instance.similarity_search_with_score.assert_called_once_with(query="find me something", k=5)
902896
mock_chroma_instance.similarity_search.assert_not_called()
903897
assert len(result) == 1
904898

src/lfx/src/lfx/components/files_and_knowledge/retrieval.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -188,9 +188,7 @@ async def _resolve_provider_variables(self, provider: str) -> dict[str, str]:
188188
if value and str(value).strip():
189189
result[var_key] = str(value)
190190
except (ValueError, KeyError, AttributeError) as e:
191-
logger.debug(
192-
f"Variable service lookup failed for '{var_key}', falling back to environment: {e}"
193-
)
191+
logger.debug(f"Variable service lookup failed for '{var_key}', falling back to environment: {e}")
194192
env_value = os.environ.get(var_key)
195193
if env_value and env_value.strip():
196194
result[var_key] = env_value

0 commit comments

Comments
 (0)