andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Add ground_truth API and refactor

Browse source
This commit is contained in:
Andras Schmelczer 2022-06-19 17:15:52 +02:00
commit 421f9bb726
1 changed files with 409 additions and 211 deletions
Download patch file Download diff file Expand all files Collapse all files

6
great_ai/src/great_ai/great_ai/deploy/routes/bootstrap_trace_endpoints.py
View file

@ -3,7 +3,7 @@ from typing import List
from fastapi import APIRouter, FastAPI, HTTPException, Response, status
from ...context import get_context
from ...views import Query, Trace, TraceView
from ...views import Query, Trace
def bootstrap_trace_endpoints(app: FastAPI) -> None:
@ -12,7 +12,7 @@ def bootstrap_trace_endpoints(app: FastAPI) -> None:
tags=["traces"],
)
@router.post("/", status_code=status.HTTP_200_OK, response_model=List[TraceView])
@router.post("/", status_code=status.HTTP_200_OK, response_model=List[Trace])
def query_traces(
query: Query,
skip: int = 0,
@ -25,7 +25,7 @@ def bootstrap_trace_endpoints(app: FastAPI) -> None:
take=take,
)[0]
@router.get("/{trace_id}", status_code=status.HTTP_200_OK, response_model=TraceView)
@router.get("/{trace_id}", status_code=status.HTTP_200_OK, response_model=Trace)
def get_trace(trace_id: str) -> Trace:
result = get_context().tracing_database.get(trace_id)
if result is None: