@xalia/agent 0.5.8 → 0.6.1

package/src/chat/db.ts

@@ -1,304 +0,0 @@
-import { createClient, SupabaseClient } from "@supabase/supabase-js";
-import type * as supabase from "../../../supabase/database.types";
-import {
-  AgentProfile,
-  ApiKey,
-  getLogger,
-  SavedAgentProfile,
-} from "@xalia/xmcp/sdk";
-import { ChatCompletionMessageParam } from "../agent/agent";
-
-const logger = getLogger();
-
-export const SUPABASE_LOCAL_URL = "http://127.0.0.1:54321";
-export const SUPABASE_LOCAL_KEY =
-  "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZS1kZW1vIiw" +
-  "icm9sZSI6InNlcnZpY2Vfcm9sZSIsImV4cCI6MTk4MzgxMjk5Nn0.EGIM96RAZx35lJz" +
-  "dJsyH-qQwv8Hdp7fsn3W0YpN81IU";
-
-/**
- * 'name' -> 'name'
- * 'space/name' -> ['space', 'name']
- */
-export function resolveCompoundName(name: string): string | [string, string] {
-  const components = name.split("/");
-  if (components.length === 1) {
-    return name;
-  }
-  if (components.length !== 2) {
-    throw "invalid compound name";
-  }
-  return components as [string, string];
-}
-
-export type UserData = {
-  uuid: supabase.Tables<"api_keys">["user_uuid"];
-  nickname: supabase.Tables<"users">["nickname"];
-};
-
-export class Database {
-  private client: SupabaseClient<Database>;
-
-  constructor(supabaseUrl: string, supabaseKey: string) {
-    this.client = createClient<Database>(supabaseUrl, supabaseKey);
-  }
-
-  async getUserDataFromApiKey(apiKey: string): Promise<UserData | undefined> {
-    const { data, error } = await this.client
-      .from("api_keys")
-      .select("user_uuid, users ( nickname )")
-      .eq("api_key", apiKey)
-      .maybeSingle<{
-        user_uuid: supabase.Tables<"api_keys">["user_uuid"];
-        users: {
-          nickname: supabase.Tables<"users">["nickname"];
-        };
-      }>();
-
-    logger.debug(
-      `[getUserDataFromApiKey]: got ${JSON.stringify({ data, error })}`
-    );
-
-    if (error) {
-      throw error;
-    }
-
-    if (data === null) {
-      return undefined;
-    }
-
-    return {
-      uuid: data.user_uuid,
-      nickname: data.users.nickname || `user ${data.user_uuid}`,
-    };
-  }
-
-  async getUserFromUuid(user_uuid: string): Promise<UserData | undefined> {
-    const { data, error } = await this.client
-      .from("users")
-      .select("*")
-      .eq("uuid", user_uuid)
-      .maybeSingle();
-    if (error) {
-      throw error;
-    }
-
-    return data;
-  }
-
-  async createUser(
-    user_uuid: string,
-    email: string,
-    nickname: string,
-    timezone?: string
-  ): Promise<void> {
-    const payload: supabase.TablesInsert<"users"> = {
-      uuid: user_uuid,
-      email,
-      nickname,
-      timezone: timezone || "UTC",
-    };
-
-    const { error } = await this.client.from("users").insert(payload);
-    if (error) {
-      throw error;
-    }
-  }
-
-  async addApiKey(
-    user_uuid: string,
-    api_key: string,
-    name: string,
-    scopes: string[],
-    is_default: boolean = false
-  ): Promise<ApiKey> {
-    const payload: supabase.TablesInsert<"api_keys"> = {
-      user_uuid,
-      api_key,
-      name,
-      scopes,
-      is_default,
-    };
-    const { data, error } = await this.client
-      .from("api_keys")
-      .insert(payload)
-      .select("*")
-      .maybeSingle();
-    if (error) {
-      throw error;
-    }
-
-    return data;
-  }
-
-  async getSavedAgentProfileById(
-    agentProfileId: string
-  ): Promise<SavedAgentProfile | undefined> {
-    const { data, error } = await this.client
-      .from("agent_profiles")
-      .select("*")
-      .eq("uuid", agentProfileId)
-      .maybeSingle<supabase.Tables<"agent_profiles">>();
-    if (error) {
-      throw error;
-    }
-
-    return data
-      ? SavedAgentProfile.fromJSONObj(data as Record<string, unknown>)
-      : undefined;
-  }
-
-  async getSavedAgentProfileByName(
-    user_uuid: string,
-    agentProfileName: string
-  ): Promise<SavedAgentProfile | undefined> {
-    const { data, error } = await this.client
-      .from("agent_profiles")
-      .select("*")
-      .eq("user_uuid", user_uuid)
-      .eq("profile_name", agentProfileName)
-      .maybeSingle<supabase.Tables<"agent_profiles">>();
-    if (error) {
-      throw error;
-    }
-
-    return data
-      ? SavedAgentProfile.fromJSONObj(data as Record<string, unknown>)
-      : undefined;
-  }
-
-  async getAgentProfileById(
-    agentProfileId: string
-  ): Promise<AgentProfile | undefined> {
-    const { data, error } = await this.client
-      .from("agent_profiles")
-      .select("profile")
-      .eq("uuid", agentProfileId)
-      .maybeSingle<{
-        profile: supabase.Tables<"agent_profiles">["profile"];
-      }>();
-    if (error) {
-      throw error;
-    }
-    return data
-      ? AgentProfile.fromJSONObj(data.profile as Record<string, unknown>)
-      : undefined;
-  }
-
-  async createAgentProfile(
-    user_uuid: string,
-    profileName: string,
-    profile: AgentProfile
-  ): Promise<string | undefined> {
-    const payload: supabase.TablesInsert<"agent_profiles"> = {
-      profile: profile as unknown as supabase.Json,
-      user_uuid,
-      profile_name: profileName,
-    };
-    const { data, error } = await this.client
-      .from("agent_profiles")
-      .upsert(payload)
-      .select("uuid");
-    if (error) {
-      throw error;
-    }
-
-    if (!data || !data[0] || !data[0].uuid) {
-      return undefined;
-    }
-
-    return data[0].uuid;
-  }
-
-  async updateAgentProfile(uuid: string, profile: AgentProfile): Promise<void> {
-    const payload: supabase.TablesUpdate<"agent_profiles"> = {
-      profile: profile as unknown as supabase.Json,
-    };
-    const { error } = await this.client
-      .from("agent_profiles")
-      .update(payload)
-      .eq("uuid", uuid);
-    if (error) {
-      throw error;
-    }
-  }
-
-  async clearAgentProfiles(): Promise<void> {
-    await this.client.from("agent_profiles").delete().neq("uuid", "");
-  }
-
-  // TODO: is there a session model?
-
-  async getSessionById(
-    session_uuid: string
-  ): Promise<supabase.Tables<"sessions"> | undefined> {
-    const { data, error } = await this.client
-      .from("sessions")
-      .select("*")
-      .eq("uuid", session_uuid)
-      .maybeSingle();
-    if (error) {
-      throw error;
-    }
-    return data;
-  }
-
-  async getSessionByName(
-    user_uuid: string,
-    session_name: string
-  ): Promise<supabase.Tables<"sessions"> | undefined> {
-    const { data, error } = await this.client
-      .from("sessions")
-      .select("*")
-      .eq("user_uuid", user_uuid)
-      .eq("title", session_name)
-      .maybeSingle();
-    if (error) {
-      throw error;
-    }
-    return data;
-  }
-
-  async createSession(
-    user_uuid: string,
-    title: string,
-    agentProfileId: string
-  ): Promise<string | undefined> {
-    const payload: supabase.TablesInsert<"sessions"> = {
-      agent_profile_uuid: agentProfileId,
-      user_uuid,
-      title: title,
-      conversation: [],
-    };
-    const { data, error } = await this.client
-      .from("sessions")
-      .upsert(payload)
-      .select("uuid");
-    if (error) {
-      throw error;
-    }
-
-    if (!data || !data[0] || !data[0].uuid) {
-      return undefined;
-    }
-
-    return data[0].uuid;
-  }
-
-  async sessionConversationAppend(
-    session_id: string,
-    newEntries: ChatCompletionMessageParam[]
-  ): Promise<void> {
-    const { error } = await this.client.rpc("session_append_to_conversation", {
-      session_id,
-      messages: newEntries,
-    });
-    if (error) {
-      throw error;
-    }
-  }
-
-  async clearSessions(): Promise<void> {
-    await this.client.from("sessions").delete().neq("uuid", "");
-  }
-}

