symmetry-cli 1.0.7 → 1.0.11

package/dist/session-repository.js

@@ -1,127 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.SessionRepository = void 0;
-const database_1 = require("./database");
-class SessionRepository {
-    constructor() {
-        this.create = (session) => {
-            return new Promise((resolve, reject) => {
-                const sql = `INSERT INTO sessions (id, provider_id, created_at, expires_at) VALUES (?, ?, ?, ?)`;
-                this.db.run(sql, [
-                    session.id,
-                    session.providerId,
-                    session.createdAt.toISOString(),
-                    session.expiresAt.toISOString(),
-                ], function (err) {
-                    if (err) {
-                        reject(err);
-                    }
-                    else {
-                        resolve();
-                    }
-                });
-            });
-        };
-        this.get = (id) => {
-            return new Promise((resolve, reject) => {
-                const sql = `
-        SELECT s.id, s.provider_id as providerId, s.created_at as createdAt, s.expires_at as expiresAt
-        FROM sessions s
-        WHERE s.id = ?
-      `;
-                this.db.get(sql, [id], (err, row) => {
-                    if (err) {
-                        reject(err);
-                    }
-                    else if (row) {
-                        resolve({
-                            id: row.id,
-                            providerId: row.providerId,
-                            createdAt: new Date(row.createdAt),
-                            expiresAt: new Date(row.expiresAt),
-                        });
-                    }
-                    else {
-                        resolve(null);
-                    }
-                });
-            });
-        };
-        this.update = (session) => {
-            return new Promise((resolve, reject) => {
-                const sql = `UPDATE sessions SET provider_id = ?, created_at = ?, expires_at = ? WHERE id = ?`;
-                this.db.run(sql, [
-                    session.providerId,
-                    session.createdAt.toISOString(),
-                    session.expiresAt.toISOString(),
-                    session.id,
-                ], function (err) {
-                    if (err) {
-                        reject(err);
-                    }
-                    else if (this.changes === 0) {
-                        reject(new Error("No session found with the provided id"));
-                    }
-                    else {
-                        resolve();
-                    }
-                });
-            });
-        };
-        this.delete = (id) => {
-            return new Promise((resolve, reject) => {
-                const sql = `DELETE FROM sessions WHERE id = ?`;
-                this.db.run(sql, [id], function (err) {
-                    if (err) {
-                        reject(err);
-                    }
-                    else {
-                        resolve(this.changes > 0);
-                    }
-                });
-            });
-        };
-        this.getAllActiveSessions = () => {
-            return new Promise((resolve, reject) => {
-                const sql = `
-        SELECT s.id, s.provider_id as providerId, s.created_at as createdAt, s.expires_at as expiresAt,
-               p.key as peer_key, p.discovery_key, p.model_name
-        FROM sessions s
-        LEFT JOIN peers p ON s.provider_id = p.id
-        WHERE s.expires_at > datetime('now')
-      `;
-                this.db.all(sql, [], (err, rows) => {
-                    if (err) {
-                        reject(err);
-                    }
-                    else {
-                        resolve(rows.map((row) => ({
-                            id: row.id,
-                            providerId: row.providerId,
-                            createdAt: new Date(row.createdAt),
-                            expiresAt: new Date(row.expiresAt),
-                            peer_key: row.peer_key,
-                            discovery_key: row.discovery_key,
-                            model_name: row.model_name,
-                        })));
-                    }
-                });
-            });
-        };
-        this.deleteExpired = () => {
-            return new Promise((resolve, reject) => {
-                const sql = `DELETE FROM sessions WHERE expires_at <= datetime('now')`;
-                this.db.run(sql, function (err) {
-                    if (err) {
-                        reject(err);
-                    }
-                    else {
-                        resolve(this.changes);
-                    }
-                });
-            });
-        };
-        this.db = database_1.database;
-    }
-}
-exports.SessionRepository = SessionRepository;

