@vectorx/agent-runtime 0.1.0

package/types/sse-sender.d.ts

@@ -0,0 +1,32 @@
+import type { EventEmitter } from "events";
+import type { RcbContext, ServerSentEvent } from "@vectorx/functions-framework";
+export type Primitive = number | string | boolean | undefined | null;
+export type BufferLike = number | string | Buffer | ArrayBuffer | ArrayBufferView | DataView;
+export declare namespace sse {
+    interface SseEvent<T = Primitive | object | Buffer> {
+        data: T;
+        comment?: string;
+        event?: string;
+        id?: string;
+        retry?: number;
+    }
+    type Event<T> = SseEvent<T> | string | Buffer;
+    interface IServerSentEvent extends EventEmitter {
+        closed: boolean;
+        setEncoder(encoder: TextEncoder): void;
+        send<T>(event: Event<T> | Event<T>[]): boolean;
+        end(msg: SseEvent): void;
+        on(event: "close", listener: () => void): this;
+        on(event: "error", listener: (errorEvent: SseEvent) => void): this;
+    }
+}
+export declare class SSESender<T> {
+    readonly context: RcbContext;
+    private _status;
+    private _sse;
+    get status(): "initial" | "started" | "ended";
+    constructor(context: RcbContext);
+    get sse(): ServerSentEvent;
+    send(data: sse.Event<T>): void;
+    end(): void;
+}

package/types/types.d.ts

@@ -0,0 +1,71 @@
+import type { SSESender } from "./sse-sender";
+export interface SendMessageInput {
+    msg: string;
+    history?: {
+        role: "user" | "assistant";
+        content: string;
+    }[];
+    searchEnable?: boolean;
+}
+export interface GetLoneMemoryInput {
+    userId: string;
+    limit?: number;
+    filter?: {
+        startTime?: number;
+        endTime?: number;
+        type?: string;
+    };
+}
+export interface KnowledgeSearchInput {
+    collectionView: string;
+    search: {
+        content: string;
+        documentSetName?: string[];
+        options?: {
+            chunkExpand?: number[];
+            filter?: string;
+            limit?: number;
+        };
+    };
+}
+export interface UserProfile {
+    tags?: string[];
+    preferences?: Record<string, any>;
+}
+export interface DocumentSet {
+    documentSetId: string;
+    documentSetName: string;
+    author: string;
+    fileTitle: string;
+    fileMetaData: string;
+}
+export interface DocumentData {
+    text: string;
+    startPos: number;
+    endPos: number;
+    pre: string[];
+    next: string[];
+    paragraphTitle: string;
+    allParentParagraphTitles: string[];
+    documentSet: DocumentSet;
+}
+export interface KnowledgeSearchResult {
+    score: number;
+    data: DocumentData;
+}
+export interface KnowledgeSearchResponse {
+    data: {
+        documents: KnowledgeSearchResult[];
+        requestId: string;
+    };
+}
+export interface ErrorResponse {
+    code: string;
+    message: string;
+    requestId: string;
+}
+export type AgentCoreSseSender = SSESender<{
+    content: string;
+    [key: string]: unknown;
+}>;
+export type PickRequired<T, K extends keyof T> = T & Required<Pick<T, K>>;

package/types/utils/agent-helper.d.ts

@@ -0,0 +1,8 @@
+import type { RcbContext } from "@vectorx/functions-framework";
+import queryString from "query-string";
+export declare function parseAgentId(url: string): string;
+export declare function parseAgentTag(url: string): string;
+export declare function parseApiName(url: string): string;
+export declare function parseQueryFromCtx(ctx: RcbContext): queryString.ParsedQuery<string | boolean>;
+export declare function genRecordId(): string;
+export declare function genRandomStr(length: number): string;

package/types/utils/index.d.ts

@@ -0,0 +1 @@
+export * from "./agent-helper";

package/types/utils/integration-response.d.ts

@@ -0,0 +1,20 @@
+export interface IntegrationResponse {
+    statusCode: number;
+    headers: Record<string, string>;
+    body: {
+        code: string;
+        message: string;
+    };
+}
+export interface ErrorResponse {
+    code: string;
+    message: string;
+    requestId: string;
+}
+export interface BaseResponse<T = any> {
+    code: number;
+    msg: string;
+    data?: T;
+}
+export declare function createIntegrationResponse(statusCode: number, code: string, message: string): IntegrationResponse;
+export declare const OK_MESSAGE = "OK";