added pnpm

VisionaireLabs · Dec 27, 2024 · ab9259d · ab9259d
1 parent 954e49e
commit ab9259d
Show file tree

Hide file tree

Showing 11 changed files with 206 additions and 4 deletions.
diff --git a/examples/agent-kit-langgraph/.env.example b/examples/agent-kit-langgraph/.env.example
@@ -0,0 +1,4 @@
+OPENAI_API_KEY=
+RPC_URL=
+SOLANA_PRIVATE_KEY=
+TAVILY_API_KEY=       #Optional: for search functionality
diff --git a/examples/agent-kit-langgraph/.gitignore b/examples/agent-kit-langgraph/.gitignore
@@ -0,0 +1,11 @@
+# Dependencies
+node_modules/
+package-lock.json
+
+# Environment variables
+.env
+.env.local
+
+# Build output
+dist/
+build/
diff --git a/examples/agent-kit-langgraph/package.json b/examples/agent-kit-langgraph/package.json
@@ -3,8 +3,10 @@
   "version": "1.0.0",
   "description": "",
   "main": "index.js",
+  "type": "module",
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "test": "echo \"Error: no test specified\" && exit 1",
+    "dev": "node --loader ts-node/esm"
   },
   "keywords": [],
   "author": "",
@@ -15,5 +17,13 @@
     "@langchain/langgraph": "^0.2.36",
     "dotenv": "^16.4.7",
     "solana-agent-kit": "^1.3.0"
+  },
+  "devDependencies": {
+    "ts-node": "^10.9.2",
+    "typescript": "^5.0.0"
+  },
+  "ts-node": {
+    "esm": true,
+    "experimentalSpecifierResolution": "node"
   }
 }
diff --git a/examples/agent-kit-langgraph/pnpm-lock.yaml b/examples/agent-kit-langgraph/pnpm-lock.yaml
diff --git a/examples/agent-kit-langgraph/src/agents/generalAgent.ts b/examples/agent-kit-langgraph/src/agents/generalAgent.ts
@@ -0,0 +1,32 @@
+import { createReactAgent } from "@langchain/langgraph/prebuilt";
+import { gpt4o } from "../utils/model.js";
+import { solanaAgentState } from "../utils/state.js";
+import { TavilySearchResults } from "@langchain/community/tools/tavily_search";
+import { HumanMessage } from "@langchain/core/messages";
+
+// Initialize tools array
+const searchTools = [];
+
+// Only add Tavily search if API key is available
+if (process.env.TAVILY_API_KEY) {
+  searchTools.push(new TavilySearchResults());
+}
+
+export const generalAgent = createReactAgent({
+  llm: gpt4o,
+  tools: searchTools,
+});
+
+export const generalistNode = async (state: typeof solanaAgentState.State) => {
+  const { messages } = state;
+
+  const result = await generalAgent.invoke({ messages });
+
+  return { messages: [...result.messages] };
+};
+
+const messages = [new HumanMessage("What is the best way to buy SOL?")];
+
+const result = await generalAgent.invoke({ messages });
+
+console.log(result.messages);
diff --git a/examples/agent-kit-langgraph/index.ts → examples/agent-kit-langgraph/src/index.ts b/examples/agent-kit-langgraph/index.ts → examples/agent-kit-langgraph/src/index.ts
diff --git a/examples/agent-kit-langgraph/utils/model.ts → ...es/agent-kit-langgraph/src/utils/model.ts b/examples/agent-kit-langgraph/utils/model.ts → ...es/agent-kit-langgraph/src/utils/model.ts
diff --git a/examples/agent-kit-langgraph/utils/route.ts → ...es/agent-kit-langgraph/src/utils/route.ts b/examples/agent-kit-langgraph/utils/route.ts → ...es/agent-kit-langgraph/src/utils/route.ts
diff --git a/.../agent-kit-langgraph/utils/solanaAgent.ts → ...nt-kit-langgraph/src/utils/solanaAgent.ts b/.../agent-kit-langgraph/utils/solanaAgent.ts → ...nt-kit-langgraph/src/utils/solanaAgent.ts
diff --git a/examples/agent-kit-langgraph/utils/state.ts → ...es/agent-kit-langgraph/src/utils/state.ts b/examples/agent-kit-langgraph/utils/state.ts → ...es/agent-kit-langgraph/src/utils/state.ts