This commit is contained in:
Timothy Jaeryang Baek 2024-12-01 00:23:06 -08:00
parent 368e11e2b2
commit 39c2f70778
2 changed files with 2 additions and 2 deletions

View File

@ -113,6 +113,6 @@ class TASKS(str, Enum):
TAGS_GENERATION = "tags_generation" TAGS_GENERATION = "tags_generation"
EMOJI_GENERATION = "emoji_generation" EMOJI_GENERATION = "emoji_generation"
QUERY_GENERATION = "query_generation" QUERY_GENERATION = "query_generation"
AUTOCOMPLETION_GENERATION = "autocompletion_generation" AUTOCOMPLETE_GENERATION = "autocomplete_generation"
FUNCTION_CALLING = "function_calling" FUNCTION_CALLING = "function_calling"
MOA_RESPONSE_GENERATION = "moa_response_generation" MOA_RESPONSE_GENERATION = "moa_response_generation"

View File

@ -2066,7 +2066,7 @@ async def generate_autocompletion(form_data: dict, user=Depends(get_verified_use
"messages": [{"role": "user", "content": content}], "messages": [{"role": "user", "content": content}],
"stream": False, "stream": False,
"metadata": { "metadata": {
"task": str(TASKS.AUTOCOMPLETION_GENERATION), "task": str(TASKS.AUTOCOMPLETE_GENERATION),
"task_body": form_data, "task_body": form_data,
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
}, },