Upload edgar_client.py

edgar_client.py

@@ -423,8 +423,11 @@ class EdgarDataClient:
             for i in range(len(form_types_list)):
                 form_type = form_types_list[i]
                 
-                # Filter by form type if specified
-                if form_types and form_type not in form_types:
+                # Normalize form type (e.g., "10-K/A" -> "10-K", "20-F/A" -> "20-F")
+                normalized_form_type = form_type.split('/')[0]
+                
+                # Filter by form type if specified (using normalized type)
+                if form_types and normalized_form_type not in form_types:
                     continue
                 
                 filing_date = filing_dates[i] if i < len(filing_dates) else ""
@@ -551,15 +554,19 @@ class EdgarDataClient:
                     # Extract year from filing_date (format: YYYY-MM-DD)
                     file_year = int(filing_date[:4]) if len(filing_date) >= 4 else 0
                     
+                    # Normalize form type (e.g., "10-K/A" -> "10-K") for consistent key matching
+                    normalized_form_type = form_type.split('/')[0]
+                    
                     # Store filing - allow filing_date year to differ from fiscal year
                     # (20-F/10-K for FY2024 may be filed in 2025)
                     # We'll match by fiscal year in the facts data instead
-                    key = f"{form_type}_{file_year}"
+                    # Use normalized form type in key to ensure amended forms (10-K/A) match
+                    key = f"{normalized_form_type}_{file_year}"
                     if key not in filings_map:
                         filings_map[key] = {
                             "accession_number": accession_number,
                             "primary_document": primary_document,
-                            "form_type": form_type,
+                            "form_type": form_type,  # Keep original form_type for reference
                             "filing_date": filing_date,
                             "file_year": file_year
                         }