add transactions and related inquiries

dist/chatClient.d.ts

@@ -1,31 +1,41 @@
 import { ChainInfo } from "@keplr-wallet/types";
 import { NesaClient } from "./client";
+import { TokenPrice, InferenceAgent } from "./codec/agent/v1/agent";
+interface TokenNumber {
+    inputTokens: number;
+    outputTokens: number;
+}
 interface ConfigOptions {
     modelName: string;
     lockAmount?: string;
     chainInfo?: ChainInfo;
     walletName?: string;
-    singlePaymentAmount?: string;
+    priceUnit?: string;
     lowBalance?: string;
     privateKey?: string;
     mnemonic?: string;
     chatId?: string;
+    isByPass?: boolean;
+    agentUrl?: string;
+    authToken?: string;
 }
 interface QuestionParams {
     messages: any;
     model: string;
     stream?: boolean;
+    session_id?: string;
+    model_params?: Map<string, any>;
+    additional_params?: Map<string, any>;
     frequency_penalty?: any;
     presence_penalty?: any;
     temperature?: any;
     top_p?: any;
-    session_id?: string;
 }
 declare class ChatClient {
     modelName: string;
     chainInfo: ChainInfo;
     lockAmount: string;
-    singlePaymentAmount: string;
+    priceUnit: string;
     lowBalance: string;
     lockAmountDenom: string;
     chatId: string;
@@ -37,6 +47,7 @@ declare class ChatClient {
     private isChatting;
     private isRegisteringSession;
     private agentUrl;
+    private agentChatUrl;
     private assistantRoleName;
     private lastNesaClientPromise;
     private lastUserMinimumLockPromise;
@@ -51,19 +62,25 @@ declare class ChatClient {
     private mnemonic;
     private isEverRequestSession;
     private tokenPrice;
+    private minerSessionId;
+    private agentSessionId;
+    private isByPass;
+    private authToken;
     constructor(options: ConfigOptions);
     initWallet(): any;
     getNesaClient(): Promise<NesaClient>;
     getChainParams(nesaClient: NesaClient): any;
     version(): string;
     getSignaturePayment(): string;
-    checkSinglePaymentAmount(): string;
+    checkSinglePaymentAmount(totalSignedPayment: string): string;
+    computePaymentAmount(tokenNumber: TokenNumber, tokenPrice: TokenPrice): string;
     requestChatQueue(readableStream: any, question: QuestionParams): void;
     requestCloseHeartbeat(): void;
     requestAgentInfo(result: any, readableStream: any): any;
     checkSignBroadcastResult(readableStream?: any): Promise<unknown>;
     requestChatStatus(): any;
     requestSession(): Promise<any>;
     requestChat(question: QuestionParams): Promise<any>;
+    connectAgent(selectAgent: InferenceAgent, readableStream?: any): Promise<unknown>;
 }
 export default ChatClient;

dist/client.d.ts

@@ -4,10 +4,12 @@ import { SigningStargateClient, SigningStargateClientOptions, GasPrice, Event, Q
 import { CometClient } from "@cosmjs/tendermint-rpc";
 import { Logger } from "./logger";
 import { VRF } from "./codec/agent/v1/tx";
-import { Payment, Params, SessionStatus } from "./codec/agent/v1/agent";
+import { Payment, Params, SessionStatus, TokenPrice, AgentModelStatus } from "./codec/agent/v1/agent";
 import { Coin } from "./codec/cosmos/base/v1beta1/coin";
-import { AgentExtension } from "./queries";
+import { setupAgentExtension, setupDHTExtension } from "./queries";
 import { StdFee } from "@cosmjs/amino";
+import { PageRequest } from "./codec/cosmos/base/query/v1beta1/pagination";
+import { Availability, InferenceType } from "./codec/dht/v1/orchestrator";
 export type NesaClientOptions = SigningStargateClientOptions & {
     logger?: Logger;
     gasPrice: GasPrice;
@@ -30,9 +32,9 @@ export declare class NesaClient {
     readonly signByModel: {
         [modelName: string]: SigningStargateClient;
     };
-    readonly query: QueryClient & AgentExtension;
+    readonly query: QueryClient & ReturnType<typeof setupAgentExtension> & ReturnType<typeof setupDHTExtension>;
     readonly queryByModel: {
-        [modelName: string]: QueryClient & AgentExtension;
+        [modelName: string]: QueryClient & ReturnType<typeof setupAgentExtension> & ReturnType<typeof setupDHTExtension>;
     };
     readonly tm: CometClient;
     readonly tmByModel: {
@@ -53,15 +55,30 @@ 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<{
+    signRegisterSession(sessionId: string, modelName: string, fee: StdFee, lockBalance: Coin, vrf: VRF, tokenPrice: TokenPrice): Promise<{
         sessionId: string;
+        tokenPrice: TokenPrice;
         transactionHash: string;
     }>;
-    registerSession(sessionId: string, modelName: string, lockBalance?: Coin, vrf?: VRF): Promise<RegisterSessionResult>;
+    registerSession(sessionId: string, modelName: string, lockBalance: Coin, vrf: VRF, tokenPrice: TokenPrice): Promise<RegisterSessionResult>;
     submitPayment(sessionId: string, signature: Uint8Array, payment?: Payment): Promise<MsgResult>;
+    registerModel(creator: string, modelName: string, blockCids: string[], allowList: string[], tokenPrice?: TokenPrice): Promise<MsgResult>;
+    updateModel(modelName: string, allowList: string[], tokenPrice: TokenPrice): Promise<MsgResult>;
     getParams(): Promise<import("./codec/agent/v1/query").QueryParamsResponse>;
     getInferenceAgent(account: string, modelName: string, limit: Long, key: Uint8Array): Promise<import("./codec/agent/v1/query").QueryInferenceAgentResponse>;
+    getAgentByModel(modelName: string, status: AgentModelStatus): Promise<import("./codec/agent/v1/query").QueryAgentByModelResponse>;
     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>;
+    getSessionByAgent(account: string, status: SessionStatus | undefined, expireTime: Date, limit: Long, orderDesc: boolean, key: Uint8Array): Promise<import("./codec/agent/v1/query").QuerySessionByAgentResponse>;
+    getSessionByChallenge(account: string, limit: Long, key: Uint8Array): Promise<import("./codec/agent/v1/query").QuerySessionByChallengeResponse>;
     getVRFSeed(account: string): Promise<import("./codec/agent/v1/query").QueryVRFSeedResponse>;
+    getDhtParams(): Promise<import("./codec/dht/v1/query").QueryParamsResponse>;
+    getModel(modelName: string): Promise<import("./codec/dht/v1/query").QueryGetModelResponse>;
+    getModelBlocks(modelName: string, pagination?: PageRequest): Promise<import("./codec/dht/v1/query").QueryGetModelBlocksResponse>;
+    getNode(nodeId: string): Promise<import("./codec/dht/v1/query").QueryGetNodeResponse>;
+    getMiner(nodeId: string): Promise<import("./codec/dht/v1/query").QueryGetMinerResponse>;
+    getOrchestrator(nodeId: string): Promise<import("./codec/dht/v1/query").QueryGetOrchestratorResponse>;
+    getAllOrchestrator(pagination?: PageRequest): Promise<import("./codec/dht/v1/query").QueryGetAllOrchestratorResponse>;
+    getOrchestratorsByParams(inferenceType: InferenceType, availability: Availability, limit: number, key: Uint8Array): Promise<import("./codec/dht/v1/query").QueryGetOrchestratorsByParamsResponse>;
+    getOrchestratorHeartbeat(nodeId: string): Promise<import("./codec/dht/v1/query").QueryGetOrchestratorHeartbeatResponse>;
+    getMinerHeartbeat(nodeId: string): Promise<import("./codec/dht/v1/query").QueryGetMinerHeartbeatResponse>;
 }

dist/codec/agent/module/v1/module.d.ts

@@ -1,28 +1,18 @@
-import Long from "long";
-import _m0 from "protobufjs/minimal";
+import * as _m0 from "protobufjs/minimal";
+import { JsonSafe } from "../../../json-safe";
 export declare const protobufPackage = "nesaorg.nesachain.agent.module.v1";
 export interface Module {
     authority: string;
 }
 export declare const Module: {
+    typeUrl: string;
     encode(message: Module, writer?: _m0.Writer): _m0.Writer;
     decode(input: _m0.Reader | Uint8Array, length?: number): Module;
     fromJSON(object: any): Module;
-    toJSON(message: Module): unknown;
+    toJSON(message: Module): JsonSafe<Module>;
     fromPartial<I extends {
         authority?: string | undefined;
     } & {
         authority?: string | undefined;
-    } & { [K in Exclude<keyof I, "authority">]: never; }>(object: I): Module;
+    } & Record<Exclude<keyof I, "authority">, never>>(object: I): Module;
 };
-type Builtin = Date | Function | Uint8Array | string | number | boolean | undefined;
-export type DeepPartial<T> = T extends Builtin ? T : T extends Long ? string | number | Long : T extends Array<infer U> ? Array<DeepPartial<U>> : T extends ReadonlyArray<infer U> ? ReadonlyArray<DeepPartial<U>> : T extends {} ? {
-    [K in keyof T]?: DeepPartial<T[K]>;
-} : Partial<T>;
-type KeysOfUnion<T> = T extends T ? keyof T : never;
-export type Exact<P, I extends P> = P extends Builtin ? P : P & {
-    [K in keyof P]: Exact<P[K], I[K]>;
-} & {
-    [K in Exclude<keyof I, KeysOfUnion<P>>]: never;
-};
-export {};