chore: some functions (#13)

Author: hmbanan666

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

@@ -2,7 +2,7 @@ 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'
+import { getPartnersByCityTool, getPartnersBySurnameTool, getPartnersCountTool } from '~~/server/services/tools'
 
 export default defineEventHandler(async (event) => {
   try {
@@ -43,8 +43,9 @@ export default defineEventHandler(async (event) => {
       instructions: 'У тебя есть доступ к данным партнеров сети. Отвечай всегда на русском в мужском роде.',
       model: new OpenAIChatCompletionsModel(client, ai.model),
       tools: [
-        getPartnersTool,
         getPartnersByCityTool,
+        getPartnersBySurnameTool,
+        getPartnersCountTool,
       ],
     })
 

apps/web-app/server/api/partner/list.get.ts

@@ -1,5 +1,5 @@
 import { repository } from '@roll-stack/database'
 
 export default defineEventHandler(async () => {
-  return repository.partner.list()
+  return repository.partner.listWithData()
 })

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

@@ -2,13 +2,14 @@ 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',
+export const getPartnersCountTool = tool({
+  name: 'get_partners_count',
+  description: 'Get the number of partners',
   needsApproval: false,
   parameters: z.object({}),
   execute: async () => {
-    return repository.partner.list()
+    const partners = await repository.partner.list()
+    return partners.length
   },
 })
 
@@ -21,7 +22,19 @@ export const getPartnersByCityTool = tool({
   }),
   execute: async ({ city }) => {
     const partners = await repository.partner.list()
-
     return partners.filter((partner) => partner.city?.toLowerCase().includes(city.toLowerCase()))
   },
 })
+
+export const getPartnersBySurnameTool = tool({
+  name: 'get_partners_by_surname',
+  description: 'Get partners filtered by surname using case-insensitive partial matching',
+  needsApproval: false,
+  parameters: z.object({
+    surname: z.string(),
+  }),
+  execute: async ({ surname }) => {
+    const partners = await repository.partner.list()
+    return partners.filter((partner) => partner.surname?.toLowerCase().includes(surname.toLowerCase()))
+  },
+})

packages/database/src/repository/partner.ts

@@ -14,6 +14,12 @@ export class Partner {
   }
 
   static async list() {
+    return useDatabase().query.partners.findMany({
+      where: (partners, { eq }) => eq(partners.isActive, true),
+    })
+  }
+
+  static async listWithData() {
     return useDatabase().query.partners.findMany({
       where: (partners, { eq }) => eq(partners.isActive, true),
       with: {