@polka-codes/core 0.1.4 → 0.1.6

package/dist/Agent/AgentBase.d.ts

@@ -1,7 +1,7 @@
-import type Anthropic from '@anthropic-ai/sdk';
-import type { AiServiceBase, MessageParam } from '../AiService';
-import type { Logger } from '../logger';
-import { type ToolHandler, type ToolInfo } from '../tools';
+import type Anthropic from "@anthropic-ai/sdk";
+import type { AiServiceBase, MessageParam } from "../AiService/index.js";
+import type { Logger } from "../logger.js";
+import { type ToolHandler, type ToolInfo } from "../tools/index.js";
 export type TaskEvent = {
     kind: string;
     info: TaskInfo;
@@ -43,7 +43,7 @@ export declare abstract class AgentBase {
     }): Promise<TaskInfo>;
     get model(): {
         id: string;
-        info: import("../AiService").ModelInfo;
+        info: import("../AiService/index.js").ModelInfo;
     };
 }
 export {};

package/dist/Agent/AgentBase.js

@@ -1,6 +1,6 @@
-import { ToolResponseType } from '../tools';
-import { parseAssistantMessage } from './parseAssistantMessage';
-import { responsePrompts } from './prompts';
+import { ToolResponseType } from "../tools/index.js";
+import { parseAssistantMessage } from "./parseAssistantMessage.js";
+import { responsePrompts } from "./prompts.js";
 export class AgentBase {
     ai;
     config;
@@ -28,75 +28,75 @@ export class AgentBase {
         }
         let nextRequest = [
             {
-                type: 'text',
+                type: "text",
                 text,
             },
         ];
         let iterations = 0;
         while (nextRequest) {
             if (iterations >= taskInfo.options.maxIterations) {
-                callback({ kind: 'max_iterations_reached', info: taskInfo });
+                callback({ kind: "max_iterations_reached", info: taskInfo });
                 break;
             }
             const response = await this.#request(taskInfo, nextRequest, callback);
             nextRequest = await this.#handleResponse(taskInfo, response, callback);
             ++iterations;
         }
-        callback({ kind: 'end_task', info: taskInfo });
+        callback({ kind: "end_task", info: taskInfo });
         return taskInfo;
     }
     async #request(info, userContent, callback) {
-        await callback({ kind: 'start_request', info, userContent });
+        await callback({ kind: "start_request", info, userContent });
         info.messages.push({
-            role: 'user',
+            role: "user",
             content: userContent,
         });
-        this.logger.trace(info.messages, 'Sending messages to AI');
+        this.logger.trace(info.messages, "Sending messages to AI");
         // TODO: use a truncated messages if needed to avoid exceeding the token limit
         const stream = this.ai.send(this.config.systemPrompt, info.messages);
-        let currentAssistantMessage = '';
+        let currentAssistantMessage = "";
         for await (const chunk of stream) {
             switch (chunk.type) {
-                case 'usage':
+                case "usage":
                     info.inputTokens = chunk.inputTokens;
                     info.outputTokens = chunk.outputTokens;
                     info.cacheWriteTokens = chunk.cacheWriteTokens ?? 0;
                     info.cacheReadTokens = chunk.cacheReadTokens ?? 0;
                     info.totalCost = chunk.totalCost;
-                    await callback({ kind: 'usage', info });
+                    await callback({ kind: "usage", info });
                     break;
-                case 'text':
+                case "text":
                     currentAssistantMessage += chunk.text;
-                    await callback({ kind: 'text', info, newText: chunk.text });
+                    await callback({ kind: "text", info, newText: chunk.text });
                     break;
             }
         }
         // TODO: error handling
         if (!currentAssistantMessage) {
-            throw new Error('No assistant message received');
+            throw new Error("No assistant message received");
         }
         info.messages.push({
-            role: 'assistant',
+            role: "assistant",
             content: currentAssistantMessage,
         });
         const ret = parseAssistantMessage(currentAssistantMessage, this.config.tools, this.config.toolNamePrefix);
