Nottybro commited on
Commit
c837482
·
verified ·
1 Parent(s): b0344af

fix: add cost field to QueryResponse

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,7 @@ class QueryResponse(BaseModel):
43
  credits_remaining: int
44
  complexity: Optional[dict] = None
45
  retrieval_source: Optional[str] = None
 
46
 
47
  @app.get("/")
48
  def root(): return {"status": "ACRA API is live \U0001F680", "docs": "/docs"}
@@ -79,6 +80,7 @@ async def query(body: QueryRequest,
79
  credits_remaining = remaining,
80
  complexity = result.get("complexity"),
81
  retrieval_source = result.get("retrieval_source"),
 
82
  )
83
 
84
  @app.get("/v1/usage")
 
43
  credits_remaining: int
44
  complexity: Optional[dict] = None
45
  retrieval_source: Optional[str] = None
46
+ cost: Optional[dict] = None
47
 
48
  @app.get("/")
49
  def root(): return {"status": "ACRA API is live \U0001F680", "docs": "/docs"}
 
80
  credits_remaining = remaining,
81
  complexity = result.get("complexity"),
82
  retrieval_source = result.get("retrieval_source"),
83
+ cost = result.get("cost"),
84
  )
85
 
86
  @app.get("/v1/usage")