aljaz/nostrdvm
1
0
Fork 0
mirror of https://github.com/aljazceru/nostrdvm.git synced 2025-12-24 01:14:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

add new summarization dvm, be more generous with generic dvms

Browse Source
This commit is contained in:
Believethehype
2024-08-29 16:00:38 +02:00
parent 42319ab751
commit e483ba4d28
5 changed files with 210 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
nostr_dvm/utils/backend_utils.py
View File

@@ -135,8 +135,8 @@ async def check_task_is_supported(event: Event, client, config=None):
# See if current dvm supports the task
task = await get_task(event, client=client, dvm_config=dvm_config)
if task not in (x.TASK for x in dvm_config.SUPPORTED_DVMS):
return False, task
# if task not in (x.TASK for x in dvm_config.SUPPORTED_DVMS) and not task == "generic":
# return False, task
# See if current dvm can handle input for given task
for dvm in dvm_config.SUPPORTED_DVMS:
if dvm.TASK == task:
@@ -144,7 +144,6 @@ async def check_task_is_supported(event: Event, client, config=None):
return False, task
return True, task
except Exception as e:
print("Check task: " + str(e))
@@ -185,7 +184,7 @@ def check_url_is_readable(url):
def get_amount_per_task(task, dvm_config, duration=1):
# duration is either static 1 (for images etc) or in seconds by default (e.g. audio/video)
for dvm in dvm_config.SUPPORTED_DVMS: # this is currently just one
if dvm.TASK == task:
if dvm.TASK == task or dvm.TASK == "generic":
amount = dvm.FIX_COST + (dvm.PER_UNIT_COST * duration)
return amount
else: