fix

app.py

@@ -41,20 +41,39 @@ class GitHubMCPServer:
             
             print(f"📊 Loaded {len(dataset)} CVE records from Hugging Face")
             
+            # Debug: Print first few records to understand dataset structure
+            print("🔍 Dataset structure analysis:")
+            for i in range(min(3, len(dataset))):
+                print(f"Record {i}: {dict(dataset[i])}")
+            
             # Create documents from CVE data
             documents = []
             for idx, record in enumerate(dataset):
-                # Extract relevant fields from the dataset
-                cve_id = record.get('cve_id', f'CVE-{idx}')
-                cwe_code = record.get('cwe_code', 'Unknown')
-                cwe_name = record.get('cwe_name', 'Unknown')
-                cvss_score = record.get('cvss_score', record.get('cvss', 'N/A'))
-                summary = record.get('summary', record.get('description', 'No summary available'))
+                # Extract relevant fields from the dataset - check multiple possible field names
+                cve_id = (record.get('cve_id') or 
+                         record.get('CVE_ID') or 
+                         record.get('id') or 
+                         record.get('cve') or 
+                         f'CVE-UNKNOWN-{idx}')
+                
+                cwe_code = record.get('cwe_code', record.get('CWE', 'Unknown'))
+                cwe_name = record.get('cwe_name', record.get('cwe_description', 'Unknown'))
+                cvss_score = record.get('cvss_score', record.get('cvss', record.get('CVSS', 'N/A')))
+                summary = (record.get('summary') or 
+                          record.get('description') or 
+                          record.get('Description') or 
+                          'No summary available')
                 
-                # Skip records without essential information
+                # Skip records without essential information or invalid CVE IDs
                 if not summary or summary == 'No summary available':
                     continue
                 
+                # Validate CVE ID format (should be CVE-YYYY-NNNNN)
+                import re
+                if not re.match(r'^CVE-\d{4}-\d+$', str(cve_id)):
+                    print(f"⚠️ Skipping invalid CVE ID: {cve_id}")
+                    continue
+                
                 # Create document content
                 content = f"""
 CVE ID: {cve_id}