package/dist/types.js

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/dist/utils.js

@@ -1,53 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getChatDataFromProvider = void 0;
-exports.safeParseJson = safeParseJson;
-exports.createMessage = createMessage;
-exports.isStreamWithDataPrefix = isStreamWithDataPrefix;
-exports.safeParseStreamResponse = safeParseStreamResponse;
-const constants_1 = require("./constants");
-function safeParseJson(data) {
-    try {
-        return JSON.parse(data);
-    }
-    catch (e) {
-        return undefined;
-    }
-}
-function createMessage(key, data) {
-    return JSON.stringify({ key, data });
-}
-function isStreamWithDataPrefix(stringBuffer) {
-    return stringBuffer.startsWith('data:');
-}
-function safeParseStreamResponse(stringBuffer) {
-    try {
-        if (isStreamWithDataPrefix(stringBuffer)) {
-            return JSON.parse(stringBuffer.split('data:')[1]);
-        }
-        return JSON.parse(stringBuffer);
-    }
-    catch (e) {
-        return undefined;
-    }
-}
-const getChatDataFromProvider = (provider, data) => {
-    var _a, _b;
-    switch (provider) {
-        case constants_1.apiProviders.Ollama:
-        case constants_1.apiProviders.OpenWebUI:
-            return ((_a = data === null || data === void 0 ? void 0 : data.choices[0].delta) === null || _a === void 0 ? void 0 : _a.content)
-                ? data === null || data === void 0 ? void 0 : data.choices[0].delta.content
-                : '';
-        case constants_1.apiProviders.LlamaCpp:
-            return data === null || data === void 0 ? void 0 : data.content;
-        case constants_1.apiProviders.LiteLLM:
-        default:
-            if ((data === null || data === void 0 ? void 0 : data.choices[0].delta.content) === 'undefined')
-                return '';
-            return ((_b = data === null || data === void 0 ? void 0 : data.choices[0].delta) === null || _b === void 0 ? void 0 : _b.content)
-                ? data === null || data === void 0 ? void 0 : data.choices[0].delta.content
-                : '';
-    }
-};
-exports.getChatDataFromProvider = getChatDataFromProvider;

package/dist/websocket-server.js

@@ -1,52 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.WsServer = void 0;
-const http_1 = require("http");
-const ws_1 = require("ws");
-class WsServer {
-    constructor(port, peerRepository, swarm) {
-        this.wssListeners = () => {
-            var _a;
-            (_a = this._wss) === null || _a === void 0 ? void 0 : _a.on("connection", (ws) => {
-                this.sendStats(ws);
-                const interval = setInterval(() => this.sendStats(ws), 5000);
-                ws.on("close", () => {
-                    clearInterval(interval);
-                });
-            });
-        };
-        this._peerRepository = peerRepository;
-        this._swarm = swarm;
-        this.start(port);
-    }
-    start(port) {
-        const server = (0, http_1.createServer)();
-        this._wss = new ws_1.WebSocketServer({ server });
-        server.listen(port);
-        this.wssListeners();
-    }
-    async sendStats(ws) {
-        const stats = await this.getStats(this._swarm);
-        ws.send(JSON.stringify(stats));
-    }
-    async broadcastStats() {
-        var _a;
-        const stats = await this.getStats(this._swarm);
-        (_a = this._wss) === null || _a === void 0 ? void 0 : _a.clients.forEach((client) => {
-            if (client.readyState === ws_1.WebSocket.OPEN) {
-                client.send(JSON.stringify(stats));
-            }
-        });
-    }
-    async getStats(swarm) {
-        const activePeers = await this._peerRepository.getActivePeerCount();
-        const activeModels = await this._peerRepository.getActiveModelCount();
-        const swarmConnections = swarm.connections.size;
-        return {
-            activePeers,
-            activeModels,
-            swarmConnections,
-        };
-    }
-}
-exports.WsServer = WsServer;

