upgrade taskloop

README.md

@@ -159,3 +159,13 @@ npm run build:plugin
 ```
 
 Then just press F5, いただきます (Let's begin)
+
+---
+
+## CI Pipeline
+
+✅ npm run build
+✅ npm run build:task-loop
+✅ openmcp-client UT
+✅ openmcp-sdk UT
+✅ vscode extension UT

renderer/src/components/main-panel/chat/core/task-loop.ts

@@ -64,7 +64,7 @@ export class TaskLoop {
 
     constructor(
         private readonly taskOptions: TaskLoopOptions = {
-            maxEpochs: 20,
+            maxEpochs: 50,
             maxJsonParseRetry: 3,
             adapter: undefined,
             verbose: 0
@@ -513,7 +513,7 @@ export class TaskLoop {
 
         let jsonParseErrorRetryCount = 0;
         const {
-            maxEpochs = 20,
+            maxEpochs = 50,
             verbose = 0
         } = this.taskOptions || {};
 

resources/openmcp-sdk-release/package.json

@@ -37,6 +37,8 @@
 		"open": "^10.1.2",
         "ws": "^8.18.1",
 		"cli-table3": "^0.6.5",
-        "https-proxy-agent": "^7.0.6"
+        "https-proxy-agent": "^7.0.6",
+        "pino": "^9.6.0",
+		"pino-pretty": "^13.0.0"
     }
 }

service/src/hook/adapter.ts

@@ -189,7 +189,13 @@ export interface OmAgentConfiguration {
     }
 }
 
-import { MessageState, type ChatMessage, type ChatSetting, type TaskLoop, type TextMessage } from '../../task-loop.js';
+export interface DefaultLLM {
+    baseURL: string;
+    apiToken?: string;
+    model: string;
+}
+
+import { MessageState, TaskLoopOptions, type ChatMessage, type ChatSetting, type TaskLoop, type TextMessage } from '../../task-loop.js';
 
 export function UserMessage(content: string): TextMessage {
     return {
@@ -218,8 +224,9 @@ export function AssistantMessage(content: string): TextMessage {
 }
 
 export class OmAgent {
-    public _adapter: TaskLoopAdapter;
-    public _loop?: TaskLoop;
+    private _adapter: TaskLoopAdapter;
+    private _loop?: TaskLoop;
+    private _defaultLLM?: DefaultLLM;
 
     constructor() {
         this._adapter = new TaskLoopAdapter();
@@ -257,6 +264,10 @@ export class OmAgent {
     public loadMcpConfig(configPath: string) {
         const config = JSON.parse(fs.readFileSync(configPath, 'utf-8')) as OmAgentConfiguration;
         const { mcpServers, defaultLLM } = config;
+
+        // set default llm
+        this.setDefaultLLM(defaultLLM);
+
         for (const key in mcpServers) {
             const mcpConfig = mcpServers[key];
             if ('command' in mcpConfig) {
@@ -284,28 +295,48 @@ export class OmAgent {
         }
     }
 
-    public async getLoop() {
+    private async getLoop(loopOption?: TaskLoopOptions) {
         if (this._loop) {
             return this._loop;
         }
 
+        const {
+            verbose = 1,
+            maxEpochs = 50,
+            maxJsonParseRetry = 3,
+        } = loopOption || {}
+
         const adapter = this._adapter;
         const { TaskLoop } = await import('../../task-loop.js');
-        this._loop = new TaskLoop({ adapter, verbose: 1 });
+        this._loop = new TaskLoop({ adapter, verbose, maxEpochs, maxJsonParseRetry });
         return this._loop;
     }
 
+    public setDefaultLLM(option: DefaultLLM) {
+        this._defaultLLM = option;
+    }
+
     public async start(
         messages: ChatMessage[] | string,
-        settings?: ChatSetting
+        settings?: ChatSetting & Partial<TaskLoopOptions>
     ) {
         if (messages.length === 0) {
             throw new Error('messages is empty');
         }
 
-        const loop = await this.getLoop();
+        // detach taskloop option from settings and set default value
+        const {
+            maxEpochs = 50,
+            maxJsonParseRetry = 3,
+            verbose = 1
+        } = settings || {};
+
+        const loop = await this.getLoop({ maxEpochs, maxJsonParseRetry, verbose });
         const storage = await loop.createStorage(settings);
         
+        // set input message
+        // user can invoke [UserMessage("CONTENT")] to make messages quickly
+        // or use string directly
         let userMessage: string;
         if (typeof messages === 'string') {
             userMessage = messages;
@@ -318,6 +349,20 @@ export class OmAgent {
             }
         }
 
+        // select correct llm config
+        // user can set llm config via omagent.setDefaultLLM()
+        // or write "defaultLLM" in mcpconfig.json to specify
+        if (this._defaultLLM) {
+            loop.setLlmConfig({
+                baseUrl: this._defaultLLM.baseURL,
+                userToken: this._defaultLLM.apiToken,
+                userModel: this._defaultLLM.model,
+            });
+        } else {
+            // throw error to user and give the suggestion
+            throw new Error('default LLM is not set, please set it via omagent.setDefaultLLM() or write "defaultLLM" in mcpconfig.json');
+        }
+
         return await loop.start(storage, userMessage);
     }
 }

service/src/index.ts

@@ -1,4 +1,4 @@
 export { routeMessage } from './common/router.js';
-export { VSCodeWebViewLike, TaskLoopAdapter, OmAgent, OmAgentConfiguration } from './hook/adapter.js';
+export { VSCodeWebViewLike, TaskLoopAdapter, OmAgent, OmAgentConfiguration, UserMessage, AssistantMessage } from './hook/adapter.js';
 export { setVscodeWorkspace, setRunningCWD } from './hook/setting.js';
 export { clientMap } from './mcp/connect.service.js';