'commit'

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
+