Merge branch 'main' into cline

Author: yrobla

alembic.ini

@@ -15,5 +15,11 @@ prepend_sys_path = .
 # version_locations. The default within new alembic.ini files is "os", which uses os.pathsep.
 version_path_separator = os  # Use os.pathsep.
 
+# template used to generate migration file names
+file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s
+# timezone to use when rendering the date within the migration file
+# as well as the filename.
+timezone = UTC
+
 # DB connection string
 sqlalchemy.url = sqlite:///codegate_volume/db/codegate.db

migrations/versions/30d0144e1a50_init_db.py renamed to migrations/versions/2025_01_15_1249-30d0144e1a50_init_db.py

@@ -69,6 +69,8 @@ async def create_workspace(request: v1_models.CreateWorkspaceRequest) -> v1_mode
                 "Invalid workspace name. " "Please use only alphanumeric characters and dashes"
             ),
         )
+    except crud.WorkspaceCrudError as e:
+        raise HTTPException(status_code=400, detail=str(e))
     except Exception:
         raise HTTPException(status_code=500, detail="Internal server error")
 

migrations/versions/5c2f3eee5f90_introduce_workspaces.py renamed to migrations/versions/2025_01_16_1849-5c2f3eee5f90_introduce_workspaces.py

@@ -186,6 +186,8 @@ async def _add_workspace(self, flags: Dict[str, str], args: List[str]) -> str:
             return "Invalid workspace name: It should be alphanumeric and dashes"
         except AlreadyExistsError:
             return f"Workspace **{new_workspace_name}** already exists"
+        except crud.WorkspaceCrudError:
+            return "An error occurred while adding the workspace"
         except Exception:
             return "An error occurred while adding the workspace"
 

migrations/versions/a692c8b52308_add_workspace_system_prompt.py renamed to migrations/versions/2025_01_20_1546-a692c8b52308_add_workspace_system_prompt.py

@@ -17,6 +17,12 @@ class WorkspaceAlreadyActiveError(WorkspaceCrudError):
     pass
 
 
+DEFAULT_WORKSPACE_NAME = "default"
+
+# These are reserved keywords that cannot be used for workspaces
+RESERVED_WORKSPACE_KEYWORDS = [DEFAULT_WORKSPACE_NAME, "active"]
+
+
 class WorkspaceCrud:
 
     def __init__(self):
@@ -29,6 +35,10 @@ async def add_workspace(self, new_workspace_name: str) -> Workspace:
         Args:
             name (str): The name of the workspace
         """
+        if new_workspace_name == "":
+            raise WorkspaceCrudError("Workspace name cannot be empty.")
+        if new_workspace_name in RESERVED_WORKSPACE_KEYWORDS:
+            raise WorkspaceCrudError(f"Workspace name {new_workspace_name} is reserved.")
         db_recorder = DbRecorder()
         workspace_created = await db_recorder.add_workspace(new_workspace_name)
         return workspace_created
@@ -102,7 +112,7 @@ async def soft_delete_workspace(self, workspace_name: str):
         """
         if workspace_name == "":
             raise WorkspaceCrudError("Workspace name cannot be empty.")
-        if workspace_name == "default":
+        if workspace_name == DEFAULT_WORKSPACE_NAME:
             raise WorkspaceCrudError("Cannot delete default workspace.")
 
         selected_workspace = await self._db_reader.get_workspace_by_name(workspace_name)