import { AIEvalClient } from "@evalgate/sdk";

const evalgate = AIEvalClient.init();

export async function POST(request: Request) {
  const { message } = await request.json();
  const traceId = `support-${Date.now()}`;

  const answer = await answerSupportQuestion(message);

  const trace = await evalgate.traces.create({
    name: "support-chat",
    traceId,
    metadata: {
      runtime: "nextjs",
      model: "configured-by-app",
    },
  });

  await evalgate.traces.createSpan(trace.id, {
    name: "answerSupportQuestion",
    spanId: `${traceId}-llm`,
    startTime: new Date().toISOString(),
    metadata: {
      inputLength: message.length,
      outputLength: answer.length,
    },
  });

  return Response.json({ answer, traceId });
}

from datetime import datetime, timezone

from evalgate_sdk import AIEvalClient
from evalgate_sdk.types import CreateSpanParams, CreateTraceParams
from fastapi import FastAPI

app = FastAPI()
evalgate = AIEvalClient.init()


@app.post("/support")
async def support(payload: dict):
    message = payload["message"]
    trace_id = f"support-{int(datetime.now(timezone.utc).timestamp())}"

    answer = await answer_support_question(message)

    trace = await evalgate.traces.create(
        CreateTraceParams(
            name="support-chat",
            trace_id=trace_id,
            metadata={"runtime": "fastapi", "model": "configured-by-app"},
        )
    )

    await evalgate.traces.create_span(
        trace.id,
        CreateSpanParams(
            name="answer_support_question",
            span_id=f"{trace_id}-llm",
            start_time=datetime.now(timezone.utc).isoformat(),
            metadata={
                "input_length": len(message),
                "output_length": len(answer),
            },
        ),
    )

    return {"answer": answer, "trace_id": trace_id}