@axiom-lattice/client-sdk 1.0.23 → 1.0.25

package/README.md

@@ -182,6 +182,39 @@ Steps for future migration:
 
 The main client class for interacting with the Axiom Lattice Agent Service API.
 
+### WeChatClient
+
+A specialized client implementation for WeChat Mini Programs that uses the native `wx` API for network requests.
+
+```typescript
+import { WeChatClient } from "@axiom-lattice/client-sdk";
+
+// Create a WeChat client
+const client = new WeChatClient({
+  baseURL: "https://api.example.com",
+  apiKey: "your-api-key",
+  assistantId: "your-assistant-id",
+  transport: "sse",
+});
+
+// Use the same API methods as the standard Client
+const threadId = await client.createThread({
+  metadata: { user: "user123" },
+});
+
+// Send a message
+const response = await client.chat.send({
+  threadId,
+  messages: [
+    {
+      role: "user",
+      content: "Hello, how can you help me?",
+      id: "msg-1",
+    },
+  ],
+});
+```
+
 ### ChunkMessageMerger
 
 A utility for processing streaming message chunks and merging them into complete messages.

package/dist/index.d.ts

@@ -9,10 +9,6 @@ import { Message, MessageChunk, AssistantMessage } from '@axiom-lattice/protocol
  * Note: Currently only "ws" is used for tool registration
  */
 type Transport = "sse" | "ws";
-/**
- * Runtime environment for the client
- */
-type RuntimeEnvironment = "web" | "wechat-miniprogram";
 /**
  * Configuration options for the Client
  */
