Commit 78a9f4b0 by 文靖昊

修改接口,适应前端也没

parent 4f4401f4
import sys import sys
sys.path.append('../') sys.path.append('../')
from fastapi import FastAPI, Header from fastapi import FastAPI, Header,Query
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from datetime import datetime,timedelta from datetime import datetime,timedelta
from src.pgdb.chat.c_db import UPostgresDB from src.pgdb.chat.c_db import UPostgresDB
...@@ -87,8 +87,8 @@ def login(phone_request: PhoneLoginRequest): ...@@ -87,8 +87,8 @@ def login(phone_request: PhoneLoginRequest):
} }
@app.get('/api/sessions/{type}') @app.get('/api/sessions/chat/')
def get_sessions(type: int,token: str = Header(None)): def get_sessions(timestamp: int = Query(None, alias="_"),token: str = Header(None)):
if not token: if not token:
return { return {
'code': 404, 'code': 404,
......
...@@ -107,17 +107,7 @@ class CRUD: ...@@ -107,17 +107,7 @@ class CRUD:
query = f'INSERT INTO turn_qa(chat_id, question, answer, turn_number, is_last) VALUES (%s,%s,%s,%s,%s)' query = f'INSERT INTO turn_qa(chat_id, question, answer, turn_number, is_last) VALUES (%s,%s,%s,%s,%s)'
self.db.execute_args(query, (chat_id, question, answer, turn_number, is_last)) self.db.execute_args(query, (chat_id, question, answer, turn_number, is_last))
def insert_turn_qa_return_id(self, chat_id, question, answer, turn_number, is_last):
conn = self.db.conn
cur = conn.cursor
query = f'INSERT INTO turn_qa(chat_id, question, answer, turn_number, is_last) VALUES (%s,%s,%s,%s,%s)'
cur.execute_args(query, (chat_id, question, answer, turn_number, is_last))
inserted_id = cur.fetchone()[0]
conn.commit()
# 关闭游标和数据库连接
cur.close()
conn.close()
return inserted_id
def insert_c_user(self, account, password): def insert_c_user(self, account, password):
query = f'INSERT INTO c_user(account, password) VALUES (%s,%s)' query = f'INSERT INTO c_user(account, password) VALUES (%s,%s)'
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment