test sqlite-web

requirements.txt

@@ -3,4 +3,5 @@ uvicorn[standard]
 captcha
 python-dotenv
 langchain_google_genai==1.0.10
-langchain-openai==0.1.25 # 使用 openai 或兼容 openai
+langchain-openai==0.1.25 # 使用 openai 或兼容 openai
+sqlite-web

routers/openai_v1.py

@@ -15,17 +15,26 @@ async def chat_completions(request:Request):
     print('chat_completions')
     auth_header = request.headers['authorization']
     user_api_key = auth_header.split()[1]  # 分割字符串并取第二个元素
+    user_id = await get_user_id(user_api_key)
+    if user_id==0:
+        return {'error': {'code': '401', 'message': 'api_key 已过期或验证不正确！'}}
+
     data = await request.json()
     try:
         model = data['model']
     except:
         model = ''
+    
     if model=='':
         model = await get_default_model()
     api_key_info = await get_api_key(model)
+
     api_key = api_key_info['api_key']
     group_name = api_key_info['group_name']
-
+    try:
+        base_url = api_key_info['base_url']
+    except:
+        base_url = ''
 
     # 下面是google api
     if group_name=='gemini': 
@@ -35,17 +44,15 @@ async def chat_completions(request:Request):
             api_key = api_key,
             model = model,
         )
-
     
     # 下面就是 chatgpt 兼容 api
-    if group_name=='glm':
-        from langchain_openai import ChatOpenAI
-        # 初始化 ChatOpenAI 模型
-        llm = ChatOpenAI(
-            model = model,
-            api_key = api_key,
-            base_url = 'https://open.bigmodel.cn/api/paas/v4', # 下一步将base_url 写到数据库里
-        )
+    from langchain_openai import ChatOpenAI
+    # 初始化 ChatOpenAI 模型
+    llm = ChatOpenAI(
+        model = model,
+        api_key = api_key,
+        base_url = base_url,
+    )
 
     messages = data['messages']
     rslt = [(item["role"], item["content"]) for item in messages] # 转换为所需的元组列表格式
@@ -53,7 +60,10 @@ async def chat_completions(request:Request):
     prompt_template = ChatPromptTemplate.from_messages(rslt)
 
     chain = prompt_template | llm | parser
-    result = chain.invoke({})
+    try:
+        result = chain.invoke({})
+    except Exception as e:
+        return {'error': str(e)}
     return result
 
 async def get_default_model():
@@ -66,7 +76,7 @@ async def get_default_model():
 
 async def get_api_key(model):
     sql = f"""
-SELECT an.api_name, ak.api_key, ag.group_name
+SELECT an.api_name, ak.api_key, an.base_url, ag.group_name
 FROM api_keys ak
 JOIN api_groups ag ON ak.api_group_id = ag.id
 JOIN api_names an ON an.api_group_id = ag.id
@@ -75,6 +85,7 @@ ORDER BY ak.last_call_at
 limit 1
 """
     results = Db(os.getenv("DB_PATH")).list_query(sql)
+    # return results
     try:
         result = results[0]
         api_key = result['api_key']
@@ -84,3 +95,12 @@ limit 1
     sql = f"update api_keys set last_call_at=datetime('now') where api_key='{api_key}'"
     Db(os.getenv("DB_PATH")).update_query(sql)
     return result
+
+async def get_user_id(api_key):
+    sql = f"select id from users where api_key='{api_key}'"
+    results = Db(os.getenv("DB_PATH")).list_query(sql)
+    try:
+        result = results[0]['id']
+    except:
+        result = 0
+    return result

routers/webtools_v1.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, APIRouter, Response, Depends
-from captcha.image import ImageCaptcha
+# from captcha.image import ImageCaptcha
 from io import BytesIO
 
 router = APIRouter()