@a3s-lab/code 0.1.0

package/dist/openai-compat.js

@@ -0,0 +1,263 @@
+/**
+ * OpenAI-Compatible Types and Conversion Layer
+ *
+ * This module provides OpenAI-compatible type definitions and conversion
+ * functions to allow seamless integration with OpenAI-style APIs.
+ *
+ * OpenAI Message Format:
+ * {
+ *   role: "user" | "assistant" | "system" | "tool",
+ *   content: string | ContentPart[],
+ *   name?: string,
+ *   tool_calls?: ToolCall[],
+ *   tool_call_id?: string,
+ * }
+ */
+// ============================================================================
+// Conversion Functions: OpenAI -> A3S
+// ============================================================================
+/**
+ * Convert OpenAI role to A3S MessageRole
+ * Since proto now uses OpenAI-compatible string roles, this is a simple passthrough
+ */
+export function openAIRoleToA3S(role) {
+    switch (role) {
+        case 'user':
+            return 'user';
+        case 'assistant':
+            return 'assistant';
+        case 'system':
+            return 'system';
+        case 'tool':
+        case 'function':
+            return 'tool';
+        default:
+            return 'user';
+    }
+}
+/**
+ * Convert OpenAI message to A3S Message
+ */
+export function openAIMessageToA3S(msg) {
+    let content;
+    if (typeof msg.content === 'string') {
+        content = msg.content;
+    }
+    else if (Array.isArray(msg.content)) {
+        // Extract text from content parts
+        content = msg.content
+            .filter((part) => part.type === 'text')
+            .map(part => part.text)
+            .join('\n');
+    }
+    else {
+        content = '';
+    }
+    return {
+        role: openAIRoleToA3S(msg.role),
+        content,
+        metadata: msg.name ? { name: msg.name } : undefined,
+    };
+}
+/**
+ * Convert array of OpenAI messages to A3S Messages
+ */
+export function openAIMessagesToA3S(messages) {
+    return messages.map(openAIMessageToA3S);
+}
+/**
+ * Convert OpenAI tool call to A3S ToolCall
+ */
+export function openAIToolCallToA3S(toolCall) {
+    return {
+        id: toolCall.id,
+        name: toolCall.function.name,
+        arguments: toolCall.function.arguments,
+    };
+}
+// ============================================================================
+// Conversion Functions: A3S -> OpenAI
+// ============================================================================
+/**
+ * Convert A3S MessageRole to OpenAI role
+ * Since proto now uses OpenAI-compatible string roles, this is a simple passthrough
+ */
+export function a3sRoleToOpenAI(role) {
+    switch (role) {
+        case 'user':
+            return 'user';
+        case 'assistant':
+            return 'assistant';
+        case 'system':
+            return 'system';
+        case 'tool':
+            return 'tool';
+        default:
+            return 'user';
+    }
+}
+/**
+ * Convert A3S Message to OpenAI message
+ */
+export function a3sMessageToOpenAI(msg) {
+    const openAIMsg = {
+        role: a3sRoleToOpenAI(msg.role),
+        content: msg.content,
+    };
+    if (msg.metadata?.name) {
+        openAIMsg.name = msg.metadata.name;
+    }
+    return openAIMsg;
+}
+/**
+ * Convert A3S ToolCall to OpenAI tool call
+ */
+export function a3sToolCallToOpenAI(toolCall) {
+    return {
+        id: toolCall.id,
+        type: 'function',
+        function: {
+            name: toolCall.name,
+            arguments: toolCall.arguments,
+        },
+    };
+}
+/**
+ * Convert A3S FinishReason to OpenAI finish_reason
+ * Since proto now uses OpenAI-compatible string values, this is a simple passthrough
+ */
+export function a3sFinishReasonToOpenAI(reason) {
+    switch (reason) {
+        case 'stop':
+            return 'stop';
+        case 'length':
+            return 'length';
+        case 'tool_calls':
+            return 'tool_calls';
+        case 'content_filter':
+            return 'content_filter';
+        default:
+            return null;
+    }
+}
+/**
+ * Convert A3S Usage to OpenAI usage
+ */
+export function a3sUsageToOpenAI(usage) {
+    if (!usage)
+        return undefined;
+    return {
+        prompt_tokens: usage.promptTokens || 0,
+        completion_tokens: usage.completionTokens || 0,
+        total_tokens: usage.totalTokens || 0,
+    };
+}
+/**
+ * Convert A3S GenerateResponse to OpenAI ChatCompletion
+ */
+export function a3sResponseToOpenAI(response, model = 'unknown') {
+    const message = response.message
+        ? a3sMessageToOpenAI(response.message)
+        : { role: 'assistant', content: '' };
+    // Add tool calls if present
+    if (response.toolCalls && response.toolCalls.length > 0) {
+        message.tool_calls = response.toolCalls.map(a3sToolCallToOpenAI);
+    }
+    return {
+        id: `chatcmpl-${response.sessionId}`,
+        object: 'chat.completion',
+        created: Math.floor(Date.now() / 1000),
+        model,
+        choices: [
+            {
+                index: 0,
+                message,
+                finish_reason: a3sFinishReasonToOpenAI(response.finishReason),
+            },
+        ],
+        usage: a3sUsageToOpenAI(response.usage),
+    };
+}
+/**
+ * Convert A3S GenerateChunk to OpenAI ChatCompletionChunk
+ * Since proto now uses OpenAI-compatible string values, conversion is simplified
+ */
+export function a3sChunkToOpenAI(chunk, model = 'unknown') {
+    const delta = {};
+    if (chunk.type === 'content' && chunk.content) {
+        delta.content = chunk.content;
+    }
+    if (chunk.type === 'tool_call' && chunk.toolCall) {
+        delta.tool_calls = [
+            {
+                id: chunk.toolCall.id,
+                type: 'function',
+                function: {
+                    name: chunk.toolCall.name,
+                    arguments: chunk.toolCall.arguments,
+                },
+            },
+        ];
+    }
+    let finishReason = null;
+    if (chunk.type === 'done' || chunk.finishReason) {
+        finishReason = chunk.finishReason ? a3sFinishReasonToOpenAI(chunk.finishReason) : 'stop';
+    }
+    return {
+        id: `chatcmpl-${chunk.sessionId}`,
+        object: 'chat.completion.chunk',
+        created: Math.floor(Date.now() / 1000),
+        model,
+        choices: [
+            {
+                index: 0,
+                delta,
+                finish_reason: finishReason,
+            },
+        ],
+    };
+}
+// ============================================================================
+// Helper Functions
+// ============================================================================
+/**
+ * Check if a message uses OpenAI format (lowercase role)
+ * Since proto now uses OpenAI-compatible string roles, all messages are in OpenAI format
+ */
+export function isOpenAIFormat(msg) {
+    const role = msg.role;
+    // Both A3S and OpenAI now use the same lowercase role strings
+    return ['user', 'assistant', 'system', 'tool', 'function'].includes(role);
+}
+/**
+ * Normalize message to A3S format (accepts both OpenAI and A3S formats)
+ * Since proto now uses OpenAI-compatible string roles, this is mostly a passthrough
+ */
+export function normalizeMessage(msg) {
+    if (isOpenAIFormat(msg)) {
+        return openAIMessageToA3S(msg);
+    }
+    return msg;
+}
+/**
+ * Normalize array of messages to A3S format
+ */
+export function normalizeMessages(messages) {
+    return messages.map(normalizeMessage);
+}
+// ============================================================================
+// Type Guards
+// ============================================================================
+/**
+ * Type guard for OpenAI text content
+ */
+export function isTextContent(content) {
+    return content.type === 'text';
+}
+/**
+ * Type guard for OpenAI image content
+ */
+export function isImageContent(content) {
+    return content.type === 'image_url';
+}
+//# sourceMappingURL=openai-compat.js.map