-        await callback({ kind: 'end_request', info });
+        await callback({ kind: "end_request", info });
         return ret;
     }
     async #handleResponse(info, response, callback) {
         const toolReponses = [];
         for (const content of response) {
             switch (content.type) {
-                case 'text':
+                case "text":
                     // no need to handle text content
                     break;
-                case 'tool_use': {
-                    await callback({ kind: 'tool_use', info, tool: content.name });
+                case "tool_use": {
+                    await callback({ kind: "tool_use", info, tool: content.name });
                     const toolResp = await this.#invokeTool(content.name, content.params);
                     switch (toolResp.type) {
                         case ToolResponseType.Reply:
                             // reply to the tool use
-                            await callback({ kind: 'tool_reply', info, tool: content.name });
+                            await callback({ kind: "tool_reply", info, tool: content.name });
                             toolReponses.push({ tool: content.name, response: toolResp.message });
                             break;
                         case ToolResponseType.Exit:
@@ -104,17 +104,17 @@ export class AgentBase {
                             return undefined;
                         case ToolResponseType.Invalid:
                             // tell AI about the invalid arguments
-                            await callback({ kind: 'tool_invalid', info, tool: content.name });
+                            await callback({ kind: "tool_invalid", info, tool: content.name });
                             toolReponses.push({ tool: content.name, response: toolResp.message });
                             break;
                         case ToolResponseType.Error:
                             // tell AI about the error
-                            await callback({ kind: 'tool_error', info, tool: content.name });
+                            await callback({ kind: "tool_error", info, tool: content.name });
                             toolReponses.push({ tool: content.name, response: toolResp.message });
                             break;
                         case ToolResponseType.Interrupted:
                             // the execution is killed
-                            await callback({ kind: 'tool_interrupted', info, tool: content.name });
+                            await callback({ kind: "tool_interrupted", info, tool: content.name });
                             return undefined;
                     }
                     break;
@@ -126,15 +126,15 @@ export class AgentBase {
             // we need to convert the loop to a state machine
             return [
                 {
-                    type: 'text',
+                    type: "text",
                     text: responsePrompts.requireUseTool,
                 },
             ];
         }
-        const finalResp = toolReponses.map(({ tool, response }) => responsePrompts.toolResults(tool, response)).join('\n');
+        const finalResp = toolReponses.map(({ tool, response }) => responsePrompts.toolResults(tool, response)).join("\n");
         return [
             {
-                type: 'text',
+                type: "text",
                 text: finalResp,
             },
         ];
@@ -155,4 +155,4 @@ export class AgentBase {
         return this.ai.model;
     }
 }
-//# sourceMappingURL=AgentBase.js.map
+//# sourceMappingURL=AgentBase.js.map

package/dist/Agent/CoderAgent/index.d.ts

@@ -1,6 +1,6 @@
-import type { AiServiceBase } from '../../AiService';
-import { type ToolHandler, type ToolInfo } from '../../tools';
-import { AgentBase } from '../AgentBase';
+import type { AiServiceBase } from "../../AiService/index.js";
+import { type ToolHandler, type ToolInfo } from "../../tools/index.js";
+import { AgentBase } from "../AgentBase.js";
 export type CoderAgentOptions = {
     ai: AiServiceBase;
     os: string;

package/dist/Agent/CoderAgent/index.js

@@ -1,7 +1,7 @@
-import { createServiceLogger } from '../../logger';
-import { allTools } from '../../tools';
-import { AgentBase } from '../AgentBase';
-import { fullSystemPrompt } from './prompts';
+import { createServiceLogger } from "../../logger.js";
+import { allTools } from "../../tools/index.js";
+import { AgentBase } from "../AgentBase.js";
+import { fullSystemPrompt } from "./prompts.js";
 const defaultTools = [
     allTools.executeCommand,
     allTools.readFile,
@@ -16,7 +16,7 @@ const defaultTools = [
 export class CoderAgent extends AgentBase {
     constructor(options) {
         const tools = options.tools || defaultTools;
-        const toolNamePrefix = 'tool_';
+        const toolNamePrefix = "tool_";
         const systemPrompt = fullSystemPrompt({
             os: options.os,
         }, tools, toolNamePrefix, options.customInstructions ?? [], options.commands ?? {});
@@ -26,7 +26,7 @@ export class CoderAgent extends AgentBase {
             toolNamePrefix,
             toolHandler: options.toolHandler,
             customInstructions: options.customInstructions,
-        }, createServiceLogger('CoderAgent'));
+        }, createServiceLogger("CoderAgent"));
     }
 }
-//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/Agent/CoderAgent/prompts.d.ts

@@ -1,4 +1,4 @@
-import type { ToolInfo } from '../../tools';
+import type { ToolInfo } from "../../tools/index.js";
 export declare const basePrompt = "You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.";
 export declare const editingFilesPrompt: (toolNamePrefix: string) => string;
 export declare const capabilities: (toolNamePrefix: string) => string;

package/dist/Agent/CoderAgent/prompts.js

@@ -1,7 +1,7 @@
 // source: https://github.com/cline/cline/blob/f6c19c29a64ca84e9360df7ab2c07d128dcebe64/src/core/prompts/system.ts#L1
-import { toolUsePrompt } from '../prompts';
+import { toolUsePrompt } from "../prompts.js";
 // TODO: restructure the prompts to avoid duplications
-export const basePrompt = 'You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.';
+export const basePrompt = "You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.";
 export const editingFilesPrompt = (toolNamePrefix) => `
 ====
 
@@ -117,9 +117,9 @@ SYSTEM INFORMATION
 
 Operating System: ${info.os}`;
 export const customInstructions = (customInstructions) => {
-    const joined = customInstructions.join('\n');
-    if (joined.trim() === '') {
-        return '';
+    const joined = customInstructions.join("\n");
+    if (joined.trim() === "") {
+        return "";
     }
     return `
 ====
@@ -133,14 +133,14 @@ ${joined}`;
 export const customCommands = (commands) => {
     const joined = Object.entries(commands)
         .map(([name, command]) => {
-        if (typeof command === 'string') {
+        if (typeof command === "string") {
             return `- ${name}\n  - Command: \`${command}\``;
         }
         return `- ${name}\n  - Command: \`${command.command}\`\n  - Description: ${command.description}`;
     })
-        .join('\n');
-    if (joined.trim() === '') {
-        return '';
+        .join("\n");
+    if (joined.trim() === "") {
+        return "";
     }
     return `
 ====
@@ -162,4 +162,4 @@ ${systemInformation(info)}
 ${customInstructions(instructions)}
 ${customCommands(commands)}
 `;
-//# sourceMappingURL=prompts.js.map
+//# sourceMappingURL=prompts.js.map

package/dist/Agent/index.d.ts

@@ -1,2 +1,2 @@
-export * from './AgentBase';
-export * from './CoderAgent';
+export * from "./AgentBase.js";
+export * from "./CoderAgent/index.js";

package/dist/Agent/index.js

@@ -1,3 +1,3 @@
-export * from './AgentBase';
-export * from './CoderAgent';
-//# sourceMappingURL=index.js.map
+export * from "./AgentBase.js";
+export * from "./CoderAgent/index.js";
+//# sourceMappingURL=index.js.map

package/dist/Agent/parseAssistantMessage.d.ts

@@ -1,10 +1,10 @@
-import type { ToolInfo } from '../tools';
+import type { ToolInfo } from "../tools/index.js";
 export interface TextContent {
-    type: 'text';
+    type: "text";
     content: string;
 }
 export interface ToolUse {
-    type: 'tool_use';
+    type: "tool_use";
     name: string;
     params: Record<string, string>;
 }

package/dist/Agent/prompts.d.ts

@@ -1,4 +1,4 @@
-import type { ToolInfo } from '../tools';
+import type { ToolInfo } from "../tools/index.js";
 export declare const toolUsePrompt: (tools: ToolInfo[], toolNamePrefix: string) => string;
 export declare const responsePrompts: {
     readonly errorInvokeTool: (tool: string, error: unknown) => string;

package/dist/AiService/AiServiceBase.d.ts

@@ -1,12 +1,12 @@
-import type { Anthropic } from '@anthropic-ai/sdk';
-import type { ModelInfo } from './ModelInfo';
+import type { Anthropic } from "@anthropic-ai/sdk";
+import type { ModelInfo } from "./ModelInfo.js";
 export type ApiStreamChunk = ApiStreamTextChunk | ApiStreamUsageChunk;
 export interface ApiStreamTextChunk {
-    type: 'text';
+    type: "text";
     text: string;
 }
 export interface ApiStreamUsageChunk {
-    type: 'usage';
+    type: "usage";
     inputTokens: number;
     outputTokens: number;
     cacheWriteTokens?: number;

package/dist/AiService/AnthropicService.d.ts

@@ -1,5 +1,5 @@
-import { AiServiceBase, type AiServiceOptions, type ApiStream, type MessageParam } from './AiServiceBase';
-import { type AnthropicModelId, type ModelInfo } from './ModelInfo';
+import { AiServiceBase, type AiServiceOptions, type ApiStream, type MessageParam } from "./AiServiceBase.js";
+import { type AnthropicModelId, type ModelInfo } from "./ModelInfo.js";
 export declare class AnthropicService extends AiServiceBase {
     #private;
     readonly model: {

package/dist/AiService/AnthropicService.js

@@ -1,9 +1,9 @@
 // source: https://github.com/cline/cline/blob/f6c19c29a64ca84e9360df7ab2c07d128dcebe64/src/api/providers/anthropic.ts
-import { Anthropic } from '@anthropic-ai/sdk';
-import { createServiceLogger } from '../logger';
-import { AiServiceBase } from './AiServiceBase';
-import { anthropicDefaultModelId, anthropicModels } from './ModelInfo';
-const logger = createServiceLogger('AnthropicService');
+import { Anthropic } from "@anthropic-ai/sdk";
+import { createServiceLogger } from "../logger.js";
+import { AiServiceBase } from "./AiServiceBase.js";
+import { anthropicDefaultModelId, anthropicModels } from "./ModelInfo.js";
+const logger = createServiceLogger("AnthropicService");
 export class AnthropicService extends AiServiceBase {
     #options;
     #client;
@@ -22,20 +22,20 @@ export class AnthropicService extends AiServiceBase {
         };
     }
     async *send(systemPrompt, messages) {
-        logger.debug({ modelId: this.model.id, messagesCount: messages.length }, 'Starting message stream');
+        logger.debug({ modelId: this.model.id, messagesCount: messages.length }, "Starting message stream");
         let stream;
         const modelId = this.model.id;
         switch (modelId) {
             // 'latest' alias does not support cache_control
-            case 'claude-3-5-sonnet-20241022':
-            case 'claude-3-5-haiku-20241022':
-            case 'claude-3-opus-20240229':
-            case 'claude-3-haiku-20240307': {
+            case "claude-3-5-sonnet-20241022":
+            case "claude-3-5-haiku-20241022":
+            case "claude-3-opus-20240229":
+            case "claude-3-haiku-20240307": {
                 /*
                         The latest message will be the new user message, one before will be the assistant message from a previous request, and the user message before that will be a previously cached user message. So we need to mark the latest user message as ephemeral to cache it for the next request, and mark the second to last user message as ephemeral to let the server know the last message to retrieve from the cache for the current request..
                         */
                 const userMsgIndices = messages.reduce((acc, msg, index) => {
-                    if (msg.role === 'user') {
+                    if (msg.role === "user") {
                         acc.push(index);
                     }
                     return acc;
@@ -49,21 +49,21 @@ export class AnthropicService extends AiServiceBase {
                     system: [
                         {
                             text: systemPrompt,
-                            type: 'text',
-                            cache_control: { type: 'ephemeral' },
+                            type: "text",
+                            cache_control: { type: "ephemeral" },
                         },
                     ], // setting cache breakpoint for system prompt so new tasks can reuse it
                     messages: messages.map((message, index) => {
                         if (index === lastUserMsgIndex || index === secondLastMsgUserIndex) {
                             return {
                                 ...message,
-                                content: typeof message.content === 'string'
+                                content: typeof message.content === "string"
                                     ? [
                                         {
-                                            type: 'text',
+                                            type: "text",
                                             text: message.content,
                                             cache_control: {
-                                                type: 'ephemeral',
+                                                type: "ephemeral",
                                             },
                                         },
                                     ]
@@ -71,7 +71,7 @@ export class AnthropicService extends AiServiceBase {
                                         ? {
                                             ...content,
                                             cache_control: {
-                                                type: 'ephemeral',
+                                                type: "ephemeral",
                                             },
                                         }
                                         : content),
@@ -88,13 +88,13 @@ export class AnthropicService extends AiServiceBase {
                     // https://github.com/anthropics/anthropic-sdk-typescript?tab=readme-ov-file#default-headers
                     // https://github.com/anthropics/anthropic-sdk-typescript/commit/c920b77fc67bd839bfeb6716ceab9d7c9bbe7393
                     switch (modelId) {
-                        case 'claude-3-5-sonnet-20241022':
-                        case 'claude-3-5-haiku-20241022':
-                        case 'claude-3-opus-20240229':
-                        case 'claude-3-haiku-20240307':
+                        case "claude-3-5-sonnet-20241022":
+                        case "claude-3-5-haiku-20241022":
+                        case "claude-3-opus-20240229":
+                        case "claude-3-haiku-20240307":
                             return {
                                 headers: {
-                                    'anthropic-beta': 'prompt-caching-2024-07-31',
+                                    "anthropic-beta": "prompt-caching-2024-07-31",
                                 },
                             };
                         default:
@@ -108,7 +108,7 @@ export class AnthropicService extends AiServiceBase {
                     model: modelId,
                     max_tokens: this.model.info.maxTokens || 8192,
                     temperature: 0,
-                    system: [{ text: systemPrompt, type: 'text' }],
+                    system: [{ text: systemPrompt, type: "text" }],
                     messages,
                     // tools,
                     // tool_choice: { type: "auto" },
@@ -117,69 +117,69 @@ export class AnthropicService extends AiServiceBase {
                 break;
             }
         }
-        logger.debug('Stream created, processing chunks');
+        logger.debug("Stream created, processing chunks");
         for await (const chunk of stream) {
             switch (chunk.type) {
-                case 'message_start': {
+                case "message_start": {
                     // tells us cache reads/writes/input/output
                     const usage = chunk.message.usage;
                     const usageInfo = {
-                        type: 'usage',
+                        type: "usage",
                         inputTokens: usage.input_tokens || 0,
                         outputTokens: usage.output_tokens || 0,
                         cacheWriteTokens: usage.cache_creation_input_tokens || undefined,
                         cacheReadTokens: usage.cache_read_input_tokens || undefined,
                     };
-                    logger.trace({ usage: usageInfo }, 'Message start usage');
+                    logger.trace({ usage: usageInfo }, "Message start usage");
                     yield usageInfo;
                     break;
                 }
-                case 'message_delta': {
+                case "message_delta": {
                     // tells us stop_reason, stop_sequence, and output tokens along the way and at the end of the message
                     const deltaUsage = {
-                        type: 'usage',
+                        type: "usage",
                         inputTokens: 0,
                         outputTokens: chunk.usage.output_tokens || 0,
                     };
-                    logger.trace({ usage: deltaUsage }, 'Message delta usage');
+                    logger.trace({ usage: deltaUsage }, "Message delta usage");
                     yield deltaUsage;
                     break;
                 }
-                case 'message_stop':
-                    logger.debug('Message stream completed');
+                case "message_stop":
+                    logger.debug("Message stream completed");
                     break;
-                case 'content_block_start':
+                case "content_block_start":
                     switch (chunk.content_block.type) {
-                        case 'text':
+                        case "text":
                             // we may receive multiple text blocks, in which case just insert a line break between them
                             if (chunk.index > 0) {
                                 yield {
-                                    type: 'text',
-                                    text: '\n',
+                                    type: "text",
+                                    text: "\n",
                                 };
                             }
                             yield {
-                                type: 'text',
+                                type: "text",
                                 text: chunk.content_block.text,
                             };
                             break;
                     }
                     break;
-                case 'content_block_delta':
+                case "content_block_delta":
                     switch (chunk.delta.type) {
-                        case 'text_delta':
+                        case "text_delta":
                             yield {
-                                type: 'text',
+                                type: "text",
                                 text: chunk.delta.text,
                             };
                             break;
                     }
                     break;
-                case 'content_block_stop':
+                case "content_block_stop":
                     break;
             }
         }
-        logger.debug('Stream ended');
+        logger.debug("Stream ended");
     }
 }
-//# sourceMappingURL=AnthropicService.js.map
+//# sourceMappingURL=AnthropicService.js.map

package/dist/AiService/DeepSeekService.d.ts

@@ -1,5 +1,5 @@
-import { AiServiceBase, type AiServiceOptions, type ApiStream, type MessageParam } from './AiServiceBase';
-import { type ModelInfo } from './ModelInfo';
+import { AiServiceBase, type AiServiceOptions, type ApiStream, type MessageParam } from "./AiServiceBase.js";
+import { type ModelInfo } from "./ModelInfo.js";
 export declare class DeepSeekService extends AiServiceBase {
     #private;
     readonly model: {

package/dist/AiService/DeepSeekService.js

@@ -1,17 +1,17 @@
 // source: https://github.com/cline/cline/blob/ce2610a6eafd860305ba9b12533db19f2a5385ad/src/api/providers/deepseek.ts
-import OpenAI from 'openai';
-import { createServiceLogger } from '../logger';
-import { AiServiceBase } from './AiServiceBase';
-import { deepSeekDefaultModelId, deepSeekModels } from './ModelInfo';
-import { convertToOpenAiMessages } from './utils';
-const logger = createServiceLogger('DeepSeekService');
+import OpenAI from "openai";
+import { createServiceLogger } from "../logger.js";
+import { AiServiceBase } from "./AiServiceBase.js";
+import { deepSeekDefaultModelId, deepSeekModels } from "./ModelInfo.js";
+import { convertToOpenAiMessages } from "./utils.js";
+const logger = createServiceLogger("DeepSeekService");
 export class DeepSeekService extends AiServiceBase {
     #client;
     model;
     constructor(options) {
         super();
         this.#client = new OpenAI({
-            baseURL: 'https://api.deepseek.com/v1',
+            baseURL: "https://api.deepseek.com/v1",
             apiKey: options.apiKey,
         });
         const id = (options.modelId || deepSeekDefaultModelId);
@@ -21,12 +21,12 @@ export class DeepSeekService extends AiServiceBase {
         };
     }
     async *send(systemPrompt, messages) {
-        logger.debug({ modelId: this.model.id, messagesCount: messages.length }, 'Starting message stream');
+        logger.debug({ modelId: this.model.id, messagesCount: messages.length }, "Starting message stream");
         const openAiMessages = [
-            { role: 'system', content: systemPrompt },
+            { role: "system", content: systemPrompt },
             ...convertToOpenAiMessages(messages),
         ];
-        logger.trace({ modelId: this.model.id, messagesCount: messages.length }, 'Sending messages to Ollama');
+        logger.trace({ modelId: this.model.id, messagesCount: messages.length }, "Sending messages to Ollama");
         const stream = await this.#client.chat.completions.create({
             model: this.model.id,
             max_completion_tokens: this.model.info.maxTokens,
@@ -39,13 +39,13 @@ export class DeepSeekService extends AiServiceBase {
             const delta = chunk.choices[0]?.delta;
             if (delta?.content) {
                 yield {
-                    type: 'text',
+                    type: "text",
                     text: delta.content,
                 };
             }
             if (chunk.usage) {
                 yield {
-                    type: 'usage',
+                    type: "usage",
                     // deepseek reports total input AND cache reads/writes, see context caching: https://api-docs.deepseek.com/guides/kv_cache
                     // where the input tokens is the sum of the cache hits/misses, while anthropic reports them as separate tokens.
                     // This is important to know for
@@ -58,7 +58,7 @@ export class DeepSeekService extends AiServiceBase {
                 };
             }
         }
-        logger.debug('Stream ended');
+        logger.debug("Stream ended");
     }
 }
-//# sourceMappingURL=DeepSeekService.js.map
+//# sourceMappingURL=DeepSeekService.js.map

package/dist/AiService/OllamaService.d.ts

@@ -1,5 +1,5 @@
-import { AiServiceBase, type AiServiceOptions, type ApiStream, type MessageParam } from './AiServiceBase';
-import { type ModelInfo } from './ModelInfo';
+import { AiServiceBase, type AiServiceOptions, type ApiStream, type MessageParam } from "./AiServiceBase.js";
+import { type ModelInfo } from "./ModelInfo.js";
 export declare class OllamaService extends AiServiceBase {
     #private;
     readonly model: {

package/dist/AiService/OllamaService.js

@@ -1,31 +1,31 @@
 // source: https://github.com/cline/cline/blob/f6c19c29a64ca84e9360df7ab2c07d128dcebe64/src/api/providers/ollama.ts
-import OpenAI from 'openai';
-import { createServiceLogger } from '../logger';
-import { AiServiceBase } from './AiServiceBase';
-import { openAiModelInfoSaneDefaults } from './ModelInfo';
-import { convertToOpenAiMessages } from './utils';
-const logger = createServiceLogger('OllamaService');
+import OpenAI from "openai";
+import { createServiceLogger } from "../logger.js";
+import { AiServiceBase } from "./AiServiceBase.js";
+import { openAiModelInfoSaneDefaults } from "./ModelInfo.js";
+import { convertToOpenAiMessages } from "./utils.js";
+const logger = createServiceLogger("OllamaService");
 export class OllamaService extends AiServiceBase {
     #client;
     model;
     constructor(options) {
         super();
         this.#client = new OpenAI({
-            baseURL: `${options.baseUrl || 'http://localhost:11434'}/v1`,
-            apiKey: 'ollama',
+            baseURL: `${options.baseUrl || "http://localhost:11434"}/v1`,
+            apiKey: "ollama",
         });
         this.model = {
-            id: options.modelId || '',
+            id: options.modelId || "",
             info: openAiModelInfoSaneDefaults,
         };
     }
     async *send(systemPrompt, messages) {
-        logger.debug({ modelId: this.model.id, messagesCount: messages.length }, 'Starting message stream');
+        logger.debug({ modelId: this.model.id, messagesCount: messages.length }, "Starting message stream");
         const openAiMessages = [
-            { role: 'system', content: systemPrompt },
+            { role: "system", content: systemPrompt },
             ...convertToOpenAiMessages(messages),
         ];
-        logger.trace({ modelId: this.model.id, messagesCount: messages.length }, 'Sending messages to Ollama');
+        logger.trace({ modelId: this.model.id, messagesCount: messages.length }, "Sending messages to Ollama");
         const stream = await this.#client.chat.completions.create({
             model: this.model.id,
             messages: openAiMessages,
@@ -36,12 +36,12 @@ export class OllamaService extends AiServiceBase {
             const delta = chunk.choices[0]?.delta;
             if (delta?.content) {
                 yield {
-                    type: 'text',
+                    type: "text",
                     text: delta.content,
                 };
             }
         }
-        logger.debug('Stream ended');
+        logger.debug("Stream ended");
     }
 }
-//# sourceMappingURL=OllamaService.js.map
+//# sourceMappingURL=OllamaService.js.map

package/dist/AiService/index.d.ts

@@ -1,8 +1,8 @@
-import type { AiServiceBase, AiServiceOptions, MessageParam } from './AiServiceBase';
-import { AnthropicService } from './AnthropicService';
-import { DeepSeekService } from './DeepSeekService';
-import type { ModelInfo } from './ModelInfo';
-import { OllamaService } from './OllamaService';
+import type { AiServiceBase, AiServiceOptions, MessageParam } from "./AiServiceBase.js";
+import { AnthropicService } from "./AnthropicService.js";
+import { DeepSeekService } from "./DeepSeekService.js";
+import type { ModelInfo } from "./ModelInfo.js";
+import { OllamaService } from "./OllamaService.js";
 export declare enum AiServiceProvider {
     Anthropic = "anthropic",
     Ollama = "ollama",

package/dist/AiService/index.js

@@ -1,6 +1,6 @@
-import { AnthropicService } from './AnthropicService';
-import { DeepSeekService } from './DeepSeekService';
-import { OllamaService } from './OllamaService';
+import { AnthropicService } from "./AnthropicService.js";
+import { DeepSeekService } from "./DeepSeekService.js";
+import { OllamaService } from "./OllamaService.js";
 export var AiServiceProvider;
 (function (AiServiceProvider) {
     AiServiceProvider["Anthropic"] = "anthropic";
@@ -17,4 +17,4 @@ export const createService = (provider, options) => {
             return new DeepSeekService(options);
     }
 };
-//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/AiService/utils.js

@@ -1,10 +1,10 @@
 // source: https://github.com/cline/cline/blob/f6c19c29a64ca84e9360df7ab2c07d128dcebe64/src/api/transform/openai-format.ts
-import { createServiceLogger } from '../logger';
-const logger = createServiceLogger('utils');
+import { createServiceLogger } from "../logger.js";
+const logger = createServiceLogger("utils");
 export function convertToOpenAiMessages(anthropicMessages) {
     const openAiMessages = [];
     for (const anthropicMessage of anthropicMessages) {
-        if (typeof anthropicMessage.content === 'string') {
+        if (typeof anthropicMessage.content === "string") {
             openAiMessages.push({
                 role: anthropicMessage.role,
                 content: anthropicMessage.content,
@@ -19,12 +19,12 @@ export function convertToOpenAiMessages(anthropicMessages) {
               { role: "assistant", content?: "" | null, tool_calls?: [{ id: "", function: { name: "", arguments: "" }, type: "function" }] },
               { role: "tool", tool_call_id: "", content: ""}
                */
-            if (anthropicMessage.role === 'user') {
+            if (anthropicMessage.role === "user") {
                 const { nonToolMessages, toolMessages } = anthropicMessage.content.reduce((acc, part) => {
-                    if (part.type === 'tool_result') {
+                    if (part.type === "tool_result") {
                         acc.toolMessages.push(part);
                     }
-                    else if (part.type === 'text' || part.type === 'image') {
+                    else if (part.type === "text" || part.type === "image") {
                         acc.nonToolMessages.push(part);
                     } // user cannot send tool_use messages
                     return acc;
@@ -34,23 +34,23 @@ export function convertToOpenAiMessages(anthropicMessages) {
                 for (const toolMessage of toolMessages) {
                     // The Anthropic SDK allows tool results to be a string or an array of text and image blocks, enabling rich and structured content. In contrast, the OpenAI SDK only supports tool results as a single string, so we map the Anthropic tool result parts into one concatenated string to maintain compatibility.
                     let content;
-                    if (typeof toolMessage.content === 'string') {
+                    if (typeof toolMessage.content === "string") {
                         content = toolMessage.content;
                     }
                     else {
                         content =
                             toolMessage.content
                                 ?.map((part) => {
-                                if (part.type === 'image') {
+                                if (part.type === "image") {
                                     toolResultImages.push(part);
-                                    return '(see following user message for image)';
+                                    return "(see following user message for image)";
                                 }
                                 return part.text;
                             })
-                                .join('\n') ?? '';
+                                .join("\n") ?? "";
                     }
                     openAiMessages.push({
-                        role: 'tool',
+                        role: "tool",
                         tool_call_id: toolMessage.tool_use_id,
                         content: content,
                     });
@@ -73,27 +73,27 @@ export function convertToOpenAiMessages(anthropicMessages) {
                 // Process non-tool messages
                 if (nonToolMessages.length > 0) {
                     openAiMessages.push({
-                        role: 'user',
+                        role: "user",
                         content: nonToolMessages.map((part) => {
-                            if (part.type === 'image') {
+                            if (part.type === "image") {
                                 return {
-                                    type: 'image_url',
+                                    type: "image_url",
                                     image_url: {
                                         url: `data:${part.source.media_type};base64,${part.source.data}`,
                                     },
                                 };
                             }
-                            return { type: 'text', text: part.text };
+                            return { type: "text", text: part.text };
                         }),
                     });
                 }
             }
-            else if (anthropicMessage.role === 'assistant') {
+            else if (anthropicMessage.role === "assistant") {
                 const { nonToolMessages, toolMessages } = anthropicMessage.content.reduce((acc, part) => {
-                    if (part.type === 'tool_use') {
+                    if (part.type === "tool_use") {
                         acc.toolMessages.push(part);
                     }
-                    else if (part.type === 'text' || part.type === 'image') {
+                    else if (part.type === "text" || part.type === "image") {
                         acc.nonToolMessages.push(part);
                     } // assistant cannot send tool_result messages
                     return acc;
@@ -103,17 +103,17 @@ export function convertToOpenAiMessages(anthropicMessages) {
                 if (nonToolMessages.length > 0) {
                     content = nonToolMessages
                         .map((part) => {
-                        if (part.type === 'image') {
-                            return ''; // impossible as the assistant cannot send images
+                        if (part.type === "image") {
+                            return ""; // impossible as the assistant cannot send images
                         }
                         return part.text;
                     })
-                        .join('\n');
+                        .join("\n");
                 }
                 // Process tool use messages
                 const tool_calls = toolMessages.map((toolMessage) => ({
                     id: toolMessage.id,
-                    type: 'function',
+                    type: "function",
                     function: {
                         name: toolMessage.name,
                         // json string
@@ -121,7 +121,7 @@ export function convertToOpenAiMessages(anthropicMessages) {
                     },
                 }));
                 openAiMessages.push({
-                    role: 'assistant',
+                    role: "assistant",
                     content,
                     // Cannot be an empty array. API expects an array with minimum length 1, and will respond with an error if it's empty
                     tool_calls: tool_calls.length > 0 ? tool_calls : undefined,
@@ -136,23 +136,23 @@ export function convertToAnthropicMessage(completion) {
     const openAiMessage = completion.choices[0].message;
     const anthropicMessage = {
         id: completion.id,
-        type: 'message',
+        type: "message",
         role: openAiMessage.role, // always "assistant"
         content: [
             {
-                type: 'text',
-                text: openAiMessage.content || '',
+                type: "text",
+                text: openAiMessage.content || "",
             },
         ],
         model: completion.model,
         stop_reason: (() => {
             switch (completion.choices[0].finish_reason) {
-                case 'stop':
-                    return 'end_turn';
-                case 'length':
-                    return 'max_tokens';
-                case 'tool_calls':
-                    return 'tool_use';
+                case "stop":
+                    return "end_turn";
+                case "length":
+                    return "max_tokens";
+                case "tool_calls":
+                    return "tool_use";
                 default:
                     return null;
             }
@@ -169,13 +169,13 @@ export function convertToAnthropicMessage(completion) {
         anthropicMessage.content.push(...openAiMessage.tool_calls.map((toolCall) => {
             let parsedInput = {};
             try {
-                parsedInput = JSON.parse(toolCall.function.arguments || '{}');
+                parsedInput = JSON.parse(toolCall.function.arguments || "{}");
             }
             catch (error) {
-                logger.warn('Failed to parse tool arguments:', error);
+                logger.warn("Failed to parse tool arguments:", error);
             }
             return {
-                type: 'tool_use',
+                type: "tool_use",
                 id: toolCall.id,
                 name: toolCall.function.name,
                 input: parsedInput,
@@ -184,4 +184,4 @@ export function convertToAnthropicMessage(completion) {
     }
     return anthropicMessage;
 }
-//# sourceMappingURL=utils.js.map
+//# sourceMappingURL=utils.js.map

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-export * from './AiService';
-export * from './Agent';
-export * from './tools';
-export * from './logger';
+export * from "./AiService/index.js";
+export * from "./Agent/index.js";
+export * from "./tools/index.js";
+export * from "./logger.js";

package/dist/index.js

@@ -1,5 +1,5 @@
-export * from './AiService';
-export * from './Agent';
-export * from './tools';
-export * from './logger';
-//# sourceMappingURL=index.js.map
+export * from "./AiService/index.js";
+export * from "./Agent/index.js";
+export * from "./tools/index.js";
+export * from "./logger.js";
+//# sourceMappingURL=index.js.map

package/dist/tools/index.d.ts

@@ -1,3 +1,3 @@
-export * from './types';
-export * from './tools';
-export * as allTools from './tools';
+export * from "./types.js";
+export * from "./tools.js";
+export * as allTools from "./tools.js";

package/dist/tools/index.js

@@ -1,4 +1,4 @@
-export * from './types';
-export * from './tools';
-export * as allTools from './tools';
-//# sourceMappingURL=index.js.map
+export * from "./types.js";
+export * from "./tools.js";
+export * as allTools from "./tools.js";
+//# sourceMappingURL=index.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "license": "AGPL-3.0",
   "type": "module",
   "exports": {