@hasna/connectors 0.4.1 → 0.5.0

package/connectors/connect-cohere/src/api/client.ts

@@ -0,0 +1,109 @@
+import type { CohereConfig } from '../types';
+import { CohereApiError } from '../types';
+
+const DEFAULT_BASE_URL = 'https://api.cohere.ai/v1';
+
+export interface RequestOptions {
+  method?: 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH';
+  params?: Record<string, string | number | boolean | undefined>;
+  body?: Record<string, unknown> | unknown[] | string;
+  headers?: Record<string, string>;
+}
+
+export class CohereClient {
+  private readonly apiKey: string;
+  private readonly baseUrl: string;
+
+  constructor(config: CohereConfig) {
+    if (!config.apiKey) {
+      throw new Error('API key is required');
+    }
+    this.apiKey = config.apiKey;
+    this.baseUrl = config.baseUrl || DEFAULT_BASE_URL;
+  }
+
+  private buildUrl(path: string, params?: Record<string, string | number | boolean | undefined>): string {
+    const url = new URL(`${this.baseUrl}${path}`);
+
+    if (params) {
+      Object.entries(params).forEach(([key, value]) => {
+        if (value !== undefined && value !== null && value !== '') {
+          url.searchParams.append(key, String(value));
+        }
+      });
+    }
+
+    return url.toString();
+  }
+
+  async request<T>(path: string, options: RequestOptions = {}): Promise<T> {
+    const { method = 'GET', params, body, headers = {} } = options;
+
+    const url = this.buildUrl(path, params);
+
+    const requestHeaders: Record<string, string> = {
+      'Authorization': `Bearer ${this.apiKey}`,
+      'Accept': 'application/json',
+      ...headers,
+    };
+
+    if (body && ['POST', 'PUT', 'PATCH'].includes(method)) {
+      requestHeaders['Content-Type'] = 'application/json';
+    }
+
+    const fetchOptions: RequestInit = {
+      method,
+      headers: requestHeaders,
+    };
+
+    if (body && ['POST', 'PUT', 'PATCH'].includes(method)) {
+      fetchOptions.body = typeof body === 'string' ? body : JSON.stringify(body);
+    }
+
+    const response = await fetch(url, fetchOptions);
+
+    if (response.status === 204) {
+      return {} as T;
+    }
+
+    let data: unknown;
+    const contentType = response.headers.get('content-type') || '';
+
+    if (contentType.includes('application/json')) {
+      const text = await response.text();
+      if (text) {
+        try {
+          data = JSON.parse(text);
+        } catch {
+          data = text;
+        }
+      }
+    } else {
+      data = await response.text();
+    }
+
+    if (!response.ok) {
+      const errorMessage = typeof data === 'object' && data !== null
+        ? (data as Record<string, string>).message || JSON.stringify(data)
+        : String(data || response.statusText);
+      throw new CohereApiError(errorMessage, response.status);
+    }
+
+    return data as T;
+  }
+
+  async get<T>(path: string, params?: Record<string, string | number | boolean | undefined>): Promise<T> {
+    return this.request<T>(path, { method: 'GET', params });
+  }
+
+  async post<T>(path: string, body?: Record<string, unknown> | unknown[] | string | object): Promise<T> {
+    return this.request<T>(path, { method: 'POST', body: body as Record<string, unknown> });
+  }
+
+  getApiKeyPreview(): string {
+    if (this.apiKey.length > 10) {
+      return `${this.apiKey.substring(0, 6)}...${this.apiKey.substring(this.apiKey.length - 4)}`;
+    }
+    return '***';
+  }
+}

package/connectors/connect-cohere/src/api/index.ts

