Upload 59 files

app/api/routes/analyze.py

@@ -91,6 +91,21 @@ async def analyze_text(request: AnalyzeRequest):
         
         db = next(get_db())
         
+        # Helper function to parse date strings
+        def parse_date(date_str):
+            if not date_str:
+                return None
+            from datetime import datetime
+            try:
+                # Try YYYY-MM-DD format
+                return datetime.strptime(date_str[:10], "%Y-%m-%d")
+            except:
+                try:
+                    # Try YYYY format
+                    return datetime.strptime(date_str[:4], "%Y")
+                except:
+                    return None
+        
         # Process entities
         for entity in result.entities:
             entity_data = ExtractedEntityResponse(
@@ -116,6 +131,9 @@ async def analyze_text(request: AnalyzeRequest):
                         if coords:
                             lat, lng = coords
                     
+                    # Parse event_date if available
+                    event_date = parse_date(getattr(entity, 'event_date', None))
+                    
                     # Create new entity
                     new_entity = Entity(
                         name=entity.name,
@@ -124,6 +142,7 @@ async def analyze_text(request: AnalyzeRequest):
                         source="llm_extraction",
                         latitude=lat,
                         longitude=lng,
+                        event_date=event_date,
                         properties={"role": entity.role, "aliases": entity.aliases}
                     )
                     db.add(new_entity)
@@ -166,10 +185,14 @@ async def analyze_text(request: AnalyzeRequest):
                     ).first()
                     
                     if not existing_rel:
+                        # Parse event_date if available
+                        rel_event_date = parse_date(getattr(rel, 'event_date', None))
+                        
                         new_rel = Relationship(
                             source_id=source_entity.id,
                             target_id=target_entity.id,
                             type=rel.relationship_type,
+                            event_date=rel_event_date,
                             properties={"context": rel.context}
                         )
                         db.add(new_rel)

app/api/routes/ingest.py

@@ -5,6 +5,7 @@ Endpoints para importar dados de fontes externas
 from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks
 from sqlalchemy.orm import Session
 from typing import Optional, List
+from datetime import datetime
 import asyncio
 
 from app.core.database import get_db
@@ -17,6 +18,21 @@ from app.services.geocoding import geocode
 router = APIRouter(prefix="/ingest", tags=["Data Ingestion"])
 
 
+def parse_event_date(date_str):
+    """Parse date string to datetime object"""
+    if not date_str:
+        return None
+    try:
+        # Try YYYY-MM-DD format
+        return datetime.strptime(date_str[:10], "%Y-%m-%d")
+    except:
+        try:
+            # Try YYYY format
+            return datetime.strptime(date_str[:4], "%Y")
+        except:
+            return None
+
+
 # ========== Wikipedia ==========
 
 @router.get("/wikipedia/search")
@@ -96,6 +112,9 @@ async def import_from_wikipedia(
                         if coords:
                             lat, lng = coords
                     
+                    # Parse event_date
+                    event_date = parse_event_date(getattr(ext_entity, 'event_date', None))
+                    
                     new_ent = Entity(
                         name=ext_entity.name,
                         type=ext_entity.type if ext_entity.type in ["person", "organization", "location", "event"] else "person",
@@ -103,6 +122,7 @@ async def import_from_wikipedia(
                         source="wikipedia_extraction",
                         latitude=lat,
                         longitude=lng,
+                        event_date=event_date,
                         properties={"role": ext_entity.role, "aliases": ext_entity.aliases, "extracted_from": title}
                     )
                     db.add(new_ent)
@@ -124,10 +144,14 @@ async def import_from_wikipedia(
                     ).first()
                     
                     if not existing_rel:
+                        # Parse relationship event_date
+                        rel_event_date = parse_event_date(getattr(rel, 'event_date', None))
+                        
                         new_rel = Relationship(
                             source_id=source_ent.id,
                             target_id=target_ent.id,
                             type=rel.relationship_type,
+                            event_date=rel_event_date,
                             properties={"context": rel.context, "extracted_from": title}
                         )
                         db.add(new_rel)