adding tests for files api and vector stores

Signed-off-by: Francisco Javier Arceo <[email protected]>

Author: franciscojavierarceo

tests/integration/tool_runtime/test_rag_tool.py

@@ -183,7 +183,7 @@ def test_vector_db_insert_from_url_and_query(
     assert any("llama2" in chunk.content.lower() for chunk in response2.chunks)
 
 
-def test_rag_tool_openai_api_integration(client_with_empty_registry, embedding_model_id, embedding_dimension):
+def test_rag_tool_openai_apis(client_with_empty_registry, embedding_model_id, embedding_dimension):
     vector_db_id = "test_openai_vector_db"
 
     client_with_empty_registry.vector_dbs.register(
@@ -221,6 +221,16 @@ def test_rag_tool_openai_api_integration(client_with_empty_registry, embedding_m
         chunk_size_in_tokens=256,
     )
 
+    files_list = client_with_empty_registry.files.list()
+    assert len(files_list.data) >= len(documents), (
+        f"Expected at least {len(documents)} files, got {len(files_list.data)}"
+    )
+
+    vector_store_files = client_with_empty_registry.vector_io.openai_list_files_in_vector_store(
+        vector_store_id=actual_vector_db_id
+    )
+    assert len(vector_store_files.data) >= len(documents), f"Expected at least {len(documents)} files in vector store"
+
     response = client_with_empty_registry.tool_runtime.rag_tool.query(
         vector_db_ids=[actual_vector_db_id],
         content="Tell me about machine learning and deep learning",
@@ -231,8 +241,8 @@ def test_rag_tool_openai_api_integration(client_with_empty_registry, embedding_m
     assert "machine learning" in content_text or "deep learning" in content_text
 
 
-def test_rag_tool_error_resilience(client_with_empty_registry, embedding_model_id, embedding_dimension):
-    vector_db_id = "test_error_resilience"
+def test_rag_tool_exception_handling(client_with_empty_registry, embedding_model_id, embedding_dimension):
+    vector_db_id = "test_exception_handling"
 
     client_with_empty_registry.vector_dbs.register(
         vector_db_id=vector_db_id,