@@ -0,0 +1,59 @@
+import type {
+  CohereConfig,
+  ChatRequest,
+  ChatResponse,
+  EmbedRequest,
+  EmbedResponse,
+  RerankRequest,
+  RerankResponse,
+  ClassifyRequest,
+  ClassifyResponse,
+  ModelsResponse,
+} from '../types';
+import { CohereClient } from './client';
+
+export class Cohere {
+  private readonly client: CohereClient;
+
+  constructor(config: CohereConfig) {
+    this.client = new CohereClient(config);
+  }
+
+  static fromEnv(): Cohere {
+    const apiKey = process.env.COHERE_API_KEY;
+    if (!apiKey) {
+      throw new Error('COHERE_API_KEY environment variable is required');
+    }
+    return new Cohere({ apiKey });
+  }
+
+  getApiKeyPreview(): string {
+    return this.client.getApiKeyPreview();
+  }
+
+  async chat(request: ChatRequest): Promise<ChatResponse> {
+    return this.client.post<ChatResponse>('/chat', request);
+  }
+
+  async embed(request: EmbedRequest): Promise<EmbedResponse> {
+    return this.client.post<EmbedResponse>('/embed', request);
+  }
+
+  async rerank(request: RerankRequest): Promise<RerankResponse> {
+    return this.client.post<RerankResponse>('/rerank', request);
+  }
+
+  async classify(request: ClassifyRequest): Promise<ClassifyResponse> {
+    return this.client.post<ClassifyResponse>('/classify', request);
+  }
+
+  async listModels(): Promise<ModelsResponse> {
+    return this.client.get<ModelsResponse>('/models');
+  }
+
+  getClient(): CohereClient {
+    return this.client;
+  }
+}
+
+export { CohereClient } from './client';

package/connectors/connect-cohere/src/cli/index.ts

