main
HuangHai 7 days ago
parent 3915725761
commit 61f5fff4ed

@ -1,9 +1,6 @@
import asyncio import asyncio
import json import json
<<<<<<< HEAD
=======
import os.path import os.path
>>>>>>> 66c0614648a1e8f5f7b9274bdb7218f082104b24
import subprocess import subprocess
import tempfile import tempfile
import urllib import urllib
@ -21,20 +18,9 @@ from starlette.staticfiles import StaticFiles
from Util.LightRagUtil import * from Util.LightRagUtil import *
from Util.PostgreSQLUtil import init_postgres_pool from Util.PostgreSQLUtil import init_postgres_pool
<<<<<<< HEAD
# 在程序开始时添加以下配置
logging.basicConfig(
level=logging.INFO, # 设置日志级别为INFO
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
# 或者如果你想更详细地控制日志输出
=======
rag_instances = {} rag_instances = {}
rag_lock = asyncio.Lock() rag_lock = asyncio.Lock()
# 想更详细地控制日志输出
>>>>>>> 66c0614648a1e8f5f7b9274bdb7218f082104b24
logger = logging.getLogger('lightrag') logger = logging.getLogger('lightrag')
logger.setLevel(logging.DEBUG) logger.setLevel(logging.DEBUG)
handler = logging.StreamHandler() handler = logging.StreamHandler()
@ -65,12 +51,10 @@ app.mount("/static", StaticFiles(directory="Static"), name="static")
async def rag(request: fastapi.Request): async def rag(request: fastapi.Request):
data = await request.json() data = await request.json()
topic = data.get("topic") # Chinese, Math topic = data.get("topic") # Chinese, Math
mode = data.get("mode", "hybrid") # 默认为hybrid模式
# 拼接路径 # 拼接路径
WORKING_PATH = "./Topic/" + topic WORKING_PATH = "./Topic/" + topic
# 查询的问题 # 查询的问题
query = data.get("query") query = data.get("query")
<<<<<<< HEAD
# 关闭参考资料 # 关闭参考资料
user_prompt = "\n 1、不要输出参考资料 或者 References " user_prompt = "\n 1、不要输出参考资料 或者 References "
user_prompt = user_prompt + "\n 2、资料中提供化学反应方程式的一定要严格按提供的Latex公式输出绝对不允许对Latex公式进行修改 " user_prompt = user_prompt + "\n 2、资料中提供化学反应方程式的一定要严格按提供的Latex公式输出绝对不允许对Latex公式进行修改 "

Loading…
Cancel
Save