feat: ai agent (#12)

* feat: ai agent

* chore: some fixes

* chore: update

Author: hmbanan666

.npmrc

@@ -1,2 +1,2 @@
 auto-install-peers=true
-shamefully-hoist=true
+node-linker=hoisted

apps/web-app/.env.example

@@ -1,18 +1,24 @@
 # Session password for Nuxt Auth Utils
-NUXT_SESSION_PASSWORD=""
+NUXT_SESSION_PASSWORD=
 
 # Main database
-DATABASE_URL=""
+DATABASE_URL=
 
 # S3 file storage
-NUXT_S3_BUCKET=""
-NUXT_S3_REGION=""
-NUXT_S3_ENDPOINT=""
-NUXT_S3_ACCESS_KEY_ID=""
-NUXT_S3_SECRET_ACCESS_KEY=""
+NUXT_S3_ACCESS_KEY_ID=
+NUXT_S3_BUCKET=
+NUXT_S3_ENDPOINT=
+NUXT_S3_REGION=
+NUXT_S3_SECRET_ACCESS_KEY=
 
 # URL to media server (probably s3 bucket with static URL)
-NUXT_PUBLIC_MEDIA_URL=""
+NUXT_PUBLIC_MEDIA_URL=
+
+# AI
+NUXT_AI_API_KEY=
+NUXT_AI_BASE_URL=
+NUXT_AI_MODEL=
+NUXT_AI_SERVICE_TOKEN=
 
 # App version
-VERSION=""
+VERSION=

apps/web-app/app/components/Navigation.vue

@@ -87,7 +87,6 @@ const menuItems = computed(() => [
     onSelect: () => {
       openDrawer.value = true
     },
-    badge: 'Нисушка себе!',
   },
   {
     label: t('app.menu.products'),

apps/web-app/nuxt.config.ts

@@ -12,6 +12,12 @@ export default defineNuxtConfig({
       accessKeyId: '',
       secretAccessKey: '',
     },
+    ai: {
+      model: '',
+      baseUrl: '',
+      apiKey: '',
+      serviceToken: '',
+    },
     public: {
       mediaUrl: '',
     },

apps/web-app/package.json

@@ -16,6 +16,7 @@
     "@dicebear/collection": "catalog:",
     "@dicebear/core": "catalog:",
     "@neoconfetti/vue": "catalog:",
+    "@openai/agents": "catalog:",
     "@pinia/nuxt": "catalog:",
     "@roll-stack/database": "workspace:*",
     "@roll-stack/ui": "workspace:*",
@@ -25,6 +26,7 @@
     "ioredis": "catalog:",
     "libphonenumber-js": "catalog:",
     "nuxt-tiptap-editor": "catalog:",
+    "openai": "catalog:",
     "pinia": "catalog:",
     "sharp": "catalog:"
   },

apps/web-app/server/api/agent/index.post.ts

@@ -0,0 +1,60 @@
+import { Buffer } from 'node:buffer'
+import { timingSafeEqual } from 'node:crypto'
+import { Agent, OpenAIChatCompletionsModel, run } from '@openai/agents'
+import OpenAI from 'openai'
+import { getPartnersByCityTool, getPartnersTool } from '~~/server/services/tools'
+
+export default defineEventHandler(async (event) => {
+  try {
+    const { ai } = useRuntimeConfig()
+
+    // Requires bearer token
+    const bearer = getHeader(event, 'authorization')
+    if (!bearer?.startsWith('Bearer ')) {
+      throw createError({
+        statusCode: 401,
+        message: 'Unauthorized',
+      })
+    }
+
+    const token = bearer.slice(7) // Remove 'Bearer ' prefix
+    if (!ai.serviceToken || !timingSafeEqual(Buffer.from(token), Buffer.from(ai.serviceToken))) {
+      throw createError({
+        statusCode: 401,
+        message: 'Unauthorized',
+      })
+    }
+
+    const body = await readBody(event)
+    if (!body?.message) {
+      throw createError({
+        statusCode: 400,
+        message: 'Message is required',
+      })
+    }
+
+    const client = new OpenAI({
+      apiKey: ai.apiKey,
+      baseURL: ai.baseUrl,
+    })
+
+    const agent = new Agent({
+      name: 'Дата агент сети доставок "Суши Love"',
+      instructions: 'У тебя есть доступ к данным партнеров сети. Отвечай всегда на русском в мужском роде.',
+      model: new OpenAIChatCompletionsModel(client, ai.model),
+      tools: [
+        getPartnersTool,
+        getPartnersByCityTool,
+      ],
+    })
+
+    const result = await run(agent, body.message)
+
+    return {
+      ok: true,
+      message: result.finalOutput,
+    }
+  } catch (error) {
+    throw errorResolver(error)
+  }
+})

apps/web-app/server/services/tools.ts

@@ -0,0 +1,27 @@
+import { tool } from '@openai/agents'
+import { repository } from '@roll-stack/database'
+import { z } from 'zod'
+
+export const getPartnersTool = tool({
+  name: 'get_all_partners',
+  description: 'Get all partners',
+  needsApproval: false,
+  parameters: z.object({}),
+  execute: async () => {
+    return repository.partner.list()
+  },
+})
+
+export const getPartnersByCityTool = tool({
+  name: 'get_partners_by_city',
+  description: 'Get partners filtered by city name using case-insensitive partial matching',
+  needsApproval: false,
+  parameters: z.object({
+    city: z.string(),
+  }),
+  execute: async ({ city }) => {
+    const partners = await repository.partner.list()
+
+    return partners.filter((partner) => partner.city?.toLowerCase().includes(city.toLowerCase()))
+  },
+})