new tool(test)

app.py

@@ -90,7 +90,7 @@ Summary: {summary}
             # Initialize BM25 retriever
             self.cve_retriever = BM25Retriever.from_documents(
                 processed_docs, 
-                k=10  # Return top 10 most relevant documents
+                k=3
             )
             
             print(f"✅ CVE Retriever initialized with {len(processed_docs)} document chunks")
@@ -243,6 +243,45 @@ Summary: {summary}
         except Exception as e:
             return f"❌ Error retrieving CVE information: {str(e)}"
     
+    def simple_cve_search(self, query: str, k: int = 3) -> str:
+        """Simple CVE search that returns only CVE IDs and descriptions for multi-agent workflow"""
+        if not self.cve_retriever:
+            return "❌ CVE retriever not properly initialized. Please check Hugging Face dataset access."
+        
+        try:
+            # Set retriever to return k results
+            original_k = self.cve_retriever.k
+            self.cve_retriever.k = k
+            
+            # Retrieve relevant documents
+            docs = self.cve_retriever.invoke(query)
+            
+            # Restore original k
+            self.cve_retriever.k = original_k
+            
+            if not docs:
+                return f"No relevant CVE information found for query: '{query}'"
+            
+            # Format simple results - just CVE ID and description
+            result = f"Top {len(docs)} CVE matches for '{query}':\n\n"
+            
+            for i, doc in enumerate(docs, 1):
+                metadata = doc.metadata
+                cve_id = metadata.get('cve_id', 'Unknown')
+                
+                # Extract summary from content
+                content_lines = doc.page_content.split('\n')
+                summary_line = next((line for line in content_lines if line.startswith('Summary:')), '')
+                summary = summary_line.replace('Summary: ', '').strip() if summary_line else 'No summary available'
+                
+                result += f"{i}. {cve_id}\n"
+                result += f"   {summary}\n\n"
+            
+            return result.strip()
+            
+        except Exception as e:
+            return f"❌ Error retrieving CVE information: {str(e)}"
+    
     def get_nvd_cve_details(self, cve_id: str) -> str:
         """
         Fetches detailed CVE information from NVD (National Vulnerability Database).
@@ -526,6 +565,17 @@ demo = gr.TabbedInterface(
             title="🔬 Smart CVE Analysis (RAG + NVD)",
             description="Automatically searches CVE database AND fetches detailed NVD information for top CVEs",
             api_name="search_and_fetch_cve_details"
+        ),
+        gr.Interface(
+            fn=github_server.simple_cve_search,
+            inputs=[
+                gr.Textbox(label="Vulnerability Query", placeholder="SQL injection, XSS, command injection, etc."),
+                gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Number of Results", info="Number of CVE matches to return")
+            ],
+            outputs=gr.Textbox(label="Simple CVE Search Results", lines=15),
+            title="🔍 Simple CVE Search",
+            description="Simple CVE search returning only CVE IDs and descriptions (for multi-agent workflow)",
+            api_name="simple_cve_search"
         )
     ],
     [
@@ -534,7 +584,8 @@ demo = gr.TabbedInterface(
         "Repository Scanner",
         "CVE Database",
         "NVD CVE Details",
-        "🔬 Smart CVE Analysis"
+        "🔬 Smart CVE Analysis",
+        "🔍 Simple CVE Search"
     ],
     title="🐙 GitHub MCP Server with CVE Knowledge Base & NVD Integration"
 )