import os

from fastapi import FastAPI

from fastapi.middleware.cors import CORSMiddleware

from pydantic import BaseModel

from core.agents import agent_network

from tracing.logger import execution_traces

app = FastAPI(title="Autonomous AI Agent Network API")

app.add_middleware(

CORSMiddleware,

allow_origins=["*"],

allow_credentials=True,

allow_methods=["*"],

allow_headers=["*"],

)

class QueryRequest(BaseModel):

query: str

@app.post("/api/run")

async def run_network(payload: QueryRequest):

execution_traces.clear()

initial_state = {

"query": payload.query,

"research_notes": "",

"data_analysis": "",

"final_report": "",

"logs": []

}

output = agent_network.invoke(initial_state)

return {

"status": "SUCCESS",

"final_report": output["final_report"],

"traces": execution_traces

}

@app.get("/api/traces")

async def get_traces():

return {"traces": execution_traces}