github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-01 08:11:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:switch BaseDao from meta_data

Browse Source
This commit is contained in:
aries_ckt 2023-10-19 14:48:29 +08:00
parent 9b662c09f1
commit 08f6c5c2df
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pilot/openapi/api_v1/feedback/feed_back_db.py
View File

@ -1,13 +1,11 @@
from datetime import datetime from datetime import datetime
from sqlalchemy import Column, Integer, Text, String, DateTime from sqlalchemy import Column, Integer, Text, String, DateTime
from sqlalchemy.ext.declarative import declarative_base
from pilot.connections.rdbms.base_dao import BaseDao from pilot.base_modules.meta_data.base_dao import BaseDao
from pilot.base_modules.meta_data.meta_data import Base, engine, session
from pilot.openapi.api_v1.feedback.feed_back_model import FeedBackBody from pilot.openapi.api_v1.feedback.feed_back_model import FeedBackBody
Base = declarative_base()
class ChatFeedBackEntity(Base): class ChatFeedBackEntity(Base):
__tablename__ = "chat_feed_back" __tablename__ = "chat_feed_back"
@ -33,13 +31,15 @@ class ChatFeedBackEntity(Base):
class ChatFeedBackDao(BaseDao): class ChatFeedBackDao(BaseDao):
def __init__(self): def __init__(self):
super().__init__(database="history", orm_base=Base, create_not_exist_table=True) super().__init__(
database="dbgpt", orm_base=Base, db_engine=engine, session=session
)
def create_or_update_chat_feed_back(self, feed_back: FeedBackBody): def create_or_update_chat_feed_back(self, feed_back: FeedBackBody):
# Todo: We need to have user information first. # Todo: We need to have user information first.
def_user_name = "" def_user_name = ""
session = self.Session() session = self.get_session()
chat_feed_back = ChatFeedBackEntity( chat_feed_back = ChatFeedBackEntity(
conv_uid=feed_back.conv_uid, conv_uid=feed_back.conv_uid,
conv_index=feed_back.conv_index, conv_index=feed_back.conv_index,
@ -73,7 +73,7 @@ class ChatFeedBackDao(BaseDao):
session.close() session.close()
def get_chat_feed_back(self, conv_uid: str, conv_index: int): def get_chat_feed_back(self, conv_uid: str, conv_index: int):
session = self.Session() session = self.get_session()
result = ( result = (
session.query(ChatFeedBackEntity) session.query(ChatFeedBackEntity)
.filter(ChatFeedBackEntity.conv_uid == conv_uid) .filter(ChatFeedBackEntity.conv_uid == conv_uid)