diff --git a/ui/desktop/openapi.json b/ui/desktop/openapi.json
index e2de8305..6aeb71cb 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 2f821de0..14a147a2 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/UserMessage.tsx b/ui/desktop/src/components/UserMessage.tsx
index a871b5b1..9b733f03 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>