From f45ae8e942feb5a45fb723147b861c6af00d968e Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Thu, 21 Aug 2025 08:48:18 +0800 Subject: [PATCH] 'commit' --- dsLightRag/Routes/JiMeng.py | 33 +++++++++++++++++++++++++++++++++ dsLightRag/Start.py | 2 ++ 2 files changed, 35 insertions(+) create mode 100644 dsLightRag/Routes/JiMeng.py diff --git a/dsLightRag/Routes/JiMeng.py b/dsLightRag/Routes/JiMeng.py new file mode 100644 index 00000000..344a4257 --- /dev/null +++ b/dsLightRag/Routes/JiMeng.py @@ -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 + diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py index 59dbafbb..0f795051 100644 --- a/dsLightRag/Start.py +++ b/dsLightRag/Start.py @@ -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 相关路由 # 登录相关(不用登录)