Sdk rewrite pt1

dist/chatClient.d.ts

@@ -1,4 +1,5 @@
 import { ChainInfo } from "@keplr-wallet/types";
+import { NesaClient } from "./client";
 interface ConfigOptions {
     modelName: string;
     lockAmount?: string;
@@ -7,8 +8,10 @@ interface ConfigOptions {
     singlePaymentAmount?: string;
     lowBalance?: string;
     privateKey?: string;
+    mnemonic?: string;
+    chatId?: string;
 }
-interface questionTypes {
+interface QuestionParams {
     messages: any;
     model: string;
     stream?: boolean;
@@ -25,13 +28,14 @@ declare class ChatClient {
     singlePaymentAmount: string;
     lowBalance: string;
     lockAmountDenom: string;
+    chatId: string;
     private walletName;
     private chatQueue;
     private chatSeq;
     private totalUsedPayment;
     private totalSignedPayment;
-    private isChatinging;
-    private isRegisterSessioning;
+    private isChatting;
+    private isRegisteringSession;
     private agentUrl;
     private assistantRoleName;
     private lastNesaClientPromise;
@@ -40,26 +44,26 @@ declare class ChatClient {
     private lastInitOfflineSignerPromise;
     private chatProgressReadable;
     private nesaClient;
-    private offLinesigner;
+    private offlineSigner;
     private signaturePayment;
     private isBrowser;
     private privateKey;
+    private mnemonic;
     private isEverRequestSession;
     private tokenPrice;
     constructor(options: ConfigOptions);
     initWallet(): any;
-    getNesaClient(): any;
-    getChainParams(nesaClient: any): any;
+    getNesaClient(): Promise<NesaClient>;
+    getChainParams(nesaClient: NesaClient): any;
     version(): string;
-    checkChainInfo(): string | false;
     getSignaturePayment(): string;
     checkSinglePaymentAmount(): string;
-    requestChatQueue(readableStream: any, question: questionTypes): void;
+    requestChatQueue(readableStream: any, question: QuestionParams): void;
     requestCloseHeartbeat(): void;
     requestAgentInfo(result: any, readableStream: any): any;
     checkSignBroadcastResult(readableStream?: any): Promise<unknown>;
-    requestChatStatus(): Promise<unknown>;
-    requestSession(): Promise<unknown>;
-    requestChat(question: questionTypes): Promise<unknown>;
+    requestChatStatus(): any;
+    requestSession(): Promise<any>;
+    requestChat(question: QuestionParams): Promise<any>;
 }
 export default ChatClient;

dist/client.d.ts

@@ -2,12 +2,11 @@
 import { OfflineSigner } from "@cosmjs/proto-signing";
 import { SigningStargateClient, SigningStargateClientOptions, GasPrice, Event, QueryClient } from "@cosmjs/stargate";
 import { CometClient } from "@cosmjs/tendermint-rpc";
-import { Logger } from './logger';
-import { VRF } from './codec/agent/v1/tx';
+import { Logger } from "./logger";
+import { VRF } from "./codec/agent/v1/tx";
 import { Payment, Params, SessionStatus } from "./codec/agent/v1/agent";
 import { Coin } from "./codec/cosmos/base/v1beta1/coin";
-import { AgentExtension } from './queries';
-import { QueryParamsResponse, QueryInferenceAgentResponse, QuerySessionResponse, QueryVRFSeedResponse, QuerySessionByAgentResponse } from "./codec/agent/v1/query";
+import { AgentExtension } from "./queries";
 import { StdFee } from "@cosmjs/amino";
 export type NesaClientOptions = SigningStargateClientOptions & {
     logger?: Logger;
@@ -28,9 +27,21 @@ export type RegisterSessionResult = MsgResult & {
 export declare class NesaClient {
     readonly gasPrice: GasPrice;
     readonly sign: SigningStargateClient;
+    readonly signByModel: {
+        [modelName: string]: SigningStargateClient;
+    };
     readonly query: QueryClient & AgentExtension;
+    readonly queryByModel: {
+        [modelName: string]: QueryClient & AgentExtension;
+    };
     readonly tm: CometClient;
+    readonly tmByModel: {
+        [modelName: string]: CometClient;
+    };
     readonly senderAddress: string;
+    readonly senderAddressByModel: {
+        [modelName: string]: string;
+    };
     readonly logger: Logger;
     readonly chainId: string;
     readonly estimatedBlockTime: number;
@@ -42,12 +53,15 @@ export declare class NesaClient {
     updateParams(authority: string, params: Params): Promise<MsgResult>;
     registerInferenceAgent(url: string, version: Long): Promise<MsgResult>;
     broadcastRegisterSession(): any;
-    signRegisterSession(sessionId: string, modelName: string, fee: StdFee, lockBalance?: Coin, vrf?: VRF): Promise<any>;
+    signRegisterSession(sessionId: string, modelName: string, fee: StdFee, lockBalance?: Coin, vrf?: VRF): Promise<{
+        sessionId: string;
+        transactionHash: string;
+    }>;
     registerSession(sessionId: string, modelName: string, lockBalance?: Coin, vrf?: VRF): Promise<RegisterSessionResult>;
     submitPayment(sessionId: string, signature: Uint8Array, payment?: Payment): Promise<MsgResult>;
-    getParams(): Promise<QueryParamsResponse>;
-    getInferenceAgent(account: string, modelName: string, limit: Long, key: Uint8Array): Promise<QueryInferenceAgentResponse>;
-    getSession(sessionId: string): Promise<QuerySessionResponse>;
-    getSessionByAgent(account: string, status: SessionStatus, limit: Long, orderDesc: boolean, key: Uint8Array, expireTime?: Date): Promise<QuerySessionByAgentResponse>;
-    getVRFSeed(account: string): Promise<QueryVRFSeedResponse>;
+    getParams(): Promise<import("./codec/agent/v1/query").QueryParamsResponse>;
+    getInferenceAgent(account: string, modelName: string, limit: Long, key: Uint8Array): Promise<import("./codec/agent/v1/query").QueryInferenceAgentResponse>;
+    getSession(sessionId: string): Promise<import("./codec/agent/v1/query").QuerySessionResponse>;
+    getSessionByAgent(account: string, status: SessionStatus, limit: Long, orderDesc: boolean, key: Uint8Array, expireTime?: Date): Promise<import("./codec/agent/v1/query").QuerySessionByAgentResponse>;
+    getVRFSeed(account: string): Promise<import("./codec/agent/v1/query").QueryVRFSeedResponse>;
 }

dist/encryptUtils.d.ts

@@ -1,14 +1,29 @@
+import { NesaClient } from "./client";
+import { CosmjsOfflineSigner } from "@leapwallet/cosmos-snap-provider";
+import { VRF } from "./codec/agent/v1/tx";
+type Key = {
+    x: string;
+    y: string;
+};
 declare class EncryptUtils {
-    static privateKey: any;
-    static publicKey: any;
-    static privateKeyBuf: any;
-    static generateKey(): {
+    static privateKeyMap: {
+        [recordId: string]: Key;
+    };
+    static publicKeyMap: {
+        [recordId: string]: Key;
+    };
+    static privateKeyBufMap: {
+        [recordId: string]: Key;
+    };
+    static generateKey(recordId: string): {
         privateKey: any;
         publicKey: any;
     };
-    static sortObjectKeys(obj: Record<string, any>): Record<string, any>;
-    static signMessage(message: string, chatSeq: number, isQuestion?: boolean): string;
-    static requestVrf(client: any, offlineSigner: any): Promise<any>;
-    static signHeartbeat(message: string): string;
+    static signMessage(recordId: string, message: string, chatSeq: number, isQuestion?: boolean): string;
+    static requestVrf(recordId: string, client: NesaClient, offlineSigner: CosmjsOfflineSigner): Promise<{
+        vrf: VRF;
+        sessionId: string;
+    }>;
+    static signHeartbeat(recordId: string, message: string): string;
 }
 export default EncryptUtils;

dist/helpers/getAgentUrls.d.ts

@@ -0,0 +1,5 @@
+import type { InferenceAgent } from "../codec/agent/v1/agent";
+export declare const getAgentUrls: (selectAgent: InferenceAgent) => {
+    agentWsUrl: string;
+    agentHeartbeatUrl: string;
+};

dist/helpers/getIsChainInfoValid.d.ts

@@ -0,0 +1,2 @@
+import { ChainInfo } from "@leapwallet/cosmos-snap-provider";
+export declare const getIsChainInfoValid: (chainInfo: ChainInfo) => string | false | undefined;

dist/socket.d.ts

@@ -7,11 +7,13 @@ interface ISocket {
     heartbeat_timer: NodeJS.Timeout | string | number | undefined;
     heartbeat_interval: number;
     init: (handle: {
+        modelName?: string;
+        recordId: string;
         ws_url: string;
         onopen: () => void;
         onclose?: (e: Event) => void;
         onerror?: (e: Event | Error) => void;
-    }) => (WebSocket | null | undefined);
+    }) => WebSocket | null | undefined;
     heartbeat: Function;
     send: (data: any, callback?: Function) => void;
     close: Function;