@hyperspell/openclaw-hyperspell 0.5.0 → 0.7.2

package/client.ts

@@ -1,252 +1,322 @@
-import Hyperspell from "hyperspell"
-import type { HyperspellConfig, HyperspellSource } from "./config.ts"
-import { log } from "./logger.ts"
+import Hyperspell from "hyperspell";
+import type { HyperspellConfig, HyperspellSource } from "./config.ts";
+import { log } from "./logger.ts";
+
+export type Highlight = {
+	id: string;
+	score: number;
+	text: string;
+};
 
 export type SearchResult = {
-  resourceId: string
-  title: string | null
-  source: HyperspellSource
-  score: number | null
-  url: string | null
-  createdAt: string | null
-}
+	resourceId: string;
+	title: string | null;
+	source: HyperspellSource;
+	score: number | null;
+	url: string | null;
+	createdAt: string | null;
+	highlights: Highlight[];
+};
 
 export type SearchWithAnswerResult = {
-  answer: string | null
-  documents: SearchResult[]
-}
+	answer: string | null;
+	documents: SearchResult[];
+};
 
 export type Integration = {
-  id: string
-  name: string
-  provider: HyperspellSource
-  icon: string
-}
+	id: string;
+	name: string;
+	provider: HyperspellSource;
+	icon: string;
+};
 
 export type Connection = {
-  id: string
-  integrationId: string
-  label: string | null
-  provider: HyperspellSource
-}
+	id: string;
+	integrationId: string;
+	label: string | null;
+	provider: HyperspellSource;
+};
 
 export class HyperspellClient {
-  private client: Hyperspell
-  private config: HyperspellConfig
-
-  constructor(config: HyperspellConfig) {
-    this.config = config
-    this.client = new Hyperspell({
-      apiKey: config.apiKey,
-      userID: config.userId,
-    })
-    log.info(`client initialized${config.userId ? ` for user ${config.userId}` : ""}`)
-  }
-
-  async search(
-    query: string,
-    options?: { limit?: number; sources?: HyperspellSource[] },
-  ): Promise<SearchResult[]> {
-    const limit = options?.limit ?? this.config.maxResults
-    const sources =
-      options?.sources ?? (this.config.sources.length > 0 ? this.config.sources : undefined)
-
-    log.debugRequest("memories.search", { query, limit, sources })
-
-    const response = await this.client.memories.search({
-      query,
-      sources,
-      options: {
-        max_results: limit,
-      },
-    })
-
-    const results: SearchResult[] = response.documents.map((doc) => ({
-      resourceId: doc.resource_id,
-      title: doc.title ?? null,
-      source: doc.source as HyperspellSource,
-      score: doc.score ?? null,
-      url: doc.metadata?.url as string | null ?? null,
-      createdAt: doc.metadata?.created_at as string | null ?? null,
-    }))
-
-    log.debugResponse("memories.search", { count: results.length })
-    return results
-  }
-
-  async searchRaw(
-    query: string,
-    options?: { limit?: number; sources?: HyperspellSource[] },
-  ): Promise<Record<string, unknown>> {
-    const limit = options?.limit ?? this.config.maxResults
-    const sources =
-      options?.sources ?? (this.config.sources.length > 0 ? this.config.sources : undefined)
-
-    log.debugRequest("memories.search (raw)", { query, limit, sources })
-
-    const response = await this.client.memories.search({
-      query,
-      sources,
-      options: {
-        max_results: limit,
-      },
-    })
-
-    log.debugResponse("memories.search (raw)", { count: response.documents.length })
-
-    return response as unknown as Record<string, unknown>
-  }
-
-  async searchWithAnswer(
-    query: string,
-    options?: { limit?: number; sources?: HyperspellSource[] },
-  ): Promise<SearchWithAnswerResult> {
-    const limit = options?.limit ?? this.config.maxResults
-    const sources =
-      options?.sources ?? (this.config.sources.length > 0 ? this.config.sources : undefined)
-
-    log.debugRequest("memories.search (with answer)", { query, limit, sources })
-
-    const response = await this.client.memories.search({
-      query,
-      sources,
-      answer: true,
-      options: {
-        max_results: limit,
-      },
-    })
-
-    const documents: SearchResult[] = response.documents.map((doc) => ({
-      resourceId: doc.resource_id,
-      title: doc.title ?? null,
-      source: doc.source as HyperspellSource,
-      score: doc.score ?? null,
-      url: doc.metadata?.url as string | null ?? null,
-      createdAt: doc.metadata?.created_at as string | null ?? null,
-    }))
-
-    log.debugResponse("memories.search (with answer)", {
-      count: documents.length,
-      hasAnswer: !!response.answer,
-    })
-
-    return {
-      answer: response.answer ?? null,
-      documents,
-    }
-  }
-
-  async addMemory(
-    text: string,
-    options?: {
-      title?: string
-      resourceId?: string
-      collection?: string
-      metadata?: Record<string, string | number | boolean>
-    },
-  ): Promise<{ resourceId: string }> {
-    log.debugRequest("memories.add", {
-      textLength: text.length,
-      title: options?.title,
-      resourceId: options?.resourceId,
-      collection: options?.collection,
-    })
-
-    const result = await this.client.memories.add({
-      text,
-      title: options?.title,
-      resource_id: options?.resourceId,
-      collection: options?.collection,
-      metadata: {
-        ...options?.metadata,
-        openclaw_source: "command",
-      },
-    })
-
-    log.debugResponse("memories.add", { resourceId: result.resource_id })
-    return { resourceId: result.resource_id }
-  }
-
-  async listIntegrations(): Promise<Integration[]> {
-    log.debugRequest("integrations.list", {})
-
-    const response = await this.client.integrations.list()
-
-    const integrations: Integration[] = response.integrations.map((int) => ({
-      id: int.id,
-      name: int.name,
-      provider: int.provider as HyperspellSource,
-      icon: int.icon,
-    }))
-
-    log.debugResponse("integrations.list", { count: integrations.length })
-    return integrations
-  }
-
-  async getConnectUrl(integrationId: string): Promise<{ url: string; expiresAt: string }> {
-    log.debugRequest("integrations.connect", { integrationId })
-
-    const response = await this.client.integrations.connect(integrationId)
-
-    log.debugResponse("integrations.connect", { url: response.url })
-    return {
-      url: response.url,
-      expiresAt: response.expires_at,
-    }
-  }
-
-  async *listMemories(
-    options?: { source?: HyperspellSource; collection?: string; pageSize?: number },
-  ): AsyncGenerator<{
-    resourceId: string
-    source: HyperspellSource
-    title: string | null
-    metadata: Record<string, unknown>
-  }> {
-    log.debugRequest("memories.list", { source: options?.source, collection: options?.collection })
-
-    const params: Record<string, unknown> = {
-      size: options?.pageSize ?? 100,
-    }
-    if (options?.source) params.source = options.source
-    if (options?.collection) params.collection = options.collection
-
-    for await (const memory of this.client.memories.list(params as any)) {
-      yield {
-        resourceId: memory.resource_id,
-        source: memory.source as HyperspellSource,
-        title: memory.title ?? null,
-        metadata: (memory.metadata ?? {}) as Record<string, unknown>,
-      }
-    }
-  }
-
-  async getMemory(
-    resourceId: string,
-    source: HyperspellSource,
-  ): Promise<Record<string, unknown>> {
-    log.debugRequest("memories.get", { resourceId, source })
-
-    const response = await this.client.memories.get(resourceId, { source })
-    const raw = response as unknown as Record<string, unknown>
-
-    log.debugResponse("memories.get", { resourceId, hasData: "data" in raw })
-    return raw
-  }
-
-  async listConnections(): Promise<Connection[]> {
-    log.debugRequest("connections.list", {})
-
-    const response = await this.client.connections.list()
-
-    const connections: Connection[] = response.connections.map((conn) => ({
-      id: conn.id,
-      integrationId: conn.integration_id,
-      label: conn.label,
-      provider: conn.provider as HyperspellSource,
-    }))
-
-    log.debugResponse("connections.list", { count: connections.length })
-    return connections
-  }
+	private client: Hyperspell;
+	private config: HyperspellConfig;
+
+	constructor(config: HyperspellConfig) {
+		this.config = config;
+		this.client = new Hyperspell({
+			apiKey: config.apiKey,
+			userID: config.userId,
+		});
+		log.info(
+			`client initialized${config.userId ? ` for user ${config.userId}` : ""}`,
+		);
+	}
+
+	async search(
+		query: string,
+		options?: { limit?: number; sources?: HyperspellSource[]; after?: string; before?: string },
+	): Promise<SearchResult[]> {
+		const limit = options?.limit ?? this.config.maxResults;
+		const sources =
+			options?.sources ??
+			(this.config.sources.length > 0 ? this.config.sources : undefined);
+
+		log.debugRequest("memories.search", { query, limit, sources, after: options?.after, before: options?.before });
+
+		const response = await this.client.memories.search({
+			query,
+			sources,
+			options: {
+				max_results: limit,
+				...(options?.after ? { after: options.after } : {}),
+				...(options?.before ? { before: options.before } : {}),
+			},
+		});
+
+		const results: SearchResult[] = response.documents.map((doc) => {
+			const raw = doc as typeof doc & { highlights?: Array<{ id: string; score: number; text: string }> };
+			return {
+				resourceId: doc.resource_id,
+				title: doc.title ?? null,
+				source: doc.source as HyperspellSource,
+				score: doc.score ?? null,
+				url: (doc.metadata?.url as string | null) ?? null,
+				createdAt: (doc.metadata?.created_at as string | null) ?? null,
+				highlights: (raw.highlights ?? []).map((h) => ({ id: h.id, score: h.score, text: h.text })),
+			};
+		});
+
+		log.debugResponse("memories.search", { count: results.length });
+		return results;
+	}
+
+	async searchRaw(
+		query: string,
+		options?: { limit?: number; sources?: HyperspellSource[]; after?: string; before?: string },
+	): Promise<Record<string, unknown>> {
+		const limit = options?.limit ?? this.config.maxResults;
+		const sources =
+			options?.sources ??
+			(this.config.sources.length > 0 ? this.config.sources : undefined);
+
+		log.debugRequest("memories.search (raw)", { query, limit, sources, after: options?.after, before: options?.before });
+
+		const response = await this.client.memories.search({
+			query,
+			sources,
+			options: {
+				max_results: limit,
+				...(options?.after ? { after: options.after } : {}),
+				...(options?.before ? { before: options.before } : {}),
+			},
+		});
+
+		log.debugResponse("memories.search (raw)", {
+			count: response.documents.length,
+		});
+
+		return response as unknown as Record<string, unknown>;
+	}
+
+	async searchWithAnswer(
+		query: string,
+		options?: { limit?: number; sources?: HyperspellSource[] },
+	): Promise<SearchWithAnswerResult> {
+		const limit = options?.limit ?? this.config.maxResults;
+		const sources =
+			options?.sources ??
+			(this.config.sources.length > 0 ? this.config.sources : undefined);
+
+		log.debugRequest("memories.search (with answer)", {
+			query,
+			limit,
+			sources,
+		});
+
+		const response = await this.client.memories.search({
+			query,
+			sources,
+			answer: true,
+			options: {
+				max_results: limit,
+			},
+		});
+
+		const documents: SearchResult[] = response.documents.map((doc) => ({
+			resourceId: doc.resource_id,
+			title: doc.title ?? null,
+			source: doc.source as HyperspellSource,
+			score: doc.score ?? null,
+			url: (doc.metadata?.url as string | null) ?? null,
+			createdAt: (doc.metadata?.created_at as string | null) ?? null,
+		}));
+
+		log.debugResponse("memories.search (with answer)", {
+			count: documents.length,
+			hasAnswer: !!response.answer,
+		});
+
+		return {
+			answer: response.answer ?? null,
+			documents,
+		};
+	}
+
+	async addMemory(
+		text: string,
+		options?: {
+			title?: string;
+			resourceId?: string;
+			collection?: string;
+			date?: string;
+			metadata?: Record<string, string | number | boolean>;
+		},
+	): Promise<{ resourceId: string }> {
+		log.debugRequest("memories.add", {
+			textLength: text.length,
+			title: options?.title,
+			resourceId: options?.resourceId,
+			collection: options?.collection,
+			date: options?.date,
+		});
+
+		const result = await this.client.memories.add({
+			text,
+			title: options?.title,
+			resource_id: options?.resourceId,
+			collection: options?.collection,
+			date: options?.date,
+			metadata: {
+				...options?.metadata,
+				openclaw_source: "command",
+			},
+		});
+
+		log.debugResponse("memories.add", { resourceId: result.resource_id });
+		return { resourceId: result.resource_id };
+	}
+
+	async listIntegrations(): Promise<Integration[]> {
+		log.debugRequest("integrations.list", {});
+
+		const response = await this.client.integrations.list();
+
+		const integrations: Integration[] = response.integrations.map((int) => ({
+			id: int.id,
+			name: int.name,
+			provider: int.provider as HyperspellSource,
+			icon: int.icon,
+		}));
+
+		log.debugResponse("integrations.list", { count: integrations.length });
+		return integrations;
+	}
+
+	async getConnectUrl(
+		integrationId: string,
+	): Promise<{ url: string; expiresAt: string }> {
+		log.debugRequest("integrations.connect", { integrationId });
+
+		const response = await this.client.integrations.connect(integrationId);
+
+		log.debugResponse("integrations.connect", { url: response.url });
+		return {
+			url: response.url,
+			expiresAt: response.expires_at,
+		};
+	}
+
+	async *listMemories(options?: {
+		source?: HyperspellSource;
+		collection?: string;
+		pageSize?: number;
+	}): AsyncGenerator<{
+		resourceId: string;
+		source: HyperspellSource;
+		title: string | null;
+		metadata: Record<string, unknown>;
+	}> {
+		log.debugRequest("memories.list", {
+			source: options?.source,
+			collection: options?.collection,
+		});
+
+		const params: Record<string, unknown> = {
+			size: options?.pageSize ?? 100,
+		};
+		if (options?.source) params.source = options.source;
+		if (options?.collection) params.collection = options.collection;
+
+		for await (const memory of this.client.memories.list(params as any)) {
+			yield {
+				resourceId: memory.resource_id,
+				source: memory.source as HyperspellSource,
+				title: memory.title ?? null,
+				metadata: (memory.metadata ?? {}) as Record<string, unknown>,
+			};
+		}
+	}
+
+	async getMemory(
+		resourceId: string,
+		source: HyperspellSource,
+	): Promise<Record<string, unknown>> {
+		log.debugRequest("memories.get", { resourceId, source });
+
+		const response = await this.client.memories.get(resourceId, { source });
+		const raw = response as unknown as Record<string, unknown>;
+
+		log.debugResponse("memories.get", { resourceId, hasData: "data" in raw });
+		return raw;
+	}
+
+	async sendTrace(
+		history: string,
+		options?: {
+			sessionId?: string;
+			title?: string;
+			extract?: Array<"procedure" | "memory" | "mood">;
+			metadata?: Record<string, string | number | boolean>;
+		},
+	): Promise<{ resourceId: string; status: string }> {
+		log.debugRequest("sessions.add", {
+			historyLength: history.length,
+			sessionId: options?.sessionId,
+			extract: options?.extract,
+		});
+
+		const result = await this.client.sessions.add({
+			history,
+			session_id: options?.sessionId,
+			title: options?.title,
+			format: "openclaw",
+			extract: options?.extract ?? ["procedure"],
+			metadata: {
+				...options?.metadata,
+				openclaw_source: "agent_end",
+			},
+		});
+
+		log.debugResponse("sessions.add", {
+			resourceId: result.resource_id,
+			status: result.status,
+		});
+		return { resourceId: result.resource_id, status: result.status };
+	}
+
+	async listConnections(): Promise<Connection[]> {
+		log.debugRequest("connections.list", {});
+
+		const response = await this.client.connections.list();
+
+		const connections: Connection[] = response.connections.map((conn) => ({
+			id: conn.id,
+			integrationId: conn.integration_id,
+			label: conn.label,
+			provider: conn.provider as HyperspellSource,
+		}));
+
+		log.debugResponse("connections.list", { count: connections.length });
+		return connections;
+	}
 }