Upload 61 files

app/api/routes/ingest.py

@@ -46,12 +46,14 @@ def search_wikipedia(q: str, limit: int = 10):
 async def import_from_wikipedia(
     title: str,
     entity_type: str = "person",
+    project_id: Optional[str] = None,
     auto_extract: bool = True,
     db: Session = Depends(get_db)
 ):
     """
     Importa uma entidade da Wikipedia
     entity_type: person, organization, location
+    project_id: ID do projeto para associar a entidade
     auto_extract: Se True, usa LLM para extrair entidades relacionadas
     """
     # Check if entity already exists
@@ -76,8 +78,9 @@ async def import_from_wikipedia(
     if not data:
         raise HTTPException(status_code=404, detail="Article not found on Wikipedia")
     
-    # Create main entity
+    # Create main entity with project_id
     entity = Entity(**data)
+    entity.project_id = project_id
     db.add(entity)
     db.commit()
     db.refresh(entity)
@@ -123,6 +126,7 @@ async def import_from_wikipedia(
                         latitude=lat,
                         longitude=lng,
                         event_date=event_date,
+                        project_id=project_id,
                         properties={"role": ext_entity.role, "aliases": ext_entity.aliases, "extracted_from": title}
                     )
                     db.add(new_ent)

app/main.py

@@ -8,7 +8,7 @@ from contextlib import asynccontextmanager
 
 from app.config import settings
 from app.core.database import init_db
-from app.api.routes import entities, relationships, events, search, ingest, analyze, graph, research, chat, investigate, dados_publicos, timeline, projects
+from app.api.routes import entities, relationships, events, search, ingest, analyze, graph, research, chat, investigate, dados_publicos, timeline
 
 
 @asynccontextmanager
@@ -62,7 +62,6 @@ app.include_router(chat.router, prefix="/api/v1")
 app.include_router(investigate.router, prefix="/api/v1")
 app.include_router(dados_publicos.router, prefix="/api/v1")
 app.include_router(timeline.router, prefix="/api/v1")
-app.include_router(projects.router, prefix="/api/v1")
 
 
 @app.get("/")