'commit'
This commit is contained in:
33
dsLightRag/Routes/JiMeng.py
Normal file
33
dsLightRag/Routes/JiMeng.py
Normal file
@@ -0,0 +1,33 @@
|
||||
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
|
||||
|
@@ -19,6 +19,7 @@ from Routes.TeachingModel.api.ThemeController import router as theme_router
|
||||
from Routes.TeachingModel.api.DocumentController import router as document_router
|
||||
from Routes.TeachingModel.api.TeachingModelController import router as teaching_model_router
|
||||
from Routes.QA import router as qa_router
|
||||
from Routes.JiMeng import router as jimeng_router
|
||||
|
||||
from Util.LightRagUtil import *
|
||||
from contextlib import asynccontextmanager
|
||||
@@ -56,6 +57,7 @@ app.include_router(knowledge_router) # 知识图谱路由
|
||||
app.include_router(oss_router) # 阿里云OSS路由
|
||||
app.include_router(llm_router) # 大模型路由
|
||||
app.include_router(qa_router) # 答疑路由
|
||||
app.include_router(jimeng_router) # 即梦路由
|
||||
|
||||
# Teaching Model 相关路由
|
||||
# 登录相关(不用登录)
|
||||
|
Reference in New Issue
Block a user