@@ -0,0 +1,255 @@
+#!/usr/bin/env bun
+import { Command } from 'commander';
+import chalk from 'chalk';
+import { Cohere } from '../api';
+import {
+  getApiKey,
+  setApiKey,
+  clearConfig,
+  getConfigDir,
+  setProfileOverride,
+  getCurrentProfile,
+  setCurrentProfile,
+  listProfiles,
+  createProfile,
+  deleteProfile,
+  profileExists,
+  loadProfile,
+} from '../utils/config';
+import type { OutputFormat } from '../utils/output';
+import { success, error, info, print } from '../utils/output';
+
+const CONNECTOR_NAME = 'connect-cohere';
+const VERSION = '0.0.1';
+
+const program = new Command();
+
+program
+  .name(CONNECTOR_NAME)
+  .description('Cohere AI connector CLI - Chat, embeddings, rerank, and classify')
+  .version(VERSION)
+  .option('-f, --format <format>', 'Output format (json, pretty)', 'pretty')
+  .option('-p, --profile <profile>', 'Use a specific profile')
+  .hook('preAction', (thisCommand) => {
+    const opts = thisCommand.opts();
+    if (opts.profile) {
+      if (!profileExists(opts.profile)) {
+        error(`Profile "${opts.profile}" does not exist`);
+        process.exit(1);
+      }
+      setProfileOverride(opts.profile);
+    }
+  });
+
+function getFormat(cmd: Command): OutputFormat {
+  const parent = cmd.parent;
+  return (parent?.opts().format || 'pretty') as OutputFormat;
+}
+
+function getClient(): Cohere {
+  const apiKey = getApiKey();
+  if (!apiKey) {
+    error(`No API key configured. Run "${CONNECTOR_NAME} config set-key <key>" or set COHERE_API_KEY`);
+    process.exit(1);
+  }
+  return new Cohere({ apiKey });
+}
+
+// Profile Commands
+const profileCmd = program.command('profile').description('Manage profiles');
+
+profileCmd.command('list').description('List profiles').action(() => {
+  const profiles = listProfiles();
+  const current = getCurrentProfile();
+  if (profiles.length === 0) {
+    info('No profiles found');
+    return;
+  }
+  profiles.forEach(p => {
+    const marker = p === current ? chalk.green(' (active)') : '';
+    console.log(`  ${p}${marker}`);
+  });
+});
+
+profileCmd.command('use <name>').description('Switch profile').action((name: string) => {
+  if (!profileExists(name)) {
+    error(`Profile "${name}" does not exist`);
+    process.exit(1);
+  }
+  setCurrentProfile(name);
+  success(`Switched to profile: ${name}`);
+});
+
+profileCmd.command('create <name>').description('Create profile')
+  .option('--api-key <key>', 'API key')
+  .option('--use', 'Switch to this profile')
+  .action((name: string, opts) => {
+    if (profileExists(name)) {
+      error(`Profile "${name}" already exists`);
+      process.exit(1);
+    }
+    createProfile(name, { apiKey: opts.apiKey });
+    success(`Profile "${name}" created`);
+    if (opts.use) {
+      setCurrentProfile(name);
+      info(`Switched to profile: ${name}`);
+    }
+  });
+
+profileCmd.command('delete <name>').description('Delete profile').action((name: string) => {
+  if (name === 'default') {
+    error('Cannot delete default profile');
+    process.exit(1);
+  }
+  if (deleteProfile(name)) {
+    success(`Profile "${name}" deleted`);
+  } else {
+    error(`Profile "${name}" not found`);
+    process.exit(1);
+  }
+});
+
+profileCmd.command('show [name]').description('Show profile').action((name?: string) => {
+  const profileName = name || getCurrentProfile();
+  const config = loadProfile(profileName);
+  console.log(chalk.bold(`Profile: ${profileName}`));
+  info(`API Key: ${config.apiKey ? config.apiKey.substring(0, 8) + '...' : chalk.gray('not set')}`);
+});
+
+// Config Commands
+const configCmd = program.command('config').description('Manage configuration');
+
+configCmd.command('set-key <apiKey>').description('Set API key').action((apiKey: string) => {
+  setApiKey(apiKey);
+  success(`API key saved`);
+});
+
+configCmd.command('show').description('Show config').action(() => {
+  console.log(chalk.bold(`Profile: ${getCurrentProfile()}`));
+  info(`Config dir: ${getConfigDir()}`);
+  const apiKey = getApiKey();
+  info(`API Key: ${apiKey ? apiKey.substring(0, 8) + '...' : chalk.gray('not set')}`);
+});
+
+configCmd.command('clear').description('Clear config').action(() => {
+  clearConfig();
+  success('Config cleared');
+});
+
+// Chat Command
+program.command('chat <message>')
+  .description('Send a chat message')
+  .option('-m, --model <model>', 'Model (default: command)')
+  .option('-t, --temperature <temp>', 'Temperature')
+  .option('--max-tokens <n>', 'Max tokens')
+  .action(async (message: string, opts) => {
+    try {
+      const client = getClient();
+      const result = await client.chat({
+        message,
+        model: opts.model,
+        temperature: opts.temperature ? parseFloat(opts.temperature) : undefined,
+        max_tokens: opts.maxTokens ? parseInt(opts.maxTokens) : undefined,
+      });
+      if (getFormat(program) === 'json') {
+        print(result, 'json');
+      } else {
+        console.log(chalk.green('\nResponse:'));
+        console.log(result.text);
+        if (result.meta?.billed_units) {
+          info(`\nTokens: ${result.meta.billed_units.input_tokens || 0} in, ${result.meta.billed_units.output_tokens || 0} out`);
+        }
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+// Embed Command
+program.command('embed <texts...>')
+  .description('Generate embeddings')
+  .option('-m, --model <model>', 'Model (default: embed-english-v3.0)')
+  .option('-t, --input-type <type>', 'Input type: search_document, search_query, classification, clustering')
+  .action(async (texts: string[], opts) => {
+    try {
+      const client = getClient();
+      const result = await client.embed({
+        texts,
+        model: opts.model,
+        input_type: opts.inputType,
+      });
+      if (getFormat(program) === 'json') {
+        print(result, 'json');
+      } else {
+        texts.forEach((text, i) => {
+          console.log(chalk.cyan(`\n[${i + 1}] "${text.substring(0, 40)}${text.length > 40 ? '...' : ''}"`));
+          console.log(`Dimension: ${result.embeddings[i]?.length || 0}`);
+        });
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+// Rerank Command
+program.command('rerank <query>')
+  .description('Rerank documents')
+  .option('-d, --documents <docs...>', 'Documents to rerank')
+  .option('-m, --model <model>', 'Model (default: rerank-english-v3.0)')
+  .option('-n, --top-n <n>', 'Top N results')
+  .action(async (query: string, opts) => {
+    try {
+      if (!opts.documents || opts.documents.length === 0) {
+        error('No documents provided. Use -d option.');
+        process.exit(1);
+      }
+      const client = getClient();
+      const result = await client.rerank({
+        query,
+        documents: opts.documents,
+        model: opts.model,
+        top_n: opts.topN ? parseInt(opts.topN) : undefined,
+        return_documents: true,
+      });
+      if (getFormat(program) === 'json') {
+        print(result, 'json');
+      } else {
+        console.log(chalk.cyan(`\nQuery: "${query}"\n`));
+        result.results.forEach((r, i) => {
+          console.log(`${chalk.yellow(`#${i + 1}`)} Score: ${chalk.green(r.relevance_score.toFixed(4))}`);
+          if (r.document) {
+            console.log(`   ${r.document.text.substring(0, 60)}${r.document.text.length > 60 ? '...' : ''}`);
+          }
+        });
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+// Models Command
+program.command('models')
+  .description('List available models')
+  .action(async () => {
+    try {
+      const client = getClient();
+      const result = await client.listModels();
+      if (getFormat(program) === 'json') {
+        print(result.models, 'json');
+      } else {
+        result.models.forEach(m => {
+          console.log(chalk.cyan(`\n${m.name}`));
+          if (m.endpoints) console.log(`  Endpoints: ${m.endpoints.join(', ')}`);
+          if (m.context_length) console.log(`  Context: ${m.context_length}`);
+        });
+      }
+    } catch (err) {
+      error(String(err));
+      process.exit(1);
+    }
+  });
+
+program.parse();

package/connectors/connect-cohere/src/index.ts

@@ -0,0 +1,19 @@
+// Cohere Connector
+// TypeScript wrapper for Cohere AI API
+
+export { Cohere } from './api';
+export * from './types';
+export { CohereClient } from './api';
+
+export {
+  getApiKey,
+  setApiKey,
+  getCurrentProfile,
+  setCurrentProfile,
+  listProfiles,
+  createProfile,
+  deleteProfile,
+  loadProfile,
+  saveProfile,
+  clearConfig,
+} from './utils/config';

package/connectors/connect-cohere/src/types/index.ts

@@ -0,0 +1,132 @@
+// Cohere Connector Types
+
+export interface CohereConfig {
+  apiKey: string;
+  baseUrl?: string;
+}
+
+export type OutputFormat = 'json' | 'pretty';
+
+// Chat Types
+export interface ChatMessage {
+  role: 'USER' | 'CHATBOT' | 'SYSTEM';
+  message: string;
+}
+
+export interface ChatRequest {
+  model?: string;
+  message: string;
+  chat_history?: ChatMessage[];
+  preamble?: string;
+  temperature?: number;
+  max_tokens?: number;
+  k?: number;
+  p?: number;
+  stop_sequences?: string[];
+}
+
+export interface ChatResponse {
+  text: string;
+  generation_id?: string;
+  chat_history?: ChatMessage[];
+  finish_reason?: string;
+  meta?: {
+    api_version?: { version: string };
+    billed_units?: { input_tokens?: number; output_tokens?: number };
+  };
+}
+
+// Embed Types
+export type EmbedInputType = 'search_document' | 'search_query' | 'classification' | 'clustering';
+
+export interface EmbedRequest {
+  model?: string;
+  texts: string[];
+  input_type?: EmbedInputType;
+  truncate?: 'NONE' | 'START' | 'END';
+}
+
+export interface EmbedResponse {
+  id: string;
+  embeddings: number[][];
+  texts: string[];
+  meta?: {
+    api_version?: { version: string };
+    billed_units?: { input_tokens?: number };
+  };
+}
+
+// Rerank Types
+export interface RerankRequest {
+  model?: string;
+  query: string;
+  documents: (string | { text: string })[];
+  top_n?: number;
+  return_documents?: boolean;
+}
+
+export interface RerankResult {
+  index: number;
+  relevance_score: number;
+  document?: { text: string };
+}
+
+export interface RerankResponse {
+  id: string;
+  results: RerankResult[];
+  meta?: {
+    api_version?: { version: string };
+    billed_units?: { search_units?: number };
+  };
+}
+
+// Classify Types
+export interface ClassifyExample {
+  text: string;
+  label: string;
+}
+
+export interface ClassifyRequest {
+  model?: string;
+  inputs: string[];
+  examples: ClassifyExample[];
+}
+
+export interface Classification {
+  id: string;
+  input: string;
+  prediction: string;
+  confidence: number;
+  labels: Record<string, { confidence: number }>;
+}
+
+export interface ClassifyResponse {
+  id: string;
+  classifications: Classification[];
+  meta?: {
+    api_version?: { version: string };
+  };
+}
+
+// Models Types
+export interface CohereModel {
+  name: string;
+  endpoints?: string[];
+  finetuned?: boolean;
+  context_length?: number;
+}
+
+export interface ModelsResponse {
+  models: CohereModel[];
+}
+
+// API Error
+export class CohereApiError extends Error {
+  public readonly statusCode: number;
+
+  constructor(message: string, statusCode: number) {
+    super(message);
+    this.name = 'CohereApiError';
+    this.statusCode = statusCode;
+  }
+}