cleanup plugin model loading

Author: Gavroche

app/models/all.py

@@ -31,14 +31,8 @@ def load_plugin_tables():
         if not os.path.exists(os.path.join(plugin_dir, 'database.py')):
             continue  # Skip plugins without a database.py file
 
-        plugin_module = import_module(f'plugins.{plugin_name}.database')
-        # plugin_metadata = plugin_module.metadata
+        import_module(f'plugins.{plugin_name}.database')
 
-        # # # Copy the tables from the main metadata to the plugin metadata object
-        # # for table_name in plugin_metadata.tables:
-        # #     table = plugin_metadata.tables[table_name]
-        # #     table.schema = None  # Reset the schema to avoid conflicts
-        # #     table.metadata = metadata
 
 
 class ConversationBase(SQLModel):
@@ -68,3 +62,5 @@ class MessageRead(MessageBase):
 
 class ConversationWithMessages(ConversationRead):
     messages: List[MessageRead] = []
+
+load_plugin_tables()

app/plugins/sample_plugin/database.py

@@ -3,13 +3,12 @@
 from sqlmodel import SQLModel, Field, Relationship
 from models.all import Conversation
 
-metadata = SQLModel.metadata
-
 class SamplePluginModelBase(SQLModel):
     """A base model for SamplePluginModel"""
     # __tablename__: str = 'sample_plugin_model'
     created_at: datetime = Field(default_factory=datetime.utcnow)
     title: Optional[str]
+    conversation_id: Optional[int] = Field(default=None, foreign_key="conversation.id")
 
 class SamplePluginModel(SamplePluginModelBase, table=True):
     """A model for SamplePlugin"""