From 1f9abd64bdd1b4e0ae186060d5fea4da715f772c Mon Sep 17 00:00:00 2001
From: Lily Delalande <ldelalande@squareup.com>
Date: Thu, 3 Jul 2025 14:03:47 -0400
Subject: [PATCH 1/2] update styling of user messages

---
 ui/desktop/openapi.json                       | 6 ++++++
 ui/desktop/src/api/types.gen.ts               | 3 +++
 ui/desktop/src/components/MarkdownContent.tsx | 1 +
 3 files changed, 10 insertions(+)

diff --git a/ui/desktop/openapi.json b/ui/desktop/openapi.json
index e2de8305dac9..6aeb71cb1043 100644
--- a/ui/desktop/openapi.json
+++ b/ui/desktop/openapi.json
@@ -1342,6 +1342,12 @@
               "envs": {
                 "$ref": "#/components/schemas/Envs"
               },
+              "headers": {
+                "type": "object",
+                "additionalProperties": {
+                  "type": "string"
+                }
+              },
               "name": {
                 "type": "string",
                 "description": "The name used to identify this extension"
diff --git a/ui/desktop/src/api/types.gen.ts b/ui/desktop/src/api/types.gen.ts
index 2f821de044b6..14a147a20fd0 100644
--- a/ui/desktop/src/api/types.gen.ts
+++ b/ui/desktop/src/api/types.gen.ts
@@ -132,6 +132,9 @@ export type ExtensionConfig = {
     description?: string | null;
     env_keys?: Array<string>;
     envs?: Envs;
+    headers?: {
+        [key: string]: string;
+    };
     /**
      * The name used to identify this extension
      */
diff --git a/ui/desktop/src/components/MarkdownContent.tsx b/ui/desktop/src/components/MarkdownContent.tsx
index 634fcee88261..ea42e7116cdc 100644
--- a/ui/desktop/src/components/MarkdownContent.tsx
+++ b/ui/desktop/src/components/MarkdownContent.tsx
@@ -127,6 +127,7 @@ export default function MarkdownContent({ content, className = '' }: MarkdownCon
           prose-ol:my-2
           prose-ul:mt-0 prose-ul:mb-3
           prose-li:m-0
+          ${className.includes('user-message') ? 'prose-headings:text-white prose-strong:text-white prose-em:text-white' : ''}
 
           ${className}`}
         components={{

From cb5505a5745ca48246c50829d397d463f10998d4 Mon Sep 17 00:00:00 2001
From: Lily Delalande <ldelalande@squareup.com>
Date: Thu, 3 Jul 2025 14:13:42 -0400
Subject: [PATCH 2/2] put formatting in user message

---
 ui/desktop/src/components/MarkdownContent.tsx | 1 -
 ui/desktop/src/components/UserMessage.tsx     | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/ui/desktop/src/components/MarkdownContent.tsx b/ui/desktop/src/components/MarkdownContent.tsx
index ea42e7116cdc..634fcee88261 100644
--- a/ui/desktop/src/components/MarkdownContent.tsx
+++ b/ui/desktop/src/components/MarkdownContent.tsx
@@ -127,7 +127,6 @@ export default function MarkdownContent({ content, className = '' }: MarkdownCon
           prose-ol:my-2
           prose-ul:mt-0 prose-ul:mb-3
           prose-li:m-0
-          ${className.includes('user-message') ? 'prose-headings:text-white prose-strong:text-white prose-em:text-white' : ''}
 
           ${className}`}
         components={{
diff --git a/ui/desktop/src/components/UserMessage.tsx b/ui/desktop/src/components/UserMessage.tsx
index a871b5b1ad14..9b733f038482 100644
--- a/ui/desktop/src/components/UserMessage.tsx
+++ b/ui/desktop/src/components/UserMessage.tsx
@@ -38,7 +38,7 @@ export default function UserMessage({ message }: UserMessageProps) {
             <div ref={contentRef}>
               <MarkdownContent
                 content={displayText}
-                className="text-white prose-a:text-white user-message"
+                className="text-white prose-a:text-white prose-headings:text-white prose-strong:text-white prose-em:text-white user-message"
               />
             </div>
           </div>