@t2000/engine 0.5.0 → 0.5.2

package/dist/index.d.ts

@@ -143,6 +143,7 @@ interface EngineConfig {
     model?: string;
     maxTurns?: number;
     maxTokens?: number;
+    temperature?: number;
     costTracker?: {
         budgetLimitUsd?: number;
         inputCostPerToken?: number;
@@ -158,6 +159,7 @@ interface ChatParams {
     tools: ToolDefinition[];
     model?: string;
     maxTokens?: number;
+    temperature?: number;
     signal?: AbortSignal;
 }
 interface ToolDefinition {
@@ -232,6 +234,7 @@ declare class QueryEngine {
     private readonly model;
     private readonly maxTurns;
     private readonly maxTokens;
+    private readonly temperature;
     private readonly agent;
     private readonly mcpManager;
     private readonly walletAddress;

package/dist/index.js

@@ -1,4 +1,5 @@
 import { z } from 'zod';
+import { getSwapQuote } from '@t2000/sdk';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
@@ -1260,8 +1261,9 @@ var swapQuoteTool = buildTool({
   },
   isReadOnly: true,
   async call(input, context) {
-    const agent = requireAgent(context);
-    const result = await agent.swapQuote({
+    const walletAddress = context.agent ? context.agent.address() : getWalletAddress(context);
+    const result = await getSwapQuote({
+      walletAddress,
       from: input.from,
       to: input.to,
       amount: input.amount,
@@ -1757,6 +1759,7 @@ var QueryEngine = class {
   model;
   maxTurns;
   maxTokens;
+  temperature;
   agent;
   mcpManager;
   walletAddress;
@@ -1776,6 +1779,7 @@ var QueryEngine = class {
     this.model = config.model;
     this.maxTurns = config.maxTurns ?? DEFAULT_MAX_TURNS;
     this.maxTokens = config.maxTokens ?? DEFAULT_MAX_TOKENS;
+    this.temperature = config.temperature;
     this.systemPrompt = config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
     this.costTracker = new CostTracker(config.costTracker);
     this.tools = config.tools ?? (config.agent ? getDefaultTools() : []);
@@ -1919,6 +1923,7 @@ ${summary.join("\n")}`);
           tools: toolDefs,
           model: this.model,
           maxTokens: this.maxTokens,
+          temperature: this.temperature,
           signal
         });
         for await (const event of stream) {
@@ -2681,7 +2686,8 @@ var AnthropicProvider = class {
       max_tokens: params.maxTokens ?? this.defaultMaxTokens,
       system: params.systemPrompt,
       messages,
-      tools: tools.length > 0 ? tools : void 0
+      tools: tools.length > 0 ? tools : void 0,
+      ...params.temperature !== void 0 && { temperature: params.temperature }
     };
     const stream = params.signal ? this.client.messages.stream(streamParams, { signal: params.signal }) : this.client.messages.stream(streamParams);
     const toolInputBuffers = /* @__PURE__ */ new Map();