1 files changed, 17 insertions, 1 deletions
diff --git a/lib/vendor-document-list/service.ts b/lib/vendor-document-list/service.ts
index f3b2b633..de6f0488 100644
--- a/lib/vendor-document-list/service.ts
+++ b/lib/vendor-document-list/service.ts
@@ -3,7 +3,7 @@
 import { eq, SQL } from "drizzle-orm"
 import db from "@/db/db"
 import { documents, documentStagesView, issueStages } from "@/db/schema/vendorDocu"
-import { contracts } from "@/db/schema/vendorData"
+import { contracts } from "@/db/schema"
 import { GetVendorDcoumentsSchema } from "./validations"
 import { unstable_cache } from "@/lib/unstable-cache";
 import { filterColumns } from "@/lib/filter-columns";
@@ -293,3 +293,19 @@ export async function modifyDocument(input: ModifyDocumentInputType) {
     };
   }
 }
+
+
+export async function getContractIdsByVendor(vendorId: number): Promise<number[]> {
+  try {
+    const contractsData = await db
+      .select({ id: contracts.id })
+      .from(contracts)
+      .where(eq(contracts.vendorId, vendorId))
+      .orderBy(contracts.id)
+
+    return contractsData.map(contract => contract.id)
+  } catch (error) {
+    console.error('Error fetching contract IDs by vendor:', error)
+    return []
+  }
+}
+\ No newline at end of file