@flotorch/loadtest 0.1.2 → 0.2.0

package/dist/index.js

@@ -382,7 +382,8 @@ init_esm_shims();
 init_esm_shims();
 import { parseArgs } from "util";
 import { readFileSync, writeFileSync, existsSync, mkdirSync } from "fs";
-import { join } from "path";
+import { join, dirname } from "path";
+import { fileURLToPath as fileURLToPath2 } from "url";
 
 // src/schemas/config.zod.ts
 init_esm_shims();
@@ -455,9 +456,11 @@ var cyan = wrap("36", "39");
 var magenta = wrap("35", "39");
 
 // src/cli/args.ts
+var __dirname2 = dirname(fileURLToPath2(import.meta.url));
+var pkg = JSON.parse(readFileSync(join(__dirname2, "../../package.json"), "utf-8"));
 var VALID_COMMANDS = /* @__PURE__ */ new Set(["run", "generate", "bench", "report", "init"]);
 var HELP_TEXT = `
-${bold("FLOTorch Load Tester")}
+${bold("FLOTorch Load Tester")} ${dim(`v${pkg.version}`)}
 
 ${yellow("USAGE")}
   flotorch ${dim("<command>")} ${dim("[options]")}
@@ -480,6 +483,7 @@ ${yellow("OPTIONS")}
   ${cyan("--base-url")} ${dim("<url>")}      Override provider.baseURL
   ${cyan("--streaming")}           Enable streaming
   ${cyan("--no-streaming")}        Disable streaming
+  ${cyan("-v, --version")}         Show version number
   ${cyan("-h, --help")}            Show this help message
 `.trimStart();
 function parseCliArgs(argv) {
@@ -501,10 +505,15 @@ function parseCliArgs(argv) {
       "base-url": { type: "string" },
       streaming: { type: "boolean" },
       "no-streaming": { type: "boolean" },
+      version: { type: "boolean", short: "v" },
       help: { type: "boolean", short: "h" }
     },
     allowPositionals: true
   });
+  if (values.version) {
+    console.log(pkg.version);
+    process.exit(0);
+  }
   if (values.help) {
     console.log(HELP_TEXT);
     process.exit(0);
@@ -621,8 +630,8 @@ async function runInit(outputPath) {
   }
   let baseURL;
   if (adapter === "openai") {
-    const url = await prompt(rl, "Base URL", "https://api.openai.com/v1");
-    if (url !== "https://api.openai.com/v1") {
+    const url = await prompt(rl, "Base URL", "https://api.openai.com/v1/chat/completions");
+    if (url !== "https://api.openai.com/v1/chat/completions") {
       baseURL = url;
     }
   }
@@ -861,11 +870,11 @@ var OpenAIBackend = class _OpenAIBackend {
   apiKey;
   static create(baseURL) {
     const env = validateEnv(EnvSchema, "openai");
-    const url = baseURL ?? "https://api.openai.com/v1";
+    const url = baseURL ?? "https://api.openai.com/v1/chat/completions";
     return new _OpenAIBackend(url, env.OPENAI_API_KEY);
   }
   constructor(baseURL, apiKey) {
-    this.url = baseURL.endsWith("/chat/completions") ? baseURL : `${baseURL.replace(/\/+$/, "")}/chat/completions`;
+    this.url = baseURL;
     this.apiKey = apiKey;
   }
   async request(prompt2, model, maxTokens, systemPrompt, params, streaming, signal) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flotorch/loadtest",
-  "version": "0.1.2",
+  "version": "0.2.0",
   "description": "LLM inference load testing and benchmarking tool",
   "license": "MIT",
   "repository": {