mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2025-09-08 05:29:29 +00:00
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from sqlalchemy import JSON, Column, ForeignKey, Integer, String
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from ktransformers.server.utils.sql_utils import Base
|
|
|
|
|
|
class RunStep(Base):
|
|
__tablename__ = "run_steps"
|
|
# todo
|
|
id = Column(String, primary_key=True, index=True)
|
|
object = Column(String, default="thread.run.step")
|
|
created_at = Column(Integer)
|
|
|
|
assistant_id = Column(String, ForeignKey("assistants.id"))
|
|
thread_id = Column(String, ForeignKey("threads.id"))
|
|
run_id = Column(String, ForeignKey("runs.id"))
|
|
type = Column(String)
|
|
status = Column(String)
|
|
step_details = Column(JSON)
|
|
last_error = Column(JSON, nullable=True)
|
|
expires_at = Column(Integer, nullable=True)
|
|
cancelled_at = Column(Integer, nullable=True)
|
|
failed_at = Column(Integer, nullable=True)
|
|
completed_at = Column(Integer, nullable=True)
|
|
|
|
meta_data = Column(JSON, nullable=True)
|
|
usage = Column(JSON, nullable=True)
|
|
|
|
assistant = relationship("Assistant", back_populates="run_steps")
|
|
thread = relationship("Thread", back_populates="run_steps")
|
|
run = relationship("Run", back_populates="run_steps")
|