Mastra Plugin

import { URPC } from "@unilab/urpc-hono";
import { MastraPlugin } from "@unilab/mastra-plugin";
import { getMastraInstance } from "@unilab/mastra-plugin/agents";
import { UserEntity } from "./entities/user";
import { UserAdapter } from "./adapters/user";

const MyPlugin = {
  entities: [UserEntity],
  adapters: [
    {
      entity: "UserEntity",
      source: "database",
      adapter: new UserAdapter(),
    },
  ],
};

const app = URPC.init({
  plugins: [
    MyPlugin,
    MastraPlugin({
      mastraInstance: getMastraInstance({
        URPC,
        openrouterApiKey: process.env.OPENROUTER_API_KEY!,
        // debug: true,
      }),
      defaultAgent: "urpcSimpleAgent",
    }),
  ],
});

export default {
  port: 3000,
  fetch: app.fetch,
};

import { ChatEntity } from "@unilab/mastra-plugin/entities";
import { repo, URPC } from "@unilab/urpc";

URPC.init({
  plugins: [MastraClientPlugin()],

  baseUrl: "http://localhost:3000",
  timeout: 10000,
});

// Generate urpc code on the server side and execute it
const chatWithAI = async () => {
  const result = await repo<ChatEntity>({
    entity: "chat",
    source: "mastra",
  }).call({
    input: "Find all users with email containing 'example.com'",
    model: "google/gemini-2.0-flash-001",
    entities: ["UserEntity"],
    // agent: "urpcSimpleAgent",
  });

  console.log(result.output);
};

// Generate urpc code on the server side, execute AI generated code on the front end
const chatWithAIProxy = async () => {
  const result = await repo<ChatEntity>({
    entity: "chat",
    source: "mastra-client",
  }).call({
    input: "Find all users with email containing 'example.com'",
    model: "google/gemini-2.0-flash-001",
    entities: ["UserEntity"],
  });
  console.log(result.output);
};