@nebutra/agents 1.1.0 → 1.1.2

package/src/sdk/provider.ts

@@ -1,80 +0,0 @@
-import { createOpenAI } from "@ai-sdk/openai";
-import { createOpenRouter } from "@openrouter/ai-sdk-provider";
-import type { EmbeddingModel, LanguageModel } from "ai";
-import { type ResolvedNebutraAIConfig, resolveApiKey } from "./config";
-import { resolveModel } from "./models";
-
-/**
- * Creates a language model instance based on the resolved config.
- *
- * Provider routing:
- * - "openrouter"   → @openrouter/ai-sdk-provider (300+ models, failover)
- * - "openai"       → @ai-sdk/openai (direct OpenAI API)
- * - "siliconflow"  → @ai-sdk/openai with SiliconFlow baseURL (OpenAI-compatible)
- * - "gateway"      → Vercel AI Gateway via plain model string (OIDC auth)
- */
-export function createModel(modelOrPreset: string, config: ResolvedNebutraAIConfig): LanguageModel {
-  const modelId = resolveModel(modelOrPreset);
-  const apiKey = resolveApiKey(config);
-
-  switch (config.provider) {
-    case "openrouter": {
-      const provider = createOpenRouter({
-        apiKey,
-        ...(config.headers ? { headers: config.headers } : {}),
-        ...(config.extraBody ? { extraBody: config.extraBody } : {}),
-      });
-      return provider.chat(modelId);
-    }
-
-    case "openai": {
-      const provider = createOpenAI({ apiKey });
-      return provider(modelId);
-    }
-
-    case "siliconflow": {
-      // SiliconFlow: OpenAI-compatible API with China-optimized infra
-      // Models use "Vendor/Model" format, e.g. "Qwen/Qwen2.5-72B-Instruct"
-      const provider = createOpenAI({
-        apiKey,
-        baseURL: "https://api.siliconflow.cn/v1",
-      });
-      return provider(modelId);
-    }
-
-    case "gateway": {
-      // Vercel AI Gateway: use @ai-sdk/openai with gateway baseURL
-      // OIDC token is passed as apiKey, routed through Vercel's proxy
-      const provider = createOpenAI({
-        apiKey,
-        baseURL: "https://ai-gateway.vercel.sh/v1",
-      });
-      return provider(modelId);
-    }
-  }
-}
-
-/**
- * Creates an embedding model instance for vector operations.
- * Only supported with OpenRouter provider.
- */
-export function createEmbeddingModel(
-  modelOrPreset: string,
-  config: ResolvedNebutraAIConfig,
-): EmbeddingModel {
-  const modelId = resolveModel(modelOrPreset);
-  const apiKey = resolveApiKey(config);
-
-  if (config.provider !== "openrouter") {
-    throw new Error(
-      `[@nebutra/agents] Embedding models are currently only supported with the "openrouter" provider.`,
-    );
-  }
-
-  const provider = createOpenRouter({
-    apiKey,
-    ...(config.headers ? { headers: config.headers } : {}),
-  });
-
-  return provider.textEmbeddingModel(modelId);
-}

package/src/tenant.ts

@@ -1,52 +0,0 @@
-/**
- * Tenant-scoped agent execution context.
- *
- * Ensures every agent operation carries tenantId for RLS,
- * billing, and audit trail purposes.
- */
-
-import type { AgentContext } from "./types";
-
-/**
- * Create a fully-populated AgentContext.
- * Generates a random conversationId when none is provided.
- */
-export function createAgentContext(
-  tenantId: string,
-  userId: string,
-  conversationId?: string,
-  metadata?: Record<string, unknown>,
-): AgentContext {
-  return {
-    tenantId,
-    userId,
-    conversationId: conversationId ?? crypto.randomUUID(),
-    ...(metadata !== undefined ? { metadata } : {}),
-  };
-}
-
-/**
- * Validate that a tenant has remaining quota for agent execution.
- *
- * Returns `{ allowed: true, remaining: -1 }` (unlimited) by default.
- * Integrate with @nebutra/billing entitlements for production usage.
- */
-export async function checkAgentQuota(
-  tenantId: string,
-): Promise<{ allowed: boolean; remaining: number }> {
-  try {
-    const { getCreditBalance } = await import("@nebutra/billing/credits");
-    const balance = await getCreditBalance(tenantId);
-
-    // Simple quota check: tenant must have positive credits
-    // In production, we might check plan-specific monthly limits beforehand
-    if (balance.balance > 0) {
-      return { allowed: true, remaining: balance.balance };
-    }
-
-    return { allowed: false, remaining: 0 };
-  } catch (_err) {
-    // Failsafe open if billing is unconfigured
-    return { allowed: true, remaining: -1 };
-  }
-}

package/src/tools.ts

@@ -1,65 +0,0 @@
-/**
- * Built-in tool registry for agents.
- *
- * These are safe, tenant-scoped stubs that demonstrate the tool interface.
- * Each tool requires configuration (API keys, vector stores, etc.) before
- * it produces real results.
- */
-
-import type { AgentTool } from "./types";
-
-/** Search the web for current information. */
-export const webSearchTool: AgentTool = {
-  name: "web_search",
-  description: "Search the web for current information",
-  inputSchema: {
-    type: "object",
-    properties: { query: { type: "string" } },
-    required: ["query"],
-  },
-  execute: async (_input, _context) => {
-    return {
-      results: [],
-      note: "Configure TAVILY_API_KEY or SERPER_API_KEY for live search",
-    };
-  },
-};
-
-/** Query the tenant's database (tenant-scoped via RLS). */
-export const databaseQueryTool: AgentTool = {
-  name: "database_query",
-  description: "Query the tenant's database",
-  inputSchema: {
-    type: "object",
-    properties: { query: { type: "string" } },
-    required: ["query"],
-  },
-  execute: async (_input, context) => {
-    return {
-      note: `Query would execute in tenant ${context.tenantId} scope`,
-    };
-  },
-};
-
-/** RAG-style knowledge base retrieval. */
-export const knowledgeBaseTool: AgentTool = {
-  name: "knowledge_base",
-  description: "Search the organization's knowledge base for relevant documents",
-  inputSchema: {
-    type: "object",
-    properties: {
-      query: { type: "string" },
-      limit: { type: "number" },
-    },
-  },
-  execute: async (_input, _context) => {
-    return { documents: [], note: "Configure vector store for RAG" };
-  },
-};
-
-/** All pre-built tools. */
-export const BUILT_IN_TOOLS: readonly AgentTool[] = [
-  webSearchTool,
-  databaseQueryTool,
-  knowledgeBaseTool,
-];

