修改web请求接口
This commit is contained in:
@@ -35,10 +35,17 @@ class conversations:
|
||||
def delete(self,id:str):
|
||||
dbManage.delete(self._tableName,id=id)
|
||||
|
||||
def rename(self,id:str):
|
||||
data = {'name':''}
|
||||
def rename(self,id:str,name:str):
|
||||
data = {'name':name}
|
||||
dbManage.update(self._tableName,data,id=id)
|
||||
|
||||
def query(self,**condition):
|
||||
results = []
|
||||
records = dbManage.query(self._tableName,**condition)
|
||||
for record in records:
|
||||
results.append(record.dict())
|
||||
return results
|
||||
|
||||
class user:
|
||||
def __init__(self) -> None:
|
||||
self._tableName = 'user'
|
||||
@@ -83,23 +90,8 @@ class parameter:
|
||||
key = record['name']
|
||||
value = record['value']
|
||||
data[key] = value
|
||||
|
||||
return {
|
||||
'opening_statement':data['opening_statement'],
|
||||
'suggested_questions':data['suggested_questions'],
|
||||
'suggested_questions_after_answer':data['suggested_questions_after_answer'],
|
||||
'speech_to_text':data['speech_to_text'],
|
||||
'text_to_speech':data['text_to_speech'],
|
||||
'retriever_resource':data['retriever_resource'],
|
||||
'annotation_reply':data['annotation_reply'],
|
||||
'more_like_this':data['more_like_this'],
|
||||
'user_input_form':data['user_input_form'],
|
||||
'sensitive_word_avoidance':data['sensitive_word_avoidance'],
|
||||
'file_upload':data['file_upload'],
|
||||
'system_parameters':data['system_parameters'],
|
||||
'opening_statement':data['opening_statement'],
|
||||
}
|
||||
|
||||
return data
|
||||
|
||||
def set(self,user_id:str):
|
||||
dbManage.addRecord(self._tableName,{})
|
||||
|
||||
|
||||
@@ -20,6 +20,14 @@ class ConversationOrm(Base):
|
||||
introduction = Column(String)
|
||||
created_at = Column(Integer)
|
||||
|
||||
def update(self,data:Dict[str,Any]):
|
||||
if 'name' in data:
|
||||
self.name = data['name']
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class UserOrm(Base):
|
||||
__tablename__ = "user"
|
||||
|
||||
@@ -143,14 +151,20 @@ class DBManager:
|
||||
session.commit()
|
||||
|
||||
def update(self,tableName:str,data:Dict[str,Any],**filter):
|
||||
if not self.exist(tableName):
|
||||
return
|
||||
session = self.SessionLocal()
|
||||
ormCls = self._get_orm(tableName)
|
||||
if ormCls is None:
|
||||
return
|
||||
record = session.query(ormCls).filter_by(**filter).first()
|
||||
if record is not None:
|
||||
record.update(data)
|
||||
session.commit()
|
||||
if len(filter) > 0:
|
||||
records = session.query(ormCls).filter_by(**filter).all()
|
||||
else:
|
||||
records = session.query(ormCls).all()
|
||||
for record in records:
|
||||
if record is not None:
|
||||
record.update(data)
|
||||
session.commit()
|
||||
|
||||
def query(self,tableName:str,**filter):
|
||||
session = self.SessionLocal()
|
||||
|
||||
@@ -1,6 +1,5 @@
|
||||
|
||||
from typing import Dict, Any
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
@@ -11,3 +10,6 @@ class ChatRequestData(BaseModel):
|
||||
response_mode: str
|
||||
files: Any
|
||||
conversation_id: str = None
|
||||
|
||||
class ChatFileUploadRequest(BaseModel):
|
||||
base64: str
|
||||
Reference in New Issue
Block a user