agent.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4e51ba54-3bfd-4384-b476-e793544191d0",
   "metadata": {},
   "outputs": [],
   "source": [
    "import { load } from \"dotenv\";\n",
    "const env = await load();\n",
    "\n",
    "const process = {\n",
    "    env\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a8216718-c062-4755-a749-c9b80ff8467c",
   "metadata": {},
   "outputs": [],
   "source": [
    "import { z } from \"zod\";\n",
    "import { ChatOpenAI } from \"@langchain/openai\";\n",
    "import { JsonOutputToolsParser } from \"@langchain/core/output_parsers/openai_tools\";\n",
    "import { RunnableSequence, RunnableBranch, RunnablePassthrough } from \"@langchain/core/runnables\";\n",
    "import { zodToJsonSchema } from \"zod-to-json-schema\";\n",
    "import { ChatPromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\n",
    " \n",
    "const classifySchema = z.object({\n",
    "    type: z.enum([\"科普\", \"编程\", \"一般问题\"]).describe(\"用户提问的分类\")\n",
    "})\n",
    "\n",
    "const model = new ChatOpenAI({\n",
    "    temperature: 0 \n",
    "})\n",
    "\n",
    "const modelWithTools = model.bind({\n",
    "    tools: [\n",
    "        {\n",
    "            type: \"function\",\n",
    "            function: {\n",
    "                name: \"classifyQuestion\",\n",
    "                description: \"对用户的提问进行分类\",\n",
    "                parameters: zodToJsonSchema(classifySchema),\n",
    "            }\n",
    "        }\n",
    "    ],\n",
    "    tool_choice: {\n",
    "        type: \"function\",\n",
    "        function: {\n",
    "           name: \"classifyQuestion\"\n",
    "        }\n",
    "    }\n",
    "})\n",
    "\n",
    "const prompt = ChatPromptTemplate.fromMessages([\n",
    "    [\"system\", `仔细思考，你有充足的时间进行严谨的思考，然后对用户的问题进行分类，\n",
    "    当你无法分类到特定分类时，可以分类到 \"一般问题\"`],\n",
    "    [\"human\", \"{input}\"]\n",
    "])\n",
    "\n",
    "const classifyChain = RunnableSequence.from([\n",
    "    prompt,\n",
    "    modelWithTools,\n",
    "    new JsonOutputToolsParser(),\n",
    "    (input) => {\n",
    "        const type = input[0]?.args?.type\n",
    "        return type ? type : \"一般问题\"\n",
    "    }\n",
    "])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4dd65030-e4a8-4dae-b859-42c54251012e",
   "metadata": {},
   "outputs": [],
   "source": [
    "await classifyChain.invoke({\n",
    "    \"input\": \"鲸鱼是哺乳动物么？\"\n",
    "})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b3817a63-926c-4d2c-bc15-b630d372668d",
   "metadata": {},
   "outputs": [],
   "source": [
    "import { StringOutputParser } from \"@langchain/core/output_parsers\";\n",
    "\n",
    "const answeringModel = new ChatOpenAI({\n",
    "    temperature: 0.7,\n",
    "})\n",
    "\n",
    "const sciencePrompt = PromptTemplate.fromTemplate(\n",
    "  `作为一位科普专家，你需要解答以下问题，尽可能提供详细、准确和易于理解的答案：\n",
    "\n",
    "问题：{input}\n",
    "答案：`\n",
    ")\n",
    "    \n",
    "const programmingPrompt = PromptTemplate.fromTemplate(\n",
    "  `作为一位编程专家，你需要解答以下编程相关的问题，尽可能提供详细、准确和实用的答案：\n",
    "\n",
    "问题：{input}\n",
    "答案：`\n",
    ")\n",
    "\n",
    "const generalPrompt = PromptTemplate.fromTemplate(\n",
    "  `请回答以下一般性问题，尽可能提供全面和有深度的答案：\n",
    "\n",
    "问题：{input}\n",
    "答案：`\n",
    ")\n",
    "\n",
    "\n",
    "const scienceChain = RunnableSequence.from([\n",
    "    sciencePrompt,\n",
    "    answeringModel,\n",
    "    new StringOutputParser(),\n",
    "    {\n",
    "        output: input => input,\n",
    "        role: () => \"科普专家\"\n",
    "    }\n",
    "    \n",
    "])\n",
    "\n",
    "const programmingChain = RunnableSequence.from([\n",
    "    programmingPrompt,\n",
    "    answeringModel,\n",
    "    new StringOutputParser(),\n",
    "    {\n",
    "        output: input => input,\n",
    "        role: () => \"编程大师\"\n",
    "    }\n",
    "    \n",
    "])\n",
    "\n",
    "const generalChain = RunnableSequence.from([\n",
    "    generalPrompt,\n",
    "    answeringModel,\n",
    "    new StringOutputParser(),\n",
    "    {\n",
    "        output: input => input,\n",
    "        role: () => \"通识专家\"\n",
    "    }\n",
    "    \n",
    "])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6e76cf62-cbdf-4a72-aa0f-9173b7e81aa5",
   "metadata": {},
   "outputs": [],
   "source": [
    "const branch = RunnableBranch.from([\n",
    "  [\n",
    "    (input => input.type.includes(\"科普\")),\n",
    "    scienceChain,\n",
    "  ],\n",
    "  [\n",
    "    (input => input.type.includes(\"编程\")),\n",
    "    programmingChain,\n",
    "  ],\n",
    "  generalChain\n",
    "]);"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "da1fe98b-8887-452e-8007-a2aa78992e98",
   "metadata": {},
   "outputs": [],
   "source": [
    "const outputTemplate = PromptTemplate.fromTemplate(\n",
    "`感谢您的提问，这是来自 {role} 的专业回答：\n",
    "\n",
    "{output}\n",
    "`)\n",
    "\n",
    "\n",
    "const finalChain = RunnableSequence.from([\n",
    "    {\n",
    "        type: classifyChain,\n",
    "        input: input => input.input\n",
    "    },\n",
    "    branch,\n",
    "    (input) => outputTemplate.format(input),\n",
    "])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "543fb03e-c1ad-404a-a5a8-36ed84d726be",
   "metadata": {},
   "outputs": [],
   "source": [
    "const res = await finalChain.invoke({\n",
    "    \"input\": \"鲸鱼是哺乳动物么？\"\n",
    "})\n",
    "\n",
    "console.log(res)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "57dafeab-a265-4db2-8111-0691486cf6da",
   "metadata": {},
   "outputs": [],
   "source": [
    "const route = ({ type }) => {\n",
    "    if(type.includes(\"科普\")){\n",
    "        return scienceChain\n",
    "    }else if(type.includes(\"编程\")){\n",
    "        return programmingChain\n",
    "    }\n",
    "\n",
    "    return generalChain\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "bb9a09a0-9d4a-4a1c-a259-15d4cabd9492",
   "metadata": {},
   "outputs": [],
   "source": [
    "const finalChain2 = RunnableSequence.from([\n",
    "    {\n",
    "        type: classifyChain,\n",
    "        input: input => input.input\n",
    "    },\n",
    "    route,\n",
    "    (input) => outputTemplate.format(input),\n",
    "])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ffd3d7ee-85c1-4ae8-8e59-88d214e2e8ed",
   "metadata": {},
   "outputs": [],
   "source": [
    "await finalChain2.invoke({\n",
    "    \"input\": \"鲸鱼是哺乳动物么？\"\n",
    "})"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Deno",
   "language": "typescript",
   "name": "deno"
  },
  "language_info": {
   "file_extension": ".ts",
   "mimetype": "text/x.typescript",
   "name": "typescript",
   "nb_converter": "script",
   "pygments_lexer": "typescript",
   "version": "5.3.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}