use the correct / agreed upon X-Request-Origin header

Author: lerouxb

packages/compass-assistant/src/compass-assistant-provider.spec.tsx

@@ -85,7 +85,11 @@ const TestComponent: React.FunctionComponent<{
 
   return (
     <DrawerContentProvider>
-      <MockedProvider appNameForPrompt="MongoDB Compass" chat={chat}>
+      <MockedProvider
+        originForPrompt="mongodb-compass"
+        appNameForPrompt="MongoDB Compass"
+        chat={chat}
+      >
         <DrawerAnchor>
           <div data-testid="provider-children">Provider children</div>
           <CompassAssistantDrawer
@@ -106,7 +110,11 @@ describe('useAssistantActions', function () {
 
       return (
         <DrawerContentProvider>
-          <MockedProvider appNameForPrompt="MongoDB Compass" chat={chat}>
+          <MockedProvider
+            originForPrompt="mongodb-compass"
+            appNameForPrompt="MongoDB Compass"
+            chat={chat}
+          >
             {children}
           </MockedProvider>
         </DrawerContentProvider>
@@ -659,7 +667,10 @@ describe('CompassAssistantProvider', function () {
       render(
         <DrawerContentProvider>
           <DrawerAnchor />
-          <MockedProvider appNameForPrompt="MongoDB Compass" />
+          <MockedProvider
+            originForPrompt="mongodb-compass"
+            appNameForPrompt="MongoDB Compass"
+          />
         </DrawerContentProvider>,
         {
           preferences: {

packages/compass-assistant/src/compass-assistant-provider.tsx

@@ -256,6 +256,7 @@ export const CompassAssistantProvider = registerCompassPlugin(
       children,
     }: PropsWithChildren<{
       appNameForPrompt: string;
+      originForPrompt: string;
       chat?: Chat<AssistantMessage>;
       atlasAiService?: AtlasAiService;
     }>) => {
@@ -280,7 +281,7 @@ export const CompassAssistantProvider = registerCompassPlugin(
         initialProps.chat ??
         new Chat({
           transport: new DocsProviderTransport({
-            appName: initialProps.appNameForPrompt,
+            origin: initialProps.originForPrompt,
             instructions: buildConversationInstructionsPrompt({
               target: initialProps.appNameForPrompt,
             }),

packages/compass-assistant/src/docs-provider-transport.spec.ts

@@ -62,6 +62,7 @@ describe('DocsProviderTransport', function () {
       });
       abortController = new AbortController();
       transport = new DocsProviderTransport({
+        origin: 'mongodb-compass',
         instructions: 'Test instructions for MongoDB assistance',
         model: mockModel,
       });

packages/compass-assistant/src/docs-provider-transport.ts

@@ -17,21 +17,21 @@ export function shouldExcludeMessage({ metadata }: AssistantMessage) {
 
 export class DocsProviderTransport implements ChatTransport<AssistantMessage> {
   private model: LanguageModel;
-  private appName: string;
+  private origin: string;
   private instructions: string;
 
   constructor({
     instructions,
     model,
-    appName,
+    origin,
   }: {
     instructions: string;
     model: LanguageModel;
-    appName: string;
+    origin: string;
   }) {
     this.instructions = instructions;
     this.model = model;
-    this.appName = appName;
+    this.origin = origin;
   }
 
   static emptyStream = new ReadableStream<UIMessageChunk>({
@@ -65,7 +65,7 @@ export class DocsProviderTransport implements ChatTransport<AssistantMessage> {
       messages: convertToModelMessages(filteredMessages),
       abortSignal: abortSignal,
       headers: {
-        'X-Request-Origin': this.appName,
+        'X-Request-Origin': this.origin,
       },
       providerOptions: {
         openai: {

packages/compass-web/src/entrypoint.tsx

@@ -502,6 +502,7 @@ const CompassWeb = ({
                         >
                           <CompassInstanceStorePlugin>
                             <CompassAssistantProvider
+                              originForPrompt="atlas-data-explorer"
                               appNameForPrompt={
                                 APP_NAMES_FOR_PROMPT.DataExplorer
                               }

packages/compass/src/app/components/home.tsx

@@ -151,6 +151,7 @@ function HomeWithConnections({
     <ConnectionStorageProvider value={connectionStorage}>
       <FileInputBackendProvider createFileInputBackend={createFileInputBackend}>
         <CompassAssistantProvider
+          originForPrompt="mongodb-compass"
           appNameForPrompt={APP_NAMES_FOR_PROMPT.Compass}
         >
           <CompassConnections