@@ -33,11 +29,6 @@ interface ClientConfig {
      * Transport method (Server-Sent Events or WebSocket)
      */
     transport: Transport;
-    /**
-     * Runtime environment (web or wechat-miniprogram)
-     * Defaults to "web" if not specified
-     */
-    environment?: RuntimeEnvironment;
     /**
      * Request timeout in milliseconds (optional, defaults to 30000)
      */
@@ -271,22 +262,26 @@ declare class AuthenticationError extends Error {
  * Main client class for interacting with the Axiom Lattice Agent Service API
  */
 declare class Client {
-    private client;
     private config;
     private assistantId;
     private tenantId;
     private registeredTools;
-    private environment;
+    private headers;
     /**
      * Creates a new Client instance
      * @param config - Configuration options for the client
      */
     constructor(config: ClientConfig);
     /**
-     * Sets up axios interceptors for error handling
+     * Helper method to handle fetch responses and errors
+     * @private
+     */
+    private handleResponse;
+    /**
+     * Helper method to make fetch requests
      * @private
      */
-    private setupInterceptors;
+    private fetchWithTimeout;
     /**
      * Set tenant ID for multi-tenant environments
      * @param tenantId - Tenant identifier
@@ -336,7 +331,7 @@ declare class Client {
     /**
      * Run agent with options
      * @param options - Options for running the agent
-     * @returns A promise that resolves to the run result or a stream
+     * @returns A promise that resolves to the run result
      */
     run(options: RunOptions): Promise<any>;
     /**
@@ -349,15 +344,121 @@ declare class Client {
      */
     private streamRun;
     /**
-     * Stream run results using axios for web environment
-     * @private
+     * Chat namespace for sending messages and streaming responses
+     */
+    chat: {
+        /**
+         * Sends a message to a thread and receives a response
+         * @param options - Options for sending a message
+         * @returns A promise that resolves to the chat response
+         */
+        send: (options: ChatSendOptions) => Promise<ChatResponse>;
+        /**
+         * Sends a message to a thread and streams the response
+         * @param options - Options for streaming a message
+         * @param onEvent - Callback function that receives stream events
+         * @param onComplete - Optional callback function called when streaming completes
+         * @param onError - Optional callback function called when an error occurs
+         * @returns A function that can be called to stop the stream
+         */
+        stream: (options: ChatStreamOptions, onEvent: (event: MessageChunk) => void, onComplete?: (state?: AgentState) => void, onError?: (error: Error) => void) => (() => void);
+    };
+    /**
+     * Tools namespace for registering and unregistering client-side tools
+     */
+    tools: {
+        /**
+         * Registers a client-side tool
+         * @param options - Options for registering a tool
+         */
+        register: (options: RegisterToolOptions) => void;
+        /**
+         * Unregisters a client-side tool
+         * @param name - Tool name
+         */
+        unregister: (name: string) => void;
+    };
+}
+
+/**
+ * WeChat Mini Program client for interacting with the Axiom Lattice Agent Service API
+ */
+declare class WeChatClient {
+    private config;
+    private assistantId;
+    private tenantId;
+    private registeredTools;
+    /**
+     * Creates a new WeChatClient instance
+     * @param config - Configuration options for the client
+     */
+    constructor(config: ClientConfig);
+    /**
+     * Set tenant ID for multi-tenant environments
+     * @param tenantId - Tenant identifier
      */
-    private streamRunWeb;
+    setTenantId(tenantId: string): void;
     /**
-     * Stream run results using WeChat Mini Program request API
+     * Helper method to make WeChat HTTP requests
      * @private
      */
-    private streamRunWechat;
+    private request;
+    /**
+     * Creates a new thread
+     * @param options - Options for creating a thread
+     * @returns A promise that resolves to the thread ID
+     */
+    createThread(options: CreateThreadOptions): Promise<string>;
+    /**
+     * Retrieves thread information
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves to the thread information
+     */
+    getThread(threadId: string): Promise<Thread>;
+    /**
+     * Lists all threads
+     * @param options - Options for listing threads
+     * @returns A promise that resolves to an array of threads
+     */
+    listThreads(options?: ListThreadsOptions): Promise<Thread[]>;
+    /**
+     * Deletes a thread
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves when the thread is deleted
+     */
+    deleteThread(threadId: string): Promise<void>;
+    /**
+     * Retrieves messages from a thread
+     * @param options - Options for retrieving messages
+     * @returns A promise that resolves to an array of messages
+     */
+    getMessages(options: GetMessagesOptions): Promise<Message[]>;
+    /**
+     * Retrieves agent state
+     * @param threadId - Thread identifier
+     * @returns A promise that resolves to the agent state
+     */
+    getAgentState(threadId: string): Promise<AgentState>;
+    /**
+     * Gets agent graph visualization
+     * @returns A promise that resolves to the graph visualization data
+     */
+    getAgentGraph(): Promise<string>;
+    /**
+     * Run agent with options
+     * @param options - Options for running the agent
+     * @returns A promise that resolves to the run result
+     */
+    run(options: RunOptions): Promise<any>;
+    /**
+     * Stream run results using WeChat's downloadFile API
+     * @param options - Options for streaming run results
+     * @param onEvent - Callback function that receives stream events
+     * @param onComplete - Optional callback function called when streaming completes
+     * @param onError - Optional callback function called when an error occurs
+     * @returns A function that can be called to stop the stream
+     */
+    private streamRun;
     /**
      * Chat namespace for sending messages and streaming responses
      */
@@ -413,4 +514,4 @@ declare function createSimpleMessageMerger(): {
     reset: () => void;
 };
 
-export { AgentState, ApiError, AuthenticationError, ChatResponse, ChatSendOptions, ChatStreamOptions, Client, ClientConfig, CreateThreadOptions, GetMessagesOptions, ListThreadsOptions, NetworkError, RegisterToolOptions, RetryConfig, RunOptions, RuntimeEnvironment, StreamCallbacks, Thread, Transport, createSimpleMessageMerger };
+export { AgentState, ApiError, AuthenticationError, ChatResponse, ChatSendOptions, ChatStreamOptions, Client, ClientConfig, CreateThreadOptions, GetMessagesOptions, ListThreadsOptions, NetworkError, RegisterToolOptions, RetryConfig, RunOptions, StreamCallbacks, Thread, Transport, WeChatClient, createSimpleMessageMerger };