Add username field in settings page and add a run_app method consumers (#174)

Author: vegito22

llmstack/apps/apis.py

@@ -634,19 +634,23 @@ def patch(self, request, uid):
         # Find the versioned app data and update it
         app_data = {
             "name": request.data["name"] if "name" in request.data else versioned_app_data.data["name"],
-            "type_slug": request.data["type_slug"]
-            if "type_slug" in request.data
-            else versioned_app_data.data["type_slug"],
-            "description": request.data["description"]
-            if "description" in request.data
-            else versioned_app_data.data["description"],
+            "type_slug": (
+                request.data["type_slug"] if "type_slug" in request.data else versioned_app_data.data["type_slug"]
+            ),
+            "description": (
+                request.data["description"] if "description" in request.data else versioned_app_data.data["description"]
+            ),
             "config": request.data["config"] if "config" in request.data else versioned_app_data.data["config"],
-            "input_fields": request.data["input_fields"]
-            if "input_fields" in request.data
-            else versioned_app_data.data["input_fields"],
-            "output_template": request.data["output_template"]
-            if "output_template" in request.data
-            else versioned_app_data.data["output_template"],
+            "input_fields": (
+                request.data["input_fields"]
+                if "input_fields" in request.data
+                else versioned_app_data.data["input_fields"]
+            ),
+            "output_template": (
+                request.data["output_template"]
+                if "output_template" in request.data
+                else versioned_app_data.data["output_template"]
+            ),
             "processors": processed_processors_data,
         }
 
@@ -888,6 +892,7 @@ def run_app_internal(
         request,
         platform=None,
         preview=False,
+        version=None,
     ):
         app = get_object_or_404(App, uuid=uuid.UUID(uid))
         app_owner = get_object_or_404(Profile, user=app.owner)
@@ -928,6 +933,12 @@ def run_app_internal(
             )
             .order_by("-created_at")
             .first()
+            if version is None
+            else AppData.objects.filter(
+                app_uuid=app.uuid,
+                version=version,
+                is_draft=False,
+            ).first()
         )
 
         # If we are running a published app, use the published app data

llmstack/base/serializers.py

@@ -6,6 +6,7 @@
 class ProfileSerializer(serializers.ModelSerializer):
     user_email = serializers.EmailField(source="user.email")
     name = serializers.SerializerMethodField()
+    username = serializers.CharField(source="user.username")
     openai_key = serializers.SerializerMethodField()
     stabilityai_key = serializers.SerializerMethodField()
     cohere_key = serializers.SerializerMethodField()
@@ -60,6 +61,7 @@ class Meta:
         fields = [
             "name",
             "user_email",
+            "username",
             "token",
             "openai_key",
             "stabilityai_key",

llmstack/client/src/pages/setting.jsx

@@ -41,6 +41,10 @@ const VisuallyHiddenInput = styled("input")({
 const settingsSchema = {
   type: "object",
   properties: {
+    username: {
+      type: "string",
+      title: "Username",
+    },
     openai_key: {
       type: "string",
       title: "OpenAI API Key",
@@ -81,6 +85,9 @@ const settingsSchema = {
 };
 
 const settingsUiSchema = {
+  username: {
+    "ui:default": null,
+  },
   openai_key: {
     "ui:widget": "password",
     "ui:default": "",
@@ -125,6 +132,7 @@ const settingsUiSchema = {
 
 const SettingPage = () => {
   const [formData, setFormData] = useState({
+    username: "",
     token: "",
     openai_key: "",
     stabilityai_key: "",
@@ -149,6 +157,7 @@ const SettingPage = () => {
       () => {},
       (profile) => {
         setFormData({
+          username: profile.username,
           token: profile.token,
           openai_key: profile.openai_key,
           stabilityai_key: profile.stabilityai_key,

llmstack/server/consumers.py

@@ -76,6 +76,17 @@ async def disconnect(self, close_code):
         # TODO: Close the stream
         pass
 
+    def _run_app(self, request_uuid, request, **kwargs):
+        from llmstack.apps.apis import AppViewSet
+
+        return AppViewSet().run_app_internal_async(
+            uid=self.app_id,
+            session_id=self._session_id,
+            request_uuid=request_uuid,
+            request=request,
+            preview=self.preview,
+        )
+
     async def _respond_to_event(self, text_data):
         from llmstack.apps.apis import AppViewSet
 
@@ -94,16 +105,8 @@ async def _respond_to_event(self, text_data):
                 request = await _build_request_from_input({"input": input, "stream": True}, self.scope)
                 if is_ratelimited_fn(request, self._respond_to_event):
                     raise Ratelimited("Rate limit reached.")
-                # if is_usage_limited_fn(request, self._respond_to_event):
-                #     raise UsageLimitReached("Usage limit reached.")
-
-                output_stream = await AppViewSet().run_app_internal_async(
-                    self.app_id,
-                    self._session_id,
-                    request_uuid,
-                    request,
-                    self.preview,
-                )
+
+                output_stream = await self._run_app(request_uuid=request_uuid, request=request)
                 # Generate a uuid for the response
                 response_id = str(uuid.uuid4())