Modified Toolkit stats

superagi/models/tool_statistics.py

@@ -1,24 +1,24 @@
-from sqlalchemy import Column, Integer, String, ForeignKey
-from sqlalchemy.orm import relationship
-from superagi.models.base_model import DBBaseModel
 import json
 
+from sqlalchemy import Column, Integer, String
+
+from superagi.models.base_model import DBBaseModel
+
+
 class ToolStatistics(DBBaseModel):
     """ToolStatistics - used to store statistics for tool kits"""
     __tablename__ = 'tool_statistics'
 
     id = Column(Integer, primary_key=True)
     """id - id of the statistics entry"""
-    toolkit_id = Column(Integer, ForeignKey('tool_kits.id'))
+    toolkit_id = Column(Integer)
     """toolkit_id - foreign key reference to the tool kit"""
 
     key = Column(String)
     """key - key for the statistic"""
     value = Column(String)
     """value - value associated with the statistic"""
 
-    toolkit = relationship("ToolKit", back_populates="statistics")
-
     def __repr__(self):
         return f"ToolStatistics(id={self.id}, toolkit_id={self.toolkit_id}, key='{self.key}', value='{self.value}')"
 
@@ -41,4 +41,4 @@ def from_json(cls, json_data):
             toolkit_id=data['toolkit_id'],
             key=data['key'],
             value=data['value']
-        )
+        )