File size: 1,681 Bytes
71b4454 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | from sqlalchemy import Column, Integer, String, DateTime, Text, JSON, Float, Boolean, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from datetime import datetime
Base = declarative_base()
class ProviderConfig(Base):
__tablename__ = "provider_configs"
id = Column(Integer, primary_key=True)
provider_id = Column(String(64), unique=True, nullable=False)
name = Column(String(64), nullable=False)
api_key = Column(Text, nullable=True)
base_url = Column(String(256), nullable=True)
models = Column(JSON, default=[])
default_model = Column(String(64), nullable=True)
enabled = Column(Boolean, default=True)
priority = Column(Integer, default=0)
config = Column(JSON, default={})
created_at = Column(DateTime, default=datetime.utcnow)
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
class ModelCall(Base):
__tablename__ = "model_calls"
id = Column(Integer, primary_key=True)
call_id = Column(String(64), unique=True, nullable=False)
provider_id = Column(String(64), nullable=False)
model = Column(String(64), nullable=False)
prompt = Column(Text, nullable=False)
response = Column(Text, nullable=True)
tokens_prompt = Column(Integer, default=0)
tokens_completion = Column(Integer, default=0)
cost = Column(Float, default=0.0)
duration_ms = Column(Float, default=0.0)
success = Column(Boolean, default=True)
error = Column(Text, nullable=True)
meta = Column(JSON, default={}) # column name is 'meta' (not reserved)
created_at = Column(DateTime, default=datetime.utcnow)
|