34 lines
736 B
Python
34 lines
736 B
Python
|
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
|
||
|
|
|