style: Format RAG files with ruff

Author: d3xvn

agents-core/vision_agents/core/rag/__init__.py

@@ -19,4 +19,3 @@
     "RAGDocumentAddedEvent",
     "RAGFileAddedEvent",
 ]
-

agents-core/vision_agents/core/rag/base.py

@@ -156,4 +156,3 @@ async def clear(self) -> None:
         Default implementation does nothing. Override if supported.
         """
         pass
-

agents-core/vision_agents/core/rag/events.py

@@ -53,6 +53,3 @@ class RAGFileAddedEvent(BaseEvent):
     file_path: str = ""
     file_id: Optional[str] = None
     metadata: dict[str, Any] = field(default_factory=dict)
-
-
-

agents-core/vision_agents/core/rag/types.py

@@ -86,4 +86,3 @@ def format_citation(self) -> str:
             return f"[doc:{self.document_id[:8]}]"
 
         return "[unknown source]"
-

examples/other_examples/gemini_rag_demo/gemini_rag_demo.py

@@ -157,7 +157,9 @@ async def on_file_added(event: RAGFileAddedEvent):
 
     @rag.events.subscribe
     async def on_retrieval_complete(event: RAGRetrievalCompleteEvent):
-        print(f"   🔎 Retrieved {event.result_count} results in {event.retrieval_time_ms:.0f}ms")
+        print(
+            f"   🔎 Retrieved {event.result_count} results in {event.retrieval_time_ms:.0f}ms"
+        )
 
     # Create and upload sample files
     print("\n📄 Creating sample documents...")
@@ -196,7 +198,9 @@ async def on_retrieval_complete(event: RAGRetrievalCompleteEvent):
         # The RAG provider automatically augments the query with relevant context
         response = await llm.simple_response(query)
 
-        print(f"💡 Answer: {response.text[:500]}{'...' if len(response.text) > 500 else ''}")
+        print(
+            f"💡 Answer: {response.text[:500]}{'...' if len(response.text) > 500 else ''}"
+        )
 
     # Cleanup
     print("\n" + "-" * 60)
@@ -215,4 +219,3 @@ async def on_retrieval_complete(event: RAGRetrievalCompleteEvent):
 
 if __name__ == "__main__":
     asyncio.run(main())
-

plugins/gemini/tests/test_gemini_rag.py

@@ -117,9 +117,7 @@ async def on_file_added(event: RAGFileAddedEvent):
             file_added_events.append(event)
 
         # Create a test file
-        with tempfile.NamedTemporaryFile(
-            mode="w", suffix=".txt", delete=False
-        ) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
             f.write(
                 "Vision Agents is a framework for building AI agents that can "
                 "process video and audio in real-time. It supports multiple LLM "
@@ -161,9 +159,7 @@ async def on_retrieval(event: RAGRetrievalCompleteEvent):
             retrieval_events.append(event)
 
         # Create test files
-        with tempfile.NamedTemporaryFile(
-            mode="w", suffix=".txt", delete=False
-        ) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
             f.write(
                 "Password Reset Instructions:\n"
                 "1. Go to the login page\n"
@@ -174,9 +170,7 @@ async def on_retrieval(event: RAGRetrievalCompleteEvent):
             )
             password_file = f.name
 
-        with tempfile.NamedTemporaryFile(
-            mode="w", suffix=".txt", delete=False
-        ) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
             f.write(
                 "Account Settings:\n"
                 "You can update your profile picture, display name, and "
@@ -251,9 +245,7 @@ async def test_llm_with_rag(self):
         llm = LLM(model="gemini-2.0-flash")
 
         # Create a test file with specific information
-        with tempfile.NamedTemporaryFile(
-            mode="w", suffix=".txt", delete=False
-        ) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
             f.write(
                 "Company Policy Document\n\n"
                 "Vacation Policy:\n"
@@ -295,9 +287,7 @@ async def test_get_file_search_tool(self):
             rag.get_file_search_tool()
 
         # Create a test file
-        with tempfile.NamedTemporaryFile(
-            mode="w", suffix=".txt", delete=False
-        ) as f:
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
             f.write("Test content for file search tool.")
             temp_path = f.name
 
@@ -360,4 +350,3 @@ async def test_build_context_prompt_no_citations(self):
 
         assert "Test passage" in prompt
         assert "[unknown source]" not in prompt
-

plugins/gemini/vision_agents/plugins/gemini/rag.py

@@ -226,7 +226,9 @@ async def add_file(self, file_path: str, metadata: Optional[dict] = None) -> str
             raise FileNotFoundError(f"File not found: {file_path}")
 
         mime_type = self._get_mime_type(file_path)
-        display_name = metadata.get("display_name", path.name) if metadata else path.name
+        display_name = (
+            metadata.get("display_name", path.name) if metadata else path.name
+        )
 
         # Build custom metadata if provided
         custom_metadata = None
@@ -408,4 +410,3 @@ def get_file_search_tool(self, top_k: Optional[int] = None) -> types.Tool:
                 top_k=top_k or self._top_k,
             )
         )
-

tests/test_rag.py

@@ -1,6 +1,5 @@
 """Tests for the core RAG module."""
 
-
 from vision_agents.core.rag import (
     Chunk,
     Document,
@@ -370,4 +369,3 @@ async def test_delete_document(self):
         # Try to delete again
         deleted_again = await provider.delete_document("test-doc")
         assert deleted_again is False
-