QuivrHQ/quivr
1
1
Fork 0
mirror of https://github.com/QuivrHQ/quivr.git synced 2024-08-17 10:20:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add pricing information to AssistantOutput class

Browse Source
This commit is contained in:
Stan Girard 2024-04-16 18:13:50 +02:00
parent 3c050aa0e9
commit 9ea3fc3584
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/modules/assistant/dto/outputs.py
View File

@ -82,9 +82,15 @@ class Outputs(BaseModel):
brain: Optional[OutputBrain] = None
class Pricing(BaseModel):
cost: int = 20
description: str = "Credits per use"
class AssistantOutput(BaseModel):
name: str
description: str
pricing: Optional[Pricing] = Pricing()
tags: Optional[List[str]] = []
input_description: str
output_description: str