cf93d8d6e2
Co-authored-by: jyong <718720800@qq.com> Co-authored-by: StyleZhang <jasonapring2015@outlook.com>
25 lines
731 B
Python
25 lines
731 B
Python
import logging
|
|
import time
|
|
|
|
import click
|
|
import requests
|
|
from celery import shared_task
|
|
|
|
from core.generator.llm_generator import LLMGenerator
|
|
|
|
|
|
@shared_task
|
|
def generate_test_task():
|
|
logging.info(click.style('Start generate test', fg='green'))
|
|
start_at = time.perf_counter()
|
|
|
|
try:
|
|
#res = requests.post('https://api.openai.com/v1/chat/completions')
|
|
answer = LLMGenerator.generate_conversation_name('84b2202c-c359-46b7-a810-bce50feaa4d1', 'avb', 'ccc')
|
|
print(f'answer: {answer}')
|
|
|
|
end_at = time.perf_counter()
|
|
logging.info(click.style('Conversation test, latency: {}'.format(end_at - start_at), fg='green'))
|
|
except Exception:
|
|
logging.exception("generate test failed")
|