diff --git a/src/sdk/index.ts b/src/sdk/index.ts index afe9a5f..86580ae 100644 --- a/src/sdk/index.ts +++ b/src/sdk/index.ts @@ -5,6 +5,7 @@ export * from "../lib/schemas/query-search.js"; export * from "../lib/schemas/query-atlas.js"; export * from "../lib/schemas/query-day.js"; export * from "../lib/schemas/query-node-type.js"; +export * from "../lib/schemas/query-graph.js"; export * from "../lib/schemas/messages.js"; export * from "../lib/schemas/summarize.js"; export * from "../lib/schemas/cleanup.js"; diff --git a/src/sdk/memory-client.ts b/src/sdk/memory-client.ts index 2ec9362..58271e3 100644 --- a/src/sdk/memory-client.ts +++ b/src/sdk/memory-client.ts @@ -28,6 +28,11 @@ import { QueryDayResponse, queryDayResponseSchema, } from "../lib/schemas/query-day.js"; +import { + QueryGraphRequest, + QueryGraphResponse, + queryGraphResponseSchema, +} from "../lib/schemas/query-graph.js"; import { QueryNodeTypeRequest, QueryNodeTypeResponse, @@ -156,6 +161,15 @@ export class MemoryClient { ); } + async queryGraph(payload: QueryGraphRequest): Promise { + return this._fetch( + "POST", + "/query/graph", + queryGraphResponseSchema, + payload, + ); + } + async summarize(payload: SummarizeRequest): Promise { return this._fetch("POST", "/summarize", summarizeResponseSchema, payload); }