1// File llama.ts2import { ChatOllama } from "@langchain/ollama";34const llama = new ChatOllama({5  model: "llama3.2:latest",6});78export default llama;91011// File route.ts1213import llama from "@/models/llama";14import { AIMessage, HumanMessage } from "@langchain/core/messages";15import { tool } from "@langchain/core/tools";16import { LangChainAdapter, type Message } from "ai";17import { NextResponse } from "next/server";18import { z } from "zod";1920export async function POST(req: Request) {21  const { messages } = await req.json();22  try {23    const response = await llama24      .bindTools([25        tool(26          async (input) => {27            console.log(input);28            return input.a * input.b;29          },30          {31            name: "multiply",32            description: "Multiplies a and b.",33            schema: z.object({34              a: z.number(),35              b: z.number(),36            }),37          }38        ),39      ])40      .stream(41        messages.map((message: Message) =>42          message.role == "user"43            ? new HumanMessage(message.content)44            : new AIMessage(message.content)45        )46      );47    return LangChainAdapter.toDataStreamResponse(response);48  } catch (error: any) {49    console.log(error);50    return NextResponse.json({ error: error.message }, { status: 500 });51  }52}535455// useChat function5657const { handleSubmit, input, handleInputChange, messages } = useChat({58    api: "/api/bot",59    onError(error) {60      console.log(error);61    },62    onResponse(response) {63      console.log(response);64    },65    onFinish(message, options) {66      console.log(message);67    },68  });