import json
import logging
import time
import uuid

import fastapi
import jieba
from fastapi import APIRouter
from fastapi import HTTPException
from openai import AsyncOpenAI
from sse_starlette.sse import EventSourceResponse

from Config import Config
from ElasticSearch.Utils.EsSearchUtil import EsSearchUtil

# 创建路由路由器
router = APIRouter(prefix="/api/jimeng", tags=["即梦"])

# 配置日志
logger = logging.getLogger(__name__)

# 初始化异步 OpenAI 客户端
client = AsyncOpenAI(
    api_key=Config.ALY_LLM_API_KEY,
    base_url=Config.ALY_LLM_BASE_URL
)


@router.post("/prompt_input")
async def prompt_input(request: fastapi.Request):
    data = await request.json()
    pass