fix: add cost field to QueryResponse
Browse files
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")
|