feat(ai): enhance OpenAI configuration with additional headers

Innei · Innei · commit 9b46a6075048 · 2025-05-07T00:41:01.000+08:00
- Added 'X-Title' and 'HTTP-Referer' headers to the OpenAI client configuration for improved request context.
- Included web URL in the configuration to support better tracking and analytics.

Signed-off-by: Innei &lt;tukon479@gmail.com&gt;
diff --git a/apps/core/src/modules/ai/ai.service.ts b/apps/core/src/modules/ai/ai.service.ts
@@ -13,6 +13,7 @@ export class AiService {
   public async getOpenAiChain(options?: { maxTokens?: number }) {
     const {
       ai: { openAiKey, openAiEndpoint, openAiPreferredModel },
+      url: { webUrl },
     } = await this.configService.waitForConfigReady()
     if (!openAiKey) {
       throw new BizException(ErrorCodeEnum.AINotEnabled, 'Key not found')
@@ -23,6 +24,10 @@ export class AiService {
       apiKey: openAiKey,
       configuration: {
         baseURL: openAiEndpoint || void 0,
+        defaultHeaders: {
+          'X-Title': 'Mix Space AI Client',
+          'HTTP-Referer': webUrl,
+        },
       },
       maxTokens: options?.maxTokens,
     })