package/dist/openai-compat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-compat.js","sourceRoot":"","sources":["../ts/openai-compat.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AA+HH,+EAA+E;AAC/E,sCAAsC;AACtC,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,IAAgB;IAC9C,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,WAAW;YACd,OAAO,WAAW,CAAC;QACrB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,MAAM,CAAC;QACZ,KAAK,UAAU;YACb,OAAO,MAAM,CAAC;QAChB;YACE,OAAO,MAAM,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,GAAkB;IACnD,IAAI,OAAe,CAAC;IAEpB,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IACxB,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,kCAAkC;QAClC,OAAO,GAAG,GAAG,CAAC,OAAO;aAClB,MAAM,CAAC,CAAC,IAAI,EAA6B,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;aACjE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACtB,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,EAAE,CAAC;IACf,CAAC;IAED,OAAO;QACL,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;QAC/B,OAAO;QACP,QAAQ,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;KACpD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAyB;IAC3D,OAAO,QAAQ,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;AAC1C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAwB;IAC1D,OAAO;QACL,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;QAC5B,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;KACvC,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,sCAAsC;AACtC,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,IAAiB;IAC/C,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,WAAW;YACd,OAAO,WAAW,CAAC;QACrB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB;YACE,OAAO,MAAM,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,GAAY;IAC7C,MAAM,SAAS,GAAkB;QAC/B,IAAI,EAAE,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC;QAC/B,OAAO,EAAE,GAAG,CAAC,OAAO;KACrB,CAAC;IAEF,IAAI,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;QACvB,SAAS,CAAC,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC;IACrC,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAkB;IACpD,OAAO;QACL,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,SAAS,EAAE,QAAQ,CAAC,SAAS;SAC9B;KACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAAoB;IAEpB,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,YAAY;YACf,OAAO,YAAY,CAAC;QACtB,KAAK,gBAAgB;YACnB,OAAO,gBAAgB,CAAC;QAC1B;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAa;IAC5C,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,OAAO;QACL,aAAa,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;QACtC,iBAAiB,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;QAC9C,YAAY,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC;KACrC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,QAA0B,EAC1B,QAAgB,SAAS;IAEzB,MAAM,OAAO,GAAkB,QAAQ,CAAC,OAAO;QAC7C,CAAC,CAAC,kBAAkB,CAAC,QAAQ,CAAC,OAAO,CAAC;QACtC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IAEvC,4BAA4B;IAC5B,IAAI,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxD,OAAO,CAAC,UAAU,GAAG,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IACnE,CAAC;IAED,OAAO;QACL,EAAE,EAAE,YAAY,QAAQ,CAAC,SAAS,EAAE;QACpC,MAAM,EAAE,iBAAiB;QACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtC,KAAK;QACL,OAAO,EAAE;YACP;gBACE,KAAK,EAAE,CAAC;gBACR,OAAO;gBACP,aAAa,EAAE,uBAAuB,CAAC,QAAQ,CAAC,YAAY,CAAC;aAC9D;SACF;QACD,KAAK,EAAE,gBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC;KACxC,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,KAAoB,EACpB,QAAgB,SAAS;IAEzB,MAAM,KAAK,GAAgB,EAAE,CAAC;IAE9B,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAC9C,KAAK,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAChC,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjD,KAAK,CAAC,UAAU,GAAG;YACjB;gBACE,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE;gBACrB,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI;oBACzB,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,SAAS;iBACpC;aACF;SACF,CAAC;IACJ,CAAC;IAED,IAAI,YAAY,GACd,IAAI,CAAC;IACP,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;QAChD,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,uBAAuB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAC3F,CAAC;IAED,OAAO;QACL,EAAE,EAAE,YAAY,KAAK,CAAC,SAAS,EAAE;QACjC,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtC,KAAK;QACL,OAAO,EAAE;YACP;gBACE,KAAK,EAAE,CAAC;gBACR,KAAK;gBACL,aAAa,EAAE,YAAY;aAC5B;SACF;KACF,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,UAAU,cAAc,CAC5B,GAA4B;IAE5B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAc,CAAC;IAChC,8DAA8D;IAC9D,OAAO,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;AAC5E,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,GAA4B;IAC3D,IAAI,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,kBAAkB,CAAC,GAAG,CAAC,CAAC;IACjC,CAAC;IACD,OAAO,GAAc,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAC/B,QAAqC;IAErC,OAAO,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;AACxC,CAAC;AAED,+EAA+E;AAC/E,cAAc;AACd,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,aAAa,CAC3B,OAA0B;IAE1B,OAAO,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC;AACjC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAC5B,OAA0B;IAE1B,OAAO,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC;AACtC,CAAC"}

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "@a3s-lab/code",
+  "version": "0.1.0",
+  "description": "TypeScript SDK for A3S Code Agent Service",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "type": "module",
+  "files": [
+    "dist",
+    "proto"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "clean": "rm -rf dist",
+    "prepublishOnly": "npm run build"
+  },
+  "dependencies": {
+    "@grpc/grpc-js": "^1.9.0",
+    "@grpc/proto-loader": "^0.7.0"
+  },
+  "devDependencies": {
+    "@vitest/coverage-v8": "^1.6.1",
+    "prettier": "^3.0.0",
+    "typescript": "^5.3.0",
+    "vitest": "^1.0.0"
+  },
+  "keywords": [
+    "a3s",
+    "code-agent",
+    "grpc",
+    "sdk"
+  ],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/a3s-lab/a3s"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}