Mise à jour ingestion et traitement RNE/JORT

app/api/v1/companies.py

@@ -34,7 +34,7 @@ async def list_companies(
     return clean_nans(data)
 
 @router.get("/{company_id}", response_model=CompanyWithLinks)
-async def read_company(company_id: int):
+async def read_company(company_id: str):
     df = await get_companies_df()
     company = df[df['id'] == company_id]
     if company.empty:
@@ -46,6 +46,6 @@ async def read_company(company_id: int):
     return clean_nans(data)
 
 @router.get("/{company_id}/osint_links")
-async def read_company_links(company_id: int):
+async def read_company_links(company_id: str):
     return await get_company_links(company_id)
 

app/models/schemas.py

@@ -2,7 +2,7 @@ from pydantic import BaseModel
 from typing import List, Optional, Dict, Any
 
 class Company(BaseModel):
-    id: Optional[int] = None # Generated ID
+    id: Optional[str] = None # Generated ID
     name: str
     wilaya: str
     delegation: Optional[str] = None

app/services/data_loader.py

@@ -95,6 +95,9 @@ class DataLoader:
             if 'id' not in df.columns:
                 df['id'] = range(1, len(df) + 1)
             
+            # Ensure id is always string for consistent matching
+            df['id'] = df['id'].astype(str)
+            
             # Recalculate capital divergence if not already handled by SQL
             # (In our current SQL view, we could add this, but keeping it here for safety for now)
             threshold = float(os.getenv("CAPITAL_DIVERGENCE_THRESHOLD", 0.05))

app/services/osint_links.py

@@ -20,7 +20,7 @@ def generate_links(company_name: str, wilaya: str):
         
     return links
 
-async def get_company_links(company_id: int):
+async def get_company_links(company_id: str):
     from app.services.data_loader import get_companies_df
     df = await get_companies_df()