Make unique workspaces name

Author: aponcedeleonch

migrations/versions/5c2f3eee5f90_introduce_workspaces.py

@@ -24,7 +24,8 @@ def upgrade() -> None:
         CREATE TABLE workspaces (
             id TEXT PRIMARY KEY,  -- UUID stored as TEXT
             name TEXT NOT NULL,
-            is_active BOOLEAN NOT NULL DEFAULT 0
+            is_active BOOLEAN NOT NULL DEFAULT 0,
+            UNIQUE (name)
         );
         """
     )
@@ -37,4 +38,9 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
-    pass
+    # Drop the index for workspace_id
+    op.execute("DROP INDEX IF EXISTS idx_prompts_workspace_id;")
+    # Remove the workspace_id column from prompts table
+    op.execute("ALTER TABLE prompts DROP COLUMN workspace_id;")
+    # Drop the workspaces table
+    op.execute("DROP TABLE IF EXISTS workspaces;")

src/codegate/pipeline/workspace/workspace.py

@@ -23,12 +23,11 @@ async def _list_workspaces(self, *args):
         List all workspaces
         """
         workspaces = await self._db_recorder.get_workspaces()
-        print(workspaces)
         respond_str = ""
         for workspace in workspaces:
-            respond_str += f"{workspace.id} - {workspace.name}"
+            respond_str += f"- {workspace.name}"
             if workspace.is_active:
-                respond_str += " (active)"
+                respond_str += " **(active)**"
             respond_str += "\n"
         return respond_str
 
@@ -52,7 +51,7 @@ async def parse_execute_cmd(self, last_user_message: str) -> str:
         Args:
             last_user_message (str): The last user message
         """
-        command_and_args = last_user_message.split("codegate-workspace ")[1]
+        command_and_args = last_user_message.lower().split("codegate-workspace ")[1]
         command, *args = command_and_args.split(" ")
         return await self.execute(command, *args)