package/src/types.ts

@@ -1,114 +0,0 @@
-/**
- * Core types for the multi-agent orchestration engine.
- *
- * All tenant-scoped operations require an AgentContext carrying `tenantId`.
- * Usage tracking is emitted on every execution for downstream billing.
- */
-
-// ─── Agent Configuration ──────────────────────────────────────────────────────
-
-export interface AgentConfig {
-  /** Unique identifier for this agent */
-  readonly id: string;
-  /** Human-readable name */
-  readonly name: string;
-  /** What this agent does (used by the router for intent matching) */
-  readonly description: string;
-  /** Model identifier, e.g. "openai/gpt-5.4" or "anthropic/claude-sonnet-4.6" */
-  readonly model: string;
-  /** System prompt / instructions */
-  readonly instructions: string;
-  /** Tools this agent can invoke */
-  readonly tools?: readonly AgentTool[];
-  /** Maximum tool-loop iterations (default 20) */
-  readonly maxSteps?: number;
-  /** Memory configuration */
-  readonly memory?: MemoryConfig;
-}
-
-// ─── Tools ────────────────────────────────────────────────────────────────────
-
-export interface AgentTool {
-  readonly name: string;
-  readonly description: string;
-  readonly inputSchema: Record<string, unknown>;
-  readonly execute: (input: unknown, context: AgentContext) => Promise<unknown>;
-}
-
-// ─── Execution Context ────────────────────────────────────────────────────────
-
-export interface AgentContext {
-  readonly tenantId: string;
-  readonly userId: string;
-  readonly conversationId: string;
-  readonly metadata?: Record<string, unknown>;
-}
-
-// ─── Messages ─────────────────────────────────────────────────────────────────
-
-export interface AgentMessage {
-  readonly role: "user" | "assistant" | "system" | "tool";
-  readonly content: string;
-  readonly toolCalls?: readonly ToolCallResult[];
-  readonly timestamp: Date;
-}
-
-export interface ToolCallResult {
-  readonly toolName: string;
-  readonly args: unknown;
-  readonly result: unknown;
-}
-
-// ─── Response ─────────────────────────────────────────────────────────────────
-
-export interface AgentResponse {
-  readonly messages: readonly AgentMessage[];
-  readonly usage: TokenUsage;
-  readonly finishReason: string;
-  readonly agentId: string;
-}
-
-export interface TokenUsage {
-  readonly promptTokens: number;
-  readonly completionTokens: number;
-  readonly totalTokens: number;
-}
-
-// ─── Memory ───────────────────────────────────────────────────────────────────
-
-export interface MemoryConfig {
-  readonly shortTerm?: { readonly maxMessages: number };
-  readonly longTerm?: { readonly enabled: boolean };
-}
-
-// ─── Orchestrator ─────────────────────────────────────────────────────────────
-
-export interface OrchestratorConfig {
-  readonly agents: readonly AgentConfig[];
-  readonly router?: RouterConfig;
-  readonly defaultAgentId?: string;
-}
-
-export interface RouterConfig {
-  readonly strategy: "keyword" | "llm" | "custom";
-  readonly customRouter?: (message: string, context: AgentContext) => Promise<string>;
-}
-
-export interface PipelineStep {
-  readonly agentId: string;
-  readonly transformInput?: (prevOutput: string) => string;
-}
-
-// ─── Usage / Billing ──────────────────────────────────────────────────────────
-
-export interface AgentUsageEvent {
-  readonly tenantId: string;
-  readonly userId: string;
-  readonly agentId: string;
-  readonly model: string;
-  readonly promptTokens: number;
-  readonly completionTokens: number;
-  readonly totalTokens: number;
-  readonly durationMs: number;
-  readonly timestamp: Date;
-}

package/tsconfig.json

@@ -1,12 +0,0 @@
-{
-  "extends": "../../../tsconfig.base.json",
-  "compilerOptions": {
-    "module": "ESNext",
-    "moduleResolution": "bundler",
-    "target": "esnext",
-    "types": ["node"],
-    "incremental": false
-  },
-  "include": ["src"],
-  "exclude": ["node_modules", "dist"]
-}

package/tsup.config.ts

@@ -1,21 +0,0 @@
-import { defineConfig } from "tsup";
-
-export default defineConfig({
-  entry: [
-    "src/index.ts",
-    "src/tools.ts",
-    "src/providers/vercel-ai.ts",
-    "src/providers/langchain.ts",
-    "src/sdk/index.ts",
-    "src/sdk/config.ts",
-    "src/sdk/models.ts",
-    "src/sdk/provider.ts",
-  ],
-  format: ["esm"],
-  dts: true,
-  clean: true,
-  target: "node20",
-  outDir: "dist",
-  splitting: true,
-  external: ["ai", "@nebutra/logger", "@nebutra/cache", "@nebutra/billing"],
-});