@artemiskit/adapter-openai 0.1.2

package/dist/types.d.ts

@@ -0,0 +1,24 @@
+/**
+ * OpenAI adapter types
+ */
+export interface OpenAIAdapterConfig {
+    provider: 'openai';
+    apiKey?: string;
+    baseUrl?: string;
+    defaultModel?: string;
+    timeout?: number;
+    maxRetries?: number;
+    organization?: string;
+}
+export interface AzureOpenAIAdapterConfig {
+    provider: 'azure-openai';
+    apiKey?: string;
+    baseUrl?: string;
+    defaultModel?: string;
+    timeout?: number;
+    maxRetries?: number;
+    resourceName: string;
+    deploymentName: string;
+    apiVersion: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,QAAQ,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,wBAAwB;IACvC,QAAQ,EAAE,cAAc,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;CACpB"}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@artemiskit/adapter-openai",
+  "version": "0.1.2",
+  "description": "OpenAI and Azure OpenAI adapter for ArtemisKit LLM evaluation toolkit",
+  "type": "module",
+  "license": "Apache-2.0",
+  "author": "code-sensei",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/code-sensei/artemiskit.git",
+    "directory": "packages/adapters/openai"
+  },
+  "bugs": {
+    "url": "https://github.com/code-sensei/artemiskit/issues"
+  },
+  "keywords": [
+    "llm",
+    "openai",
+    "azure",
+    "gpt-4",
+    "adapter",
+    "artemiskit"
+  ],
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "scripts": {
+    "build": "tsc && bun build ./src/index.ts --outdir ./dist --target bun",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist",
+    "test": "bun test"
+  },
+  "dependencies": {
+    "@artemiskit/core": "workspace:*",
+    "openai": "^4.28.0",
+    "nanoid": "^5.0.0"
+  },
+  "devDependencies": {
+    "@types/bun": "^1.1.0",
+    "typescript": "^5.3.0"
+  }
+}

package/src/client.ts

@@ -0,0 +1,166 @@
+/**
+ * OpenAI SDK Adapter
+ * Supports: OpenAI, Azure OpenAI, OpenAI-compatible APIs
+ */
+
+import type {
+  AdapterConfig,
+  GenerateOptions,
+  GenerateResult,
+  ModelCapabilities,
+  ModelClient,
+} from '@artemiskit/core';
+import { nanoid } from 'nanoid';
+import OpenAI from 'openai';
+import type { AzureOpenAIAdapterConfig, OpenAIAdapterConfig } from './types';
+
+export class OpenAIAdapter implements ModelClient {
+  private client: OpenAI;
+  private config: OpenAIAdapterConfig | AzureOpenAIAdapterConfig;
+  readonly provider: string;
+
+  constructor(config: AdapterConfig) {
+    this.config = config as OpenAIAdapterConfig | AzureOpenAIAdapterConfig;
+
+    if (config.provider === 'azure-openai') {
+      const azureConfig = config as AzureOpenAIAdapterConfig;
+      this.provider = 'azure-openai';
+
+      this.client = new OpenAI({
+        apiKey: azureConfig.apiKey,
+        baseURL: `https://${azureConfig.resourceName}.openai.azure.com/openai/deployments/${azureConfig.deploymentName}`,
+        defaultQuery: { 'api-version': azureConfig.apiVersion },
+        defaultHeaders: { 'api-key': azureConfig.apiKey ?? '' },
+        timeout: azureConfig.timeout ?? 60000,
+        maxRetries: azureConfig.maxRetries ?? 2,
+      });
+    } else {
+      const openaiConfig = config as OpenAIAdapterConfig;
+      this.provider = 'openai';
+
+      this.client = new OpenAI({
+        apiKey: openaiConfig.apiKey,
+        baseURL: openaiConfig.baseUrl,
+        organization: openaiConfig.organization,
+        timeout: openaiConfig.timeout ?? 60000,
+        maxRetries: openaiConfig.maxRetries ?? 2,
+      });
+    }
+  }
+
+  async generate(options: GenerateOptions): Promise<GenerateResult> {
+    const startTime = Date.now();
+    const model = options.model || this.config.defaultModel || 'gpt-4';
+
+    const messages = this.normalizePrompt(options.prompt);
+
+    const response = await this.client.chat.completions.create({
+      model,
+      messages,
+      max_tokens: options.maxTokens,
+      temperature: options.temperature,
+      top_p: options.topP,
+      seed: options.seed,
+      stop: options.stop,
+      tools: options.tools,
+      response_format: options.responseFormat,
+    });
+
+    const latencyMs = Date.now() - startTime;
+    const choice = response.choices[0];
+
+    return {
+      id: response.id || nanoid(),
+      model: response.model,
+      text: choice.message.content || '',
+      tokens: {
+        prompt: response.usage?.prompt_tokens ?? 0,
+        completion: response.usage?.completion_tokens ?? 0,
+        total: response.usage?.total_tokens ?? 0,
+      },
+      latencyMs,
+      finishReason: this.mapFinishReason(choice.finish_reason),
+      toolCalls: choice.message.tool_calls?.map((tc) => ({
+        id: tc.id,
+        type: tc.type as 'function',
+        function: {
+          name: tc.function.name,
+          arguments: tc.function.arguments,
+        },
+      })),
+      raw: response,
+    };
+  }
+
+  async *stream(options: GenerateOptions, onChunk: (chunk: string) => void): AsyncIterable<string> {
+    const model = options.model || this.config.defaultModel || 'gpt-4';
+    const messages = this.normalizePrompt(options.prompt);
+
+    const stream = await this.client.chat.completions.create({
+      model,
+      messages,
+      max_tokens: options.maxTokens,
+      temperature: options.temperature,
+      stream: true,
+    });
+
+    for await (const chunk of stream) {
+      const content = chunk.choices[0]?.delta?.content || '';
+      if (content) {
+        onChunk(content);
+        yield content;
+      }
+    }
+  }
+
+  async embed(text: string): Promise<number[]> {
+    const model =
+      this.config.provider === 'azure-openai' ? 'text-embedding-ada-002' : 'text-embedding-3-small';
+
+    const response = await this.client.embeddings.create({
+      model,
+      input: text,
+    });
+    return response.data[0].embedding;
+  }
+
+  async capabilities(): Promise<ModelCapabilities> {
+    return {
+      streaming: true,
+      functionCalling: true,
+      toolUse: true,
+      maxContext: 128000,
+      vision: true,
+      jsonMode: true,
+    };
+  }
+
+  async close(): Promise<void> {
+    // No cleanup needed
+  }
+
+  private normalizePrompt(prompt: GenerateOptions['prompt']) {
+    if (typeof prompt === 'string') {
+      return [{ role: 'user' as const, content: prompt }];
+    }
+    return prompt.map((m) => ({
+      role: m.role as 'system' | 'user' | 'assistant',
+      content: m.content,
+    }));
+  }
+
+  private mapFinishReason(reason: string | null): GenerateResult['finishReason'] {
+    switch (reason) {
+      case 'stop':
+        return 'stop';
+      case 'length':
+        return 'length';
+      case 'tool_calls':
+        return 'tool_calls';
+      case 'content_filter':
+        return 'content_filter';
+      default:
+        return 'stop';
+    }
+  }
+}

package/src/index.ts

@@ -0,0 +1,8 @@
+/**
+ * @artemiskit/adapter-openai
+ * OpenAI SDK adapter for Artemis
+ * Supports: OpenAI, Azure OpenAI, OpenAI-compatible APIs
+ */
+
+export { OpenAIAdapter } from './client';
+export * from './types';

package/src/types.ts

@@ -0,0 +1,25 @@
+/**
+ * OpenAI adapter types
+ */
+
+export interface OpenAIAdapterConfig {
+  provider: 'openai';
+  apiKey?: string;
+  baseUrl?: string;
+  defaultModel?: string;
+  timeout?: number;
+  maxRetries?: number;
+  organization?: string;
+}
+
+export interface AzureOpenAIAdapterConfig {
+  provider: 'azure-openai';
+  apiKey?: string;
+  baseUrl?: string;
+  defaultModel?: string;
+  timeout?: number;
+  maxRetries?: number;
+  resourceName: string;
+  deploymentName: string;
+  apiVersion: string;
+}

package/tsconfig.json

@@ -0,0 +1,13 @@
+{
+  "extends": "../../../tsconfig.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "noEmit": false,
+    "declaration": true,
+    "declarationMap": true,
+    "emitDeclarationOnly": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "**/*.test.ts"]
+}