11from dataclasses import dataclass , field
2- from sqlalchemy import ForeignKey , Index , Column
2+ from optparse import Option
3+ from sqlalchemy import ForeignKey , Index , Column , String
34from sqlalchemy .orm import relationship
45from sqlalchemy .dialects .postgresql import JSONB , UUID
56from typing import TYPE_CHECKING , Optional , List
@@ -21,6 +22,8 @@ class ToolReference(CommonMixin):
2122
2223 __table_args__ = (Index ("ix_tool_reference_project_id" , "project_id" ),)
2324
25+ tool_name : str = field (metadata = {"db" : Column (String , nullable = False )})
26+
2427 project_id : asUUID = field (
2528 metadata = {
2629 "db" : Column (
@@ -30,8 +33,10 @@ class ToolReference(CommonMixin):
3033 )
3134 }
3235 )
33-
34- configs : Optional [dict ] = field (
36+ tool_description : Optional [str ] = field (
37+ default = None , metadata = {"db" : Column (String , nullable = True )}
38+ )
39+ tool_arguments_schema : Optional [dict ] = field (
3540 default = None , metadata = {"db" : Column (JSONB , nullable = True )}
3641 )
3742
0 commit comments