mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-26 03:44:58 +08:00
47 lines
1.9 KiB
Python
47 lines
1.9 KiB
Python
import sqlalchemy
|
|
|
|
from .base import Base
|
|
|
|
|
|
class LLMModel(Base):
|
|
"""LLM model"""
|
|
|
|
__tablename__ = 'llm_models'
|
|
|
|
uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True, unique=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={})
|
|
created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False, server_default=sqlalchemy.func.now())
|
|
updated_at = sqlalchemy.Column(
|
|
sqlalchemy.DateTime,
|
|
nullable=False,
|
|
server_default=sqlalchemy.func.now(),
|
|
onupdate=sqlalchemy.func.now(),
|
|
)
|
|
|
|
|
|
class EmbeddingModel(Base):
|
|
"""Embedding 模型"""
|
|
|
|
__tablename__ = 'embedding_models'
|
|
|
|
uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True, unique=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)
|
|
extra_args = sqlalchemy.Column(sqlalchemy.JSON, nullable=False, default={})
|
|
created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False, server_default=sqlalchemy.func.now())
|
|
updated_at = sqlalchemy.Column(
|
|
sqlalchemy.DateTime,
|
|
nullable=False,
|
|
server_default=sqlalchemy.func.now(),
|
|
onupdate=sqlalchemy.func.now(),
|
|
)
|