From d28c57363717d6b971cca6facf488e443509cf9d Mon Sep 17 00:00:00 2001 From: yuiseki Date: Sun, 29 Dec 2024 19:13:45 +0900 Subject: [PATCH] WIP --- scripts/agent.ts | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) diff --git a/scripts/agent.ts b/scripts/agent.ts index 18d26368..6691c556 100644 --- a/scripts/agent.ts +++ b/scripts/agent.ts @@ -1,14 +1,8 @@ import { ChatOllama } from "@langchain/ollama"; -import { AIMessage, BaseMessage, HumanMessage } from "@langchain/core/messages"; - -// Tools -import { Tool } from "langchain/tools"; -import { Wikipedia } from "../src/utils/langchain/tools/wikipedia/index.ts"; +import { HumanMessage } from "@langchain/core/messages"; // langgraph -import { createReactAgent } from "@langchain/langgraph/prebuilt"; - -const tools: Array = [new Wikipedia()]; +import { loadWikipediaAgent } from "./agents/wikipedia.ts"; const model = new ChatOllama({ // 速いがツールを使わずに返答しちゃう @@ -28,13 +22,7 @@ const model = new ChatOllama({ temperature: 0, }); -const prompt = - "You are a Wikipedia researcher. Be sure to search Wikipedia and reply based on the results. You have up to 10 chances to search Wikipedia."; -const agent = createReactAgent({ - llm: model, - tools: tools, - stateModifier: prompt, -}); +const agent = await loadWikipediaAgent(model); // Use the agent const stream = await agent.stream(