package/src/chat/messages.ts

@@ -1,266 +0,0 @@
-import {
-  ChatCompletionMessageParam,
-  ChatCompletionAssistantMessageParam,
-  ChatCompletionToolMessageParam,
-  ChatCompletionMessageToolCall,
-} from "openai/resources.mjs";
-import { Tool } from "@modelcontextprotocol/sdk/types.js";
-import { McpServerBrief } from "@xalia/xmcp/sdk";
-
-/**
- * Message from a user to the server
- */
-export type ClientUserMessage = {
-  type: "msg";
-  message: string;
-};
-
-/**
- * Message to add an MCP server to the agent
- */
-export type ClientAddMcpServer = {
-  type: "add_mcp_server";
-  server_name: string;
-  enable_all: boolean;
-};
-
-/**
- * Message to remove an MCP server from the agent
- */
-export type ClientRemoveMcpServer = {
-  type: "remove_mcp_server";
-  server_name: string;
-};
-
-/**
- * Enable a specific tool
- */
-export type ClientEnableMcpServerTool = {
-  type: "enable_mcp_server_tool";
-  server_name: string;
-  tool: string;
-};
-
-/**
- * Disable a specific tool
- */
-export type ClientDisableMcpServerTool = {
-  type: "disable_mcp_server_tool";
-  server_name: string;
-  tool: string;
-};
-
-export type ClientEnableAllMcpServerTools = {
-  type: "enable_all_mcp_server_tools";
-  server_name: string;
-};
-
-export type ClientDisableAllMcpServerTools = {
-  type: "disable_all_mcp_server_tools";
-  server_name: string;
-};
-
-export type ClientSetSystemPrompt = {
-  type: "set_system_prompt";
-  system_prompt: string;
-};
-
-export type ClientSetModel = {
-  type: "set_model";
-  model: string;
-};
-
-export type ClientToServer =
-  | ClientUserMessage
-  | ClientAddMcpServer
-  | ClientRemoveMcpServer
-  | ClientEnableMcpServerTool
-  | ClientDisableMcpServerTool
-  | ClientEnableAllMcpServerTools
-  | ClientDisableAllMcpServerTools
-  | ClientSetSystemPrompt
-  | ClientSetModel;
-
-///
-/// Server -> Client Messages
-///
-
-/**
- * Error in response to an invalid request from the client
- */
-export type ServerError = {
-  type: "error";
-  message: string;
-};
-
-/**
- * (from server) Chat history
- */
-export type ServerHistory = {
-  type: "history";
-  /// Conversation history in the form we expect.  The `name` attribute of
-  /// `ChatCompletionUserMessageParam` (role: "user") holds the original
-  /// message sender.
-  messages: ChatCompletionMessageParam[];
-};
-
-export type ServerUserJoined = {
-  type: "user_joined";
-  user: string;
-};
-
-export type ServerUserLeft = {
-  type: "user_left";
-  user: string;
-};
-
-/**
- * Message from the server, informing us of a message in the chat history
- */
-export type ServerUserMessage = {
-  type: "user_msg";
-  message_id: string;
-  message: string;
-  from: string;
-};
-
-export type ServerAgentMessage = {
-  type: "agent_msg";
-  message_id: string;
-  message: ChatCompletionAssistantMessageParam;
-};
-
-export type ServerAgentMessageChunk = {
-  type: "agent_msg_chunk";
-  message_id: string;
-  message: string;
-  end: boolean;
-};
-
-export type ServerMcpServerBriefs = {
-  type: "mcp_server_briefs";
-  server_briefs: McpServerBrief[];
-};
-
-/**
- * For information only (to keep the chat window consistent).
- */
-export type ServerToolCall = {
-  type: "agent_tool_call";
-  message_id: string;
-  message: ChatCompletionMessageToolCall;
-};
-
-/**
- * For information only (to keep the chat window consistent)
- */
-export type ServerToolCallResult = {
-  type: "tool_call_result";
-  message_id: string;
-  message: ChatCompletionToolMessageParam;
-};
-
-export type ServerTyping = {
-  type: "typing";
-  from: string;
-};
-
-export type ServerMcpServerAdded = {
-  type: "mcp_server_added";
-  server_name: string;
-  tools: Tool[];
-  enabled_tools: string[];
-};
-
-export type ServerMcpServerRemoved = {
-  type: "mcp_server_removed";
-  server_name: string;
-};
-
-export type ServerMcpServerToolEnabled = {
-  type: "mcp_server_tool_enabled";
-  server_name: string;
-  tool: string;
-};
-
-export type ServerMcpServerToolDisabled = {
-  type: "mcp_server_tool_disabled";
-  server_name: string;
-  tool: string;
-};
-
-export type ServerSystemPromptUpdated = {
-  type: "system_prompt_updated";
-  system_prompt: string;
-};
-
-export type ServerModelUpdated = {
-  type: "model_updated";
-  model: string;
-};
-
-export type ServerToClientStateUpdate =
-  | ServerMcpServerAdded
-  | ServerMcpServerRemoved
-  | ServerMcpServerToolEnabled
-  | ServerMcpServerToolDisabled
-  | ServerSystemPromptUpdated
-  | ServerModelUpdated;
-
-export type ServerOpenUrl = {
-  type: "open_url";
-  url: string;
-  display_name: string;
-};
-
-export type ServerApproveToolCall = {
-  type: "approve_tool_call";
-  tbd: string;
-};
-
-export type ServerToClientActions = ServerOpenUrl | ServerApproveToolCall;
-
-export type ServerToClient =
-  | ServerError
-  | ServerUserJoined
-  | ServerUserLeft
-  | ServerUserMessage
-  | ServerAgentMessage
-  | ServerAgentMessageChunk
-  | ServerMcpServerBriefs
-  | ServerToolCall
-  | ServerToolCallResult
-  | ServerTyping
-  | ServerToClientStateUpdate
-  | ServerToClientActions;
-
-export function decodeAssistantMessageParam(
-  msg: ChatCompletionAssistantMessageParam
-): string {
-  let text = "";
-
-  if (msg.audio) {
-    throw "decodeAssistantMessageParam; audio unimplemented";
-  }
-
-  if (msg.content) {
-    if (typeof msg.content === "string") {
-      text = msg.content;
-    } else if (msg.content instanceof Array) {
-      for (const c of msg.content) {
-        switch (c.type) {
-          case "text":
-            text += c.text;
-            break;
-          case "refusal":
-            text += c.refusal;
-            break;
-          default:
-            throw Error("unexpected AssistantMessageParam.content entry");
-        }
-      }
-    }
-  }
-
-  return text;
-}