xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-09 23:36:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: move entities

Browse Source
This commit is contained in:
Junyan Qin
2025-03-10 22:34:45 +08:00
parent a0fd152d19
commit c1f4de425a
14 changed files with 105 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
pkg/entity/__init__.py Normal file
View File

0
pkg/entity/persistence/__init__.py Normal file
View File

6
pkg/entity/persistence/base.py Normal file
View File

@@ -0,0 +1,6 @@
import sqlalchemy.orm
import pydantic
class Base(sqlalchemy.orm.DeclarativeBase):
pass

19
pkg/entity/persistence/model.py Normal file
View File

@@ -0,0 +1,19 @@
import sqlalchemy
from .base import Base
class LLMModel(Base):
"""LLM 模型"""
__tablename__ = 'llm_models'
uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True)
name = sqlalchemy.Column(sqlalchemy.String(255), nullable=False)
description = sqlalchemy.Column(sqlalchemy.String(255), nullable=False)
requester = sqlalchemy.Column(sqlalchemy.String(255), nullable=False)
requester_config = sqlalchemy.Column(sqlalchemy.JSON, nullable=False, default={})
api_keys = sqlalchemy.Column(sqlalchemy.JSON, nullable=False)
abilities = sqlalchemy.Column(sqlalchemy.JSON, nullable=False, default=[])
extra_args = sqlalchemy.Column(sqlalchemy.JSON, nullable=False, default={})
updated_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False)
created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False)

13
pkg/entity/persistence/user.py Normal file
View File

@@ -0,0 +1,13 @@
import sqlalchemy
from .base import Base
class User(Base):
__tablename__ = 'users'
id = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True)
user = sqlalchemy.Column(sqlalchemy.String(255), nullable=False)
password = sqlalchemy.Column(sqlalchemy.String(255), nullable=False)
created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False)
updated_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False)