package/src/config.ts

@@ -1,51 +0,0 @@
-import { ProviderConfig  } from "./types";
-import fs from "fs";
-import yaml from "js-yaml";
-
-export class ConfigManager {
-  private config: ProviderConfig;
-
-  constructor(configPath: string) {
-    const configFile = fs.readFileSync(configPath, "utf8");
-    const config = yaml.load(configFile) as ProviderConfig;
-    this.config = config
-    this.validate();
-  }
-
-  public getAll () {
-    return this.config;
-  }
-
-  private validate(): void {
-    const requiredFields: (keyof ProviderConfig)[] = [
-      "apiHostname",
-      "apiPath",
-      "apiPort",
-      "apiProtocol",
-      "apiProvider",
-      "modelName",
-      "path",
-      "public",
-      "serverKey",
-    ];
-
-    for (const field of requiredFields) {
-      if (!(field in this.config)) {
-        throw new Error(
-          `Missing required field in client configuration: ${field}`
-        );
-      }
-    }
-
-    if (typeof this.config.public !== "boolean") {
-      throw new Error(
-        'The "public" field in client configuration must be a boolean'
-      );
-    }
-  }
-
-  get<K extends keyof ProviderConfig>(key: K): ProviderConfig[K];
-  get(key: string): unknown {
-    return this.config[key as keyof ProviderConfig];
-  }
-}

package/src/constants.ts

@@ -1,31 +0,0 @@
-export const NORMALIZE_REGEX = /\s*\r?\n|\r/g;
-
-export const PROVIDER_HELLO_TIMEOUT = 30000
-
-export const serverMessageKeys = {
-  challenge: "challenge",
-  conectionSize: "conectionSize",
-  heartbeat: "heartbeat",
-  inference: "inference",
-  inferenceEnded: "inferenceEnded",
-  join: "join",
-  joinAck: "joinAck",
-  leave: "leave",
-  newConversation: "newConversation",
-  ping: "ping",
-  pong: "pong",
-  providerDetails: "providerDetails",
-  reportCompletion: "reportCompletion",
-  requestProvider: "requestProvider",
-  sessionValid: "sessionValid",
-  verifySession: "verifySession",
-} as const;
-
-export const apiProviders = {
-  LiteLLM: 'litellm',
-  LlamaCpp: 'llamacpp',
-  LMStudio: 'lmstudio',
-  Ollama: 'ollama',
-  Oobabooga: 'oobabooga',
-  OpenWebUI: 'openwebui',
-} as const

package/src/logger.ts

@@ -1,47 +0,0 @@
-/* eslint-disable @typescript-eslint/no-explicit-any */
-import chalk from "chalk";
-
-export enum LogLevel {
-  DEBUG,
-  ERROR,
-  INFO,
-  WARNING,
-}
-
-export class Logger {
-  private static instance: Logger;
-  private logLevel: LogLevel = LogLevel.INFO;
-
-  private constructor() {}
-
-  public static getInstance(): Logger {
-    if (!Logger.instance) {
-      Logger.instance = new Logger();
-    }
-    return Logger.instance;
-  }
-
-  public setLogLevel(level: LogLevel): void {
-    this.logLevel = level;
-  }
-
-  public info(message: string, ...args: any[]): void {
-    if (this.logLevel <= LogLevel.INFO) {
-      console.log(chalk.blue("ℹ️ INFO:"), message, ...args);
-    }
-  }
-
-  public warning(message: string, ...args: any[]): void {
-    console.log(chalk.yellow("⚠️ WARNING:"), message, ...args);
-  }
-
-  public error(message: string, ...args: any[]): void {
-    console.error(chalk.red("❌ ERROR:"), message, ...args);
-  }
-
-  public debug(message: string, ...args: any[]): void {
-    console.log(chalk.gray("🐛 DEBUG:"), message, ...args);
-  }
-}
-
-export const logger = Logger.getInstance();