@@ -64,7 +64,7 @@ def mock_uploaded_file():
6464class TestDocumentUploadIntegration :
6565 """Integration tests for document upload workflow"""
6666
67- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
67+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
6868 def test_single_file_upload_workflow (self , mock_api , mock_uploaded_file ):
6969 """Test complete workflow for uploading a single file"""
7070 from llama_stack_client import RAGDocument
@@ -113,7 +113,7 @@ def test_single_file_upload_workflow(self, mock_api, mock_uploaded_file):
113113 mock_api .client .vector_dbs .register .assert_called_once ()
114114 mock_api .client .tool_runtime .rag_tool .insert .assert_called_once ()
115115
116- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
116+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
117117 def test_multiple_files_upload_workflow (self , mock_api ):
118118 """Test uploading multiple files at once"""
119119 from llama_stack_client import RAGDocument
@@ -182,7 +182,7 @@ def test_file_type_validation(self):
182182class TestVectorDBCreation :
183183 """Integration tests for vector database creation"""
184184
185- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
185+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
186186 def test_vector_db_registration_params (self , mock_api ):
187187 """Test vector DB registration with correct parameters"""
188188 mock_api .client .providers .list .return_value = [
@@ -204,7 +204,7 @@ def test_vector_db_registration_params(self, mock_api):
204204 assert call_args [1 ]['embedding_model' ] == "all-MiniLM-L6-v2"
205205 assert call_args [1 ]['provider_id' ] == "pgvector"
206206
207- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
207+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
208208 def test_vector_db_with_custom_name (self , mock_api ):
209209 """Test creating vector DB with custom name"""
210210 mock_api .client .providers .list .return_value = [
@@ -227,7 +227,7 @@ def test_vector_db_with_custom_name(self, mock_api):
227227class TestProviderDetection :
228228 """Integration tests for provider detection"""
229229
230- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
230+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
231231 def test_vector_io_provider_detection (self , mock_api ):
232232 """Test that vector_io provider is correctly detected"""
233233 mock_api .client .providers .list .return_value = [
@@ -244,7 +244,7 @@ def test_vector_io_provider_detection(self, mock_api):
244244
245245 assert vector_io_provider == "pgvector"
246246
247- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
247+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
248248 def test_no_vector_io_provider (self , mock_api ):
249249 """Test handling when no vector_io provider is available"""
250250 mock_api .client .providers .list .return_value = [
@@ -264,7 +264,7 @@ def test_no_vector_io_provider(self, mock_api):
264264class TestDocumentInsertion :
265265 """Integration tests for document insertion into vector DB"""
266266
267- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
267+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
268268 def test_document_insertion_with_chunks (self , mock_api ):
269269 """Test document insertion with chunking"""
270270 from llama_stack_client import RAGDocument
@@ -288,7 +288,7 @@ def test_document_insertion_with_chunks(self, mock_api):
288288 call_args = mock_api .client .tool_runtime .rag_tool .insert .call_args
289289 assert call_args [1 ]['chunk_size_in_tokens' ] == 512
290290
291- @patch ('llama_stack_ui.distribution.ui.page.upload.upload .llama_stack_api' )
291+ @patch ('llama_stack_ui.distribution.ui.modules.api .llama_stack_api' )
292292 def test_empty_document_list (self , mock_api ):
293293 """Test handling of empty document list"""
294294 documents = []
0 commit comments