Base profiles

Author: michaelroudnitski

src/core/providers/openai.ts

@@ -0,0 +1,18 @@
+import {ChatOpenAI} from '@langchain/openai'
+
+export interface OpenAIConfig {
+  apiKey: string
+  maxTokens?: number
+  model: string
+  temperature?: number
+}
+
+export function createClient(config: OpenAIConfig): ChatOpenAI {
+  return new ChatOpenAI({
+    apiKey: config.apiKey,
+    maxRetries: 3,
+    maxTokens: config.maxTokens || 2000,
+    model: config.model,
+    temperature: config.temperature || 0.3,
+  })
+}

src/core/providers/watsonx.ts

@@ -1,49 +1,24 @@
 import {ChatWatsonx} from '@langchain/community/chat_models/ibm'
 
-/**
- * Configuration for watsonx.ai client
- */
 export interface WatsonxConfig {
   apiKey: string
   maxNewTokens?: number
-  model?: string
+  model: string
   projectId: string
   serviceUrl: string
   temperature?: number
 }
 
-/**
- * Creates and returns a configured watsonx.ai chat model with IAM authentication
- */
-export function createClient(config?: Partial<WatsonxConfig>): ChatWatsonx {
-  // Get configuration from environment variables or provided config
-  const serviceUrl = config?.serviceUrl || process.env.WATSONX_AI_SERVICE_URL
-  const projectId = config?.projectId || process.env.WATSONX_AI_PROJECT_ID
-  const apiKey = config?.apiKey || process.env.WATSONX_AI_APIKEY
-
-  // Validate required configuration
-  if (!serviceUrl) {
-    throw new Error('WATSONX_AI_SERVICE_URL is required')
-  }
-
-  if (!projectId) {
-    throw new Error('WATSONX_AI_PROJECT_ID is required')
-  }
-
-  if (!apiKey) {
-    throw new Error('WATSONX_AI_APIKEY is required')
-  }
-
-  // Create and return the chat model with IAM authentication
+export function createClient(config: WatsonxConfig): ChatWatsonx {
   return new ChatWatsonx({
     maxRetries: 3,
-    maxTokens: config?.maxNewTokens || 2000,
-    model: config?.model || 'ibm/granite-4-h-small',
-    projectId,
-    serviceUrl,
-    temperature: config?.temperature || 0.3,
+    maxTokens: config.maxNewTokens || 2000,
+    model: config.model,
+    projectId: config.projectId,
+    serviceUrl: config.serviceUrl,
+    temperature: config.temperature || 0.3,
     version: '2024-05-31',
-    watsonxAIApikey: apiKey,
+    watsonxAIApikey: config.apiKey,
     watsonxAIAuthType: 'iam',
   })
 }

src/lib/profile/factory.ts

@@ -0,0 +1,18 @@
+import type {BaseChatModel} from '@langchain/core/language_models/chat_models'
+
+import type {Profile} from './types.js'
+
+import {createClient as createOpenAIClient} from '../../core/providers/openai.js'
+import {createClient as createWatsonxClient} from '../../core/providers/watsonx.js'
+
+export function createProviderFromProfile(profile: Profile): BaseChatModel {
+  switch (profile.provider) {
+    case 'openai': {
+      return createOpenAIClient(profile.config)
+    }
+
+    case 'watsonx': {
+      return createWatsonxClient(profile.config)
+    }
+  }
+}

src/lib/profile/storage.ts

@@ -0,0 +1,64 @@
+import {existsSync, mkdirSync, readdirSync, readFileSync, rmSync, writeFileSync} from 'node:fs'
+import {homedir} from 'node:os'
+import {join} from 'node:path'
+
+import type {Profile} from './types.js'
+
+export function getProfilesDir(): string {
+  const home = process.env.HOME || homedir()
+  return join(home, '.config', 'translation-ai-cli', 'profiles')
+}
+
+export function ensureProfilesDir(): void {
+  const dir = getProfilesDir()
+  if (!existsSync(dir)) {
+    mkdirSync(dir, {recursive: true})
+  }
+}
+
+export function getProfilePath(name: string): string {
+  return join(getProfilesDir(), `${name}.json`)
+}
+
+export function saveProfile(profile: Profile): void {
+  ensureProfilesDir()
+  const path = getProfilePath(profile.name)
+  writeFileSync(path, JSON.stringify(profile, null, 2), {
+    encoding: 'utf8',
+    mode: 0o600, // only owner can read/write
+  })
+}
+
+export function loadProfile(name: string): Profile {
+  const path = getProfilePath(name)
+  if (!existsSync(path)) {
+    throw new Error(`Profile "${name}" does not exist`)
+  }
+
+  const content = readFileSync(path, 'utf8')
+  return JSON.parse(content) as Profile
+}
+
+export function listProfiles(): string[] {
+  const dir = getProfilesDir()
+  if (!existsSync(dir)) {
+    return []
+  }
+
+  return readdirSync(dir)
+    .filter((file) => file.endsWith('.json'))
+    .map((file) => file.replace(/\.json$/, ''))
+}
+
+export function deleteProfile(name: string): void {
+  const path = getProfilePath(name)
+  if (!existsSync(path)) {
+    throw new Error(`Profile "${name}" does not exist`)
+  }
+
+  rmSync(path)
+}
+
+export function profileExists(name: string): boolean {
+  return existsSync(getProfilePath(name))
+}

src/lib/profile/types.ts

@@ -0,0 +1,25 @@
+export interface OpenAIProfileConfig {
+  apiKey: string
+  model: string
+}
+
+export interface WatsonxProfileConfig {
+  apiKey: string
+  model: string
+  projectId: string
+  serviceUrl: string
+}
+
+export interface OpenAIProfile {
+  config: OpenAIProfileConfig
+  name: string
+  provider: 'openai'
+}
+
+export interface WatsonxProfile {
+  config: WatsonxProfileConfig
+  name: string
+  provider: 'watsonx'
+}
+
+export type Profile = OpenAIProfile | WatsonxProfile