@shenghuabi/llama 1.0.2 → 1.0.4

package/index.mjs

@@ -13,9 +13,6 @@ import { InjectionToken } from "static-injector";
 var LLamaConfigToken = new InjectionToken(
   "LLamaConfig"
 );
-var OLLAMA_MODEL_URL_TOKEN = new InjectionToken(
-  "OLLAMA_MODEL_URL"
-);
 
 // packages/llama/llama-swap.service.ts
 import * as v7 from "valibot";
@@ -2607,45 +2604,7 @@ async function readDir2(dir, level) {
 
 // packages/llama/llama-swap.service.ts
 import { dirname } from "path";
-
-// packages/llama/download/get-ollama-gguf-file.ts
-function getDownloadFileName(name) {
-  return name;
-}
-async function getOllamaGgufFile(name, options) {
-  name = name.includes("/") ? name : `library/${name}`;
-  const resultUrl = parseName(name);
-  const endpoint = resultUrl.endpoint ?? `https://${options.endpoint}/`;
-  const hff = `${resultUrl.namespace}/${resultUrl.repo}`;
-  const result = await getModelManifest(name, {
-    ...options,
-    endpoint
-  });
-  const rfilename = result.layers.find(
-    (item) => item.mediaType === `application/vnd.ollama.image.model`
-  ).digest;
-  const result2 = getDownloadFileName(rfilename);
-  const fileList = [`${endpoint}v2/${hff}/blobs/${result2}`];
-  return {
-    fileList,
-    fileName: `ollama-${name}`.replace(/\/|:/g, "-")
-  };
-}
-
-// packages/llama/llama-swap.service.ts
 import { differenceBy } from "es-toolkit";
-
-// packages/llama/download/auto-check-vendor.ts
-async function autoCheckVendor(repo, options) {
-  try {
-    await getModelManifest(repo, options);
-    return void 0;
-  } catch (error) {
-  }
-  return "ollama";
-}
-
-// packages/llama/llama-swap.service.ts
 var SysMap2 = { win32: "win", linux: "ubuntu" };
 var LlamaPlatform2 = SysMap2[process.platform];
 var remoteArgs = ["hf-repo", "hf-file", "model-url"];
@@ -2664,7 +2623,6 @@ var LlamaSwapService = class extends ExternalCallBaseService {
     return fullPath;
   });
   llamaSwapDir$$ = computed(() => path3.join(this.#config().dir, "llama-swap"));
-  #ollamaModelUrl$$ = inject(OLLAMA_MODEL_URL_TOKEN);
   #hgModelUrl$$ = inject(HUGGINGFACE_URL_TOKEN);
   #downloadConfig$$ = inject(DownloadConfigToken);
   #hgtoken$$ = inject(HUGGINGFACE_TOKEN_TOKEN);
@@ -2885,18 +2843,11 @@ var LlamaSwapService = class extends ExternalCallBaseService {
     let directory;
     const dir = path3.join(this.#config().dir, "models");
     if (modelOptions.repo) {
-      if (modelOptions.vendor) {
-        this.log?.info("使用厂商:", modelOptions.vendor);
-      }
-      const resolveResult = await (modelOptions.vendor === "ollama" ? getOllamaGgufFile(modelOptions.repo, {
-        ...modelOptions,
-        endpoint: this.#ollamaModelUrl$$(),
-        token: modelOptions.token || this.#hgtoken$$()
-      }) : getGgufFile(modelOptions.repo, {
+      const resolveResult = await getGgufFile(modelOptions.repo, {
         ...modelOptions,
         endpoint: this.#hgModelUrl$$(),
         token: modelOptions.token || this.#hgtoken$$()
-      }));
+      });
       this.log?.info("模型文件:", resolveResult);
       const filePath = path3.join(dir, resolveResult.fileName);
       if (fs2.existsSync(filePath)) {
@@ -2950,14 +2901,9 @@ var LlamaSwapService = class extends ExternalCallBaseService {
     );
   }
   async createModelConfig(model, options) {
-    const vendor = await autoCheckVendor(model, {
-      endpoint: `https://${this.#hgModelUrl$$()}/`,
-      token: this.#hgtoken$$()
-    });
     const downloadFilePath = await this.downloadModel(
       {
-        repo: model,
-        vendor
+        repo: model
       },
       options
     );
@@ -3029,7 +2975,6 @@ export {
   LlamaServerDefine,
   LlamaServerSwapItemDefine,
   LlamaSwapService,
-  OLLAMA_MODEL_URL_TOKEN,
   SamplingDefine
 };
 //# sourceMappingURL=index.mjs.map