rag-gateway/app/models/llm_model.py

from datetime import datetime
from enum import IntEnum
from typing import Optional

from sqlalchemy import Column, Integer, String, DateTime, Enum, Index, Table, ForeignKey
from pydantic import BaseModel
from sqlalchemy.orm import relationship, backref

from app.models.base_model import Base

class CommonLlmModel(Base):
    __tablename__ = 'common_llm'
    __mapper_args__ = {
        # "order_by": 'SEQ'
    }

    id = Column(String(36), primary_key=True, nullable=False)
    llm_factory = Column(String(128), nullable=False)
    model_type = Column(String(128), nullable=False)
    llm_name = Column(String(128), nullable=False)
    api_key = Column(String(1024), nullable=True)
    api_base = Column(String(255), nullable=True)

    def to_json(self):
        json = {
            'id': self.ID,
            'llm_factory': self.llm_factory,
            'model_type': self.model_type,
            'llm_name': self.llm_name,
            'api_key': self.api_key,
            'api_base': self.api_base,
        }
        return json