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