github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2026-07-17 01:58:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

support dbgptQA

Browse Source
This commit is contained in:
xuyuan23
2024-01-04 15:24:15 +08:00
parent b799486dda
commit 87f775bbb4
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pilot/scene/dbgpt_chat/chat.py
View File

@@ -7,6 +7,7 @@ from pilot.scene.base import ChatScene
from pilot.scene.base_chat import BaseChat
from pilot.server.knowledge.api import knowledge_space_service
from pilot.server.knowledge.request.request import KnowledgeSpaceRequest
from .prompt import prompt
CFG = Config()

21
pilot/user/user_db.py
View File

@@ -1,13 +1,14 @@
from pilot.base_modules.meta_data.base_dao import BaseDao
from pilot.base_modules.meta_data.meta_data import Base, engine, session
from pilot.configs.config import Config
from sqlalchemy.ext.declarative import declarative_base
from datetime import datetime
from sqlalchemy import Column, String, DateTime
from sqlalchemy import Column, Integer, String, DateTime
from pilot.connections.rdbms.base_dao import BaseDao
from pilot.user.user_request import UserRequest
import uuid
CFG = Config()
Base = declarative_base()
DB_AUTH = "auth"
TABLE_USER = "user"
@@ -35,14 +36,14 @@ class UserDao(BaseDao):
create database `auth` firstly.
"""
super().__init__(
database=DB_AUTH, orm_base=Base, db_engine=engine, session=session
database=DB_AUTH, orm_base=Base, create_not_exist_table=False
)
def get_by_user_id(self, user_id: str):
"""
get user info by user_id
"""
session = self.get_session()
session = self.Session()
user_query = session.query(UserEntity).filter(UserEntity.user_id == user_id)
result = user_query.all()
session.close()
@@ -55,7 +56,7 @@ class UserDao(BaseDao):
"""
query user by user_no
"""
session = self.get_session()
session = self.Session()
user_query = session.query(UserEntity).filter(
UserEntity.user_no == user_no
).filter(
@@ -77,7 +78,7 @@ class UserDao(BaseDao):
if user_req.user_no is not None and user_req.user_channel is not None:
result = self.get_by_user_no_and_channel(user_req.user_no, user_req.user_channel)
if len(result) == 0:
session = self.get_session()
session = self.Session()
user = UserEntity(
user_id=uuid.uuid4().hex,
nick_name=user_req.nick_name,
@@ -99,7 +100,7 @@ class UserDao(BaseDao):
"""
query user by user_no
"""
session = self.get_session()
session = self.Session()
user_query = session.query(UserEntity).order_by(
UserEntity.gmt_created.desc()
)
@@ -108,14 +109,14 @@ class UserDao(BaseDao):
return result
def update_user(self, user: UserEntity):
session = self.get_session()
session = self.Session()
session.merge(user)
session.commit()
session.close()
return True
def delete_user(self, user: UserEntity):
session = self.get_session()
session = self.Session()
if user:
session.delete(user)
session.commit()