Add workspace functionality

app_users/models.py

@@ -233,6 +233,13 @@ def get_or_create_personal_workspace(self) -> tuple["Workspace", bool]:
 
         return Workspace.objects.get_or_create_from_user(self)
 
+    def get_workspaces(self) -> models.QuerySet["Workspace"]:
+        from workspaces.models import Workspace
+
+        return Workspace.objects.filter(
+            memberships__user=self, memberships__deleted__isnull=True
+        )
+
     def get_anonymous_token(self):
         return auth.create_custom_token(self.uid).decode()
 

bots/migrations/0087_alter_publishedrun_visibility_and_more.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.1.2 on 2024-11-07 16:31
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('bots', '0086_publishedrunversion_change_notes'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='publishedrun',
+            name='visibility',
+            field=models.IntegerField(choices=[(1, 'Unlisted'), (2, 'Public'), (3, 'Internal')], default=1),
+        ),
+        migrations.AlterField(
+            model_name='publishedrunversion',
+            name='visibility',
+            field=models.IntegerField(choices=[(1, 'Unlisted'), (2, 'Public'), (3, 'Internal')], default=1),
+        ),
+    ]

bots/models.py

@@ -15,8 +15,9 @@
 from app_users.models import AppUser
 from bots.admin_links import open_in_new_tab
 from bots.custom_fields import PostgresJSONEncoder, CustomURLField
-from daras_ai_v2 import icons
+from daras_ai_v2 import icons, urls
 from daras_ai_v2.crypto import get_random_doc_id
+from daras_ai_v2.fastapi_tricks import get_route_path
 from daras_ai_v2.language_model import format_chat_entry
 from functions.models import CalledFunctionResponse
 from gooeysite.bg_db_conn import get_celery_result_db_safe
@@ -36,27 +37,58 @@
 class PublishedRunVisibility(models.IntegerChoices):
     UNLISTED = 1
     PUBLIC = 2
+    INTERNAL = 3
+
+    @classmethod
+    def choices_for_workspace(
+        cls, workspace: typing.Optional["Workspace"]
+    ) -> typing.Iterable["PublishedRunVisibility"]:
+        if not workspace or workspace.is_personal:
+            return [cls.UNLISTED, cls.PUBLIC]
+        else:
+            # TODO: Add cls.PUBLIC when team-handles are added
+            return [cls.UNLISTED, cls.INTERNAL]
+
+    def help_text(self, workspace: typing.Optional["Workspace"] = None):
+        from routers.account import profile_route, saved_route
 
-    def help_text(self):
         match self:
             case PublishedRunVisibility.UNLISTED:
                 return f"{self.get_icon()} Only me + people with a link"
+            case PublishedRunVisibility.PUBLIC if workspace and workspace.is_personal:
+                user = workspace.created_by
+                if user.handle:
+                    profile_url = user.handle.get_app_url()
+                    pretty_profile_url = urls.remove_scheme(profile_url).rstrip("/")
+                    return f'{self.get_icon()} Public on <a href="{pretty_profile_url}" target="_blank">{profile_url}</a>'
+                else:
+                    edit_profile_url = get_route_path(profile_route)
+                    return f'{self.get_icon()} Public on <a href="{edit_profile_url}" target="_blank">my profile page</a>'
             case PublishedRunVisibility.PUBLIC:
                 return f"{self.get_icon()} Public"
+            case PublishedRunVisibility.INTERNAL if workspace:
+                saved_route_url = get_route_path(saved_route)
+                return f'{self.get_icon()} Members <a href="{saved_route_url}" target="_blank">can find</a> and edit'
+            case PublishedRunVisibility.INTERNAL:
+                return f"{self.get_icon()} Members can find and edit"
 
     def get_icon(self):
         match self:
             case PublishedRunVisibility.UNLISTED:
                 return icons.lock
             case PublishedRunVisibility.PUBLIC:
                 return icons.globe
+            case PublishedRunVisibility.INTERNAL:
+                return icons.company_solid
 
-    def get_badge_html(self):
+    def get_badge_html(self) -> str:
         match self:
             case PublishedRunVisibility.UNLISTED:
                 return f"{self.get_icon()} Private"
             case PublishedRunVisibility.PUBLIC:
                 return f"{self.get_icon()} Public"
+            case PublishedRunVisibility.INTERNAL:
+                return f"{self.get_icon()} Internal"
 
 
 class Platform(models.IntegerChoices):
@@ -1686,6 +1718,7 @@ def create_with_version(
                 title=title,
                 visibility=visibility,
                 notes=notes,
+                change_notes="First Version",
             )
             return pr