Merge pull request #1 from codeSTACKr/jacob/versions

feat: Upgrade versions, small fixes

Author: codeSTACKr

.eslintrc.cjs

@@ -10,7 +10,7 @@ module.exports = {
     project: "./tsconfig.json",
     sourceType: "module",
   },
-  plugins: ["import", "@typescript-eslint", "no-instanceof"],
+  plugins: ["import", "@typescript-eslint"],
   ignorePatterns: [
     ".eslintrc.cjs",
     "scripts",
@@ -23,7 +23,6 @@ module.exports = {
   ],
   rules: {
     "no-process-env": 0,
-    "no-instanceof/no-instanceof": 2,
     "@typescript-eslint/explicit-module-boundary-types": 0,
     "@typescript-eslint/no-empty-function": 0,
     "@typescript-eslint/no-non-null-assertion": 0,
@@ -48,6 +47,7 @@ module.exports = {
     "no-await-in-loop": 0,
     "no-bitwise": 0,
     "no-console": 0,
+    "no-constant-condition": 0,
     "no-restricted-syntax": 0,
     "no-shadow": 0,
     "no-continue": 0,

data/hr_policies.ts

@@ -24,7 +24,7 @@ export const hrPolicies: HRPolicy[] = [
         * Multiple team members taking time off simultaneously
     `,
     lastUpdated: new Date("2024-01-01"),
-    requiresManagerApproval: true
+    requiresManagerApproval: true,
   },
   {
     id: "SICK001",
@@ -42,7 +42,7 @@ export const hrPolicies: HRPolicy[] = [
         * Care for immediate family members
     `,
     lastUpdated: new Date("2024-01-01"),
-    requiresManagerApproval: false
+    requiresManagerApproval: false,
   },
   {
     id: "WFH001",
@@ -61,7 +61,7 @@ export const hrPolicies: HRPolicy[] = [
       - Permanent remote work requires VP approval
     `,
     lastUpdated: new Date("2024-02-15"),
-    requiresManagerApproval: true
+    requiresManagerApproval: true,
   },
   {
     id: "BEN001",
@@ -81,7 +81,7 @@ export const hrPolicies: HRPolicy[] = [
       - New employees: 30-day enrollment window
     `,
     lastUpdated: new Date("2024-01-15"),
-    requiresManagerApproval: false
+    requiresManagerApproval: false,
   },
   {
     id: "PAR001",
@@ -98,6 +98,6 @@ export const hrPolicies: HRPolicy[] = [
       - Flexible return-to-work arrangements available
     `,
     lastUpdated: new Date("2024-01-01"),
-    requiresManagerApproval: true
-  }
-]; 
+    requiresManagerApproval: true,
+  },
+];

package.json

@@ -19,17 +19,19 @@
     "test:all": "yarn test && yarn test:int && yarn lint:langgraph"
   },
   "dependencies": {
-    "@langchain/core": "^0.1.48",
-    "@langchain/langgraph": "^0.0.12",
+    "@langchain/core": "^0.3.40",
+    "@langchain/langgraph": "^0.2.46",
+    "@langchain/langgraph-checkpoint-mongodb": "^0.0.5",
     "@langchain/mongodb": "^0.1.0",
-    "@langchain/openai": "^0.0.14",
+    "@langchain/openai": "^0.4.4",
     "dotenv": "^16.4.5",
-    "langchain": "^0.1.21",
+    "langchain": "^0.3.15",
     "mongodb": "^6.3.0",
     "ts-node": "^10.9.2",
     "zod": "^3.22.4"
   },
   "devDependencies": {
+    "@tsconfig/recommended": "^1.0.8",
     "@types/jest": "^29.5.0",
     "@types/node": "^20.14.8",
     "@typescript-eslint/eslint-plugin": "^5.59.8",

scripts/seed-db.ts

@@ -17,11 +17,11 @@ async function createSearchIndex(db: any) {
 
     // Create Atlas Vector Search index
     await db.collection("hr_policies").createIndex(
-      { "vector_embedding": 1 },
+      { vector_embedding: 1 },
       {
         name: "vector_index",
-        background: true
-      }
+        background: true,
+      },
     );
 
     // Create Atlas Vector Search index definition
@@ -32,19 +32,21 @@ async function createSearchIndex(db: any) {
           vector_embedding: {
             dimensions: 1536,
             similarity: "cosine",
-            type: "knnVector"
-          }
-        }
-      }
+            type: "knnVector",
+          },
+        },
+      },
     };
 
     // Create the vector search index using the Atlas Search API
     await db.command({
       createSearchIndexes: "hr_policies",
-      indexes: [{
-        name: "vector_search_index",
-        definition: indexDefinition
-      }]
+      indexes: [
+        {
+          name: "vector_search_index",
+          definition: indexDefinition,
+        },
+      ],
     });
 
     console.log("Created Atlas Vector Search index");
@@ -70,11 +72,11 @@ async function generateEmbeddings(policies: typeof hrPolicies) {
           lastUpdated: policy.lastUpdated,
           requiresManagerApproval: policy.requiresManagerApproval,
         },
-      })
+      }),
   );
 
   const vectors = await embeddings.embedDocuments(
-    documents.map((doc) => doc.pageContent)
+    documents.map((doc) => doc.pageContent),
   );
 
   return documents.map((doc, i) => ({
@@ -92,33 +94,33 @@ async function seedDatabase() {
     console.log("Connected to MongoDB");
 
     const db = client.db();
-    
+
     // Create collections if they don't exist
     await db.createCollection("hr_policies");
     await db.createCollection("checkpoints");
-    
+
     // Clear existing policies
     await db.collection("hr_policies").deleteMany({});
-    
+
     // Generate embeddings and insert policies
     console.log("Generating embeddings for HR policies...");
     const policiesWithEmbeddings = await generateEmbeddings(hrPolicies);
-    
+
     // Insert policies with embeddings
-    const result = await db.collection("hr_policies").insertMany(policiesWithEmbeddings);
+    const result = await db
+      .collection("hr_policies")
+      .insertMany(policiesWithEmbeddings);
     console.log(`Inserted ${result.insertedCount} HR policies with embeddings`);
 
     // Create vector search index
     console.log("Creating vector search index...");
     await createSearchIndex(db);
 
     // Create TTL index for checkpoints (expire after 24 hours of inactivity)
-    await db.collection("checkpoints").createIndex(
-      { "ts": 1 },
-      { expireAfterSeconds: 86400 }
-    );
+    await db
+      .collection("checkpoints")
+      .createIndex({ ts: 1 }, { expireAfterSeconds: 86400 });
     console.log("Created TTL index for checkpoints");
-
   } catch (error) {
     console.error("Error seeding database:", error);
     process.exit(1);
@@ -129,4 +131,4 @@ async function seedDatabase() {
 }
 
 // Run seeding
-seedDatabase().catch(console.error); 
+seedDatabase().catch(console.error);

src/index.ts

@@ -37,15 +37,13 @@ async function main() {
 
     try {
       // Start the workflow with thread configuration
-      const stream = await agent.stream([
-        new HumanMessage(question)
-      ], config);
+      const stream = await agent.stream([new HumanMessage(question)], config);
 
       for await (const step of stream) {
         for (const [taskName, update] of Object.entries(step)) {
           // Skip the final agent output
           if (taskName === "hr_support") continue;
-          
+
           const message = update as BaseMessage;
           console.log(`\n${taskName}:`);
           prettyPrintMessage(message);
@@ -54,27 +52,34 @@ async function main() {
         // Handle interrupt for human assistance
         if ("__interrupt__" in step) {
           console.log("\nNeeds human assistance!");
-          console.log("Current AI Response:", step.__interrupt__[0].value.current_response);
+          console.log(
+            "Current AI Response:",
+            step.__interrupt__[0].value.current_response,
+          );
           console.log("Waiting for human guidance...");
 
           // In a real application, you would get the human response from a UI
           // For this demo, we'll simulate a human response
-          const humanResponse = "Please refer the employee to their manager for guidance on this topic.";
-          
+          const humanResponse =
+            "Please refer the employee to their manager for guidance on this topic.";
+
           // Resume the workflow with the human response
           const resumeStream = await agent.stream(
-            new Command({ resume: humanResponse }), 
-            config
+            new Command({ resume: humanResponse }),
+            config,
           );
 
           for await (const resumeStep of resumeStream) {
             if (resumeStep.hr_support) {
               console.log("\nFinal Response:");
               console.log("AI Response:", resumeStep.hr_support.ai_response);
-              console.log("Human Guidance:", resumeStep.hr_support.human_response);
+              console.log(
+                "Human Guidance:",
+                resumeStep.hr_support.human_response,
+              );
             }
           }
-        } 
+        }
         // Handle normal response
         else if (step.hr_support) {
           console.log("\nFinal Response:");
@@ -87,4 +92,4 @@ async function main() {
   }
 }
 
-main().catch(console.error); 
+main().catch(console.error);

src/memory_agent/checkpointer.ts

@@ -6,6 +6,6 @@ export async function createCheckpointer() {
 
   return new MongoDBSaver({
     client,
-    checkpointCollectionName: "checkpoints"
+    checkpointCollectionName: "checkpoints",
   });
-} 
+}

src/memory_agent/configuration.ts

@@ -15,7 +15,7 @@ export function ensureConfiguration(config?: LangGraphRunnableConfig) {
   const configurable = config?.configurable || {};
   return {
     userId: configurable?.userId || "default",
-    model: configurable?.model || "openai/o3-mini",
+    model: configurable?.model || "openai:o3-mini",
     systemPrompt: configurable?.systemPrompt || AGENT_PROMPT,
   };
 }