feat(ai): add openrouter headers (#198)

Author: mezotv

apps/api/package.json

@@ -23,7 +23,6 @@
     "@databuddy/shared": "workspace:*",
     "@elysiajs/cors": "^1.4.0",
     "@logtail/edge": "^0.5.6",
-    "@openrouter/ai-sdk-provider": "^1.2.0",
     "@upstash/ratelimit": "^2.0.6",
     "@upstash/redis": "^1.35.6",
     "ai": "^5.0.81",

apps/api/src/agent/core/ai-service.ts

@@ -1,15 +1,11 @@
-import { createOpenRouter } from "@openrouter/ai-sdk-provider";
 import { generateObject } from "ai";
 import type { z } from "zod";
 import {
 	AIResponseJsonSchema,
 	comprehensiveSystemPrompt,
 } from "../prompts/agent";
 import type { AssistantSession } from "./assistant-session";
-
-const openrouter = createOpenRouter({
-	apiKey: process.env.AI_API_KEY,
-});
+import { openrouter } from "@databuddy/shared/utils/openrouter";
 
 const AI_MODEL = "google/gemini-2.5-flash-lite-preview-06-17";
 

apps/api/src/agent_v2/index.ts

@@ -1,11 +1,7 @@
-import { createOpenRouter } from "@openrouter/ai-sdk-provider";
 import { type ModelMessage, smoothStream, stepCountIs, streamText } from "ai";
 import { chatPrompt } from "./prompts/chat-prompt";
 import { tools } from "./tools";
-
-const openrouter = createOpenRouter({
-	apiKey: process.env.AI_API_KEY,
-});
+import { openrouter } from "@databuddy/shared/utils/openrouter";
 
 export const modes = ["chat", "agent", "agent_max"] as const;
 export type Mode = (typeof modes)[number];

apps/dashboard/package.json

@@ -3,8 +3,8 @@
   "version": "0.1.0",
   "private": true,
   "scripts": {
-    "dev": "bun --bun next dev --turbopack -p 3000",
-    "build": "next build --turbopack",
+    "dev": "bun --bun next dev -p 3000",
+    "build": "next build",
     "start": "next start -p 3000",
     "typecheck": "tsgo --noEmit",
     "build:script": "bun build ./public/databuddy.ts --outfile ./public/databuddy.js --minify"
@@ -60,7 +60,7 @@
     "leaflet": "^1.9.4",
     "lucide-react": "^0.544.0",
     "nanoid": "^5.1.6",
-    "next": "^16.0.0",
+    "next": "^16.0.3",
     "next-themes": "^0.4.6",
     "nuqs": "^2.7.1",
     "ogl": "^1.0.11",

bun.lock

@@ -12,7 +12,6 @@
     "@databuddy/shared": "workspace:*",
     "@databuddy/validation": "workspace:*",
     "@logtail/edge": "catalog:",
-    "@openrouter/ai-sdk-provider": "^1.2.0",
     "@orpc/server": "^1.10.4",
     "ai": "^5.0.81",
     "autumn-js": "catalog:",

packages/rpc/package.json

@@ -1,11 +1,7 @@
-import { createOpenRouter } from "@openrouter/ai-sdk-provider";
 import { type ModelMessage, smoothStream, stepCountIs, streamText } from "ai";
 import { chatPrompt } from "./prompts/chat-prompt";
 import { tools } from "./tools";
-
-const openrouter = createOpenRouter({
-    apiKey: process.env.AI_API_KEY,
-});
+import { openrouter } from "@databuddy/shared/utils/openrouter";
 
 export const modes = ["chat", "agent", "agent_max"] as const;
 export type Mode = (typeof modes)[number];

packages/rpc/src/agent/index.ts

@@ -30,11 +30,13 @@
     "./types/website": "./src/types/website.ts",
     "./utils/date-utils": "./src/utils/date-utils.ts",
     "./utils/discord-webhook": "./src/utils/discord-webhook.ts",
-    "./utils/ids": "./src/utils/ids.ts"
+    "./utils/ids": "./src/utils/ids.ts",
+    "./utils/openrouter": "./src/utils/openrouter.ts"
   },
   "dependencies": {
     "@databuddy/db": "workspace:*",
     "@logtail/edge": "catalog:",
+    "@openrouter/ai-sdk-provider": "^1.2.2",
     "dayjs": "catalog:",
     "drizzle-orm": "catalog:",
     "nanoid": "catalog:"

packages/shared/package.json

@@ -0,0 +1,13 @@
+import { createOpenRouter } from "@openrouter/ai-sdk-provider";
+
+const apiKey = process.env.AI_API_KEY;
+
+const headers = {
+    "HTTP-Referer": "https://www.databuddy.cc/",
+    "X-Title": "Databuddy",
+};
+
+export const openrouter = createOpenRouter({
+    apiKey,
+    headers,
+});