Skip to content

Commit 6c67545

Browse files
committed
Fix formatting
1 parent 87708d6 commit 6c67545

4 files changed

Lines changed: 8 additions & 10 deletions

File tree

src/app/endpoints/query.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ async def query_endpoint_handler(
158158
inline_rag_context = await build_rag_context(
159159
client, query_request.query, query_request.vector_store_ids, query_request.solr
160160
)
161-
161+
162162
# Prepare API request parameters
163163
responses_params = await prepare_responses_params(
164164
client,

src/llama_stack_configuration.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,9 @@ def construct_vector_stores_section(
190190
if not brag.get("rag_id"):
191191
raise ValueError(f"BYOK RAG entry is missing required 'rag_id': {brag}")
192192
if not brag.get("vector_db_id"):
193-
raise ValueError(f"BYOK RAG entry is missing required 'vector_db_id': {brag}")
193+
raise ValueError(
194+
f"BYOK RAG entry is missing required 'vector_db_id': {brag}"
195+
)
194196
rag_id = brag["rag_id"]
195197
vector_db_id = brag["vector_db_id"]
196198
if vector_db_id in existing_store_ids:

src/utils/vector_search.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -492,12 +492,8 @@ async def build_rag_context(
492492
RAGContext containing formatted context text and referenced documents
493493
"""
494494
# Fetch from all enabled RAG sources in parallel
495-
byok_chunks_task = _fetch_byok_rag(
496-
client, query, vector_store_ids
497-
)
498-
solr_chunks_task = _fetch_solr_rag(
499-
client, query, solr
500-
)
495+
byok_chunks_task = _fetch_byok_rag(client, query, vector_store_ids)
496+
solr_chunks_task = _fetch_solr_rag(client, query, solr)
501497

502498
(byok_chunks, byok_docs), (solr_chunks, solr_docs) = await asyncio.gather(
503499
byok_chunks_task, solr_chunks_task

tests/unit/utils/test_responses.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -353,7 +353,7 @@ class TestGetRAGTools:
353353

354354
def test_get_rag_tools_empty_list(self) -> None:
355355
"""Test get_rag_tools returns empty list for empty vector store IDs."""
356-
assert get_rag_tools([]) == []
356+
assert not get_rag_tools([])
357357

358358
def test_get_rag_tools_with_vector_stores(self) -> None:
359359
"""Test get_rag_tools returns correct tool format for vector stores."""
@@ -2312,7 +2312,7 @@ def test_returns_none_when_tool_rag_disabled(self, mocker: MockerFixture) -> Non
23122312
mock_config.rag.tool.byok.enabled = False
23132313
mocker.patch("utils.responses.configuration", mock_config)
23142314

2315-
assert get_rag_tools(["vs1", "vs2"]) == []
2315+
assert not get_rag_tools(["vs1", "vs2"])
23162316

23172317
def test_returns_tools_when_enabled(self, mocker: MockerFixture) -> None:
23182318
"""Test get_rag_tools returns tools when Tool RAG is enabled in config."""

0 commit comments

Comments
 (0)