unrag 0.2.9 → 0.2.10

package/package.json

@@ -6,7 +6,7 @@
   "bin": {
     "unrag": "./dist/cli/index.js"
   },
-  "version": "0.2.9",
+  "version": "0.2.10",
   "private": false,
   "license": "Apache-2.0",
   "devDependencies": {

package/registry/embedding/azure.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface AzureModule {
   azure: {
-    embedding: (model: string) => EmbeddingModel<string>;
+    embedding: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/bedrock.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface BedrockModule {
   bedrock: {
-    embedding: (model: string) => EmbeddingModel<string>;
+    embedding: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/cohere.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface CohereModule {
   cohere: {
-    embedding: (model: string) => EmbeddingModel<string>;
+    embedding: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/google.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface GoogleModule {
   google: {
-    embedding: (model: string) => EmbeddingModel<string>;
+    embedding: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/mistral.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface MistralModule {
   mistral: {
-    embedding: (model: string) => EmbeddingModel<string>;
+    embedding: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/ollama.ts

@@ -6,7 +6,7 @@ import { requireOptional } from "./_shared";
  * Ollama provider instance interface.
  */
 interface OllamaProvider {
-  textEmbeddingModel: (model: string) => EmbeddingModel<string>;
+  textEmbeddingModel: (model: string) => EmbeddingModel;
 }
 
 /**

package/registry/embedding/openai.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface OpenAiModule {
   openai: {
-    embedding: (model: string) => EmbeddingModel<string>;
+    embedding: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/together.ts

@@ -7,8 +7,8 @@ import { requireOptional } from "./_shared";
  */
 interface TogetherAiModule {
   togetherai: {
-    embeddingModel?: (model: string) => EmbeddingModel<string>;
-    textEmbeddingModel?: (model: string) => EmbeddingModel<string>;
+    embeddingModel?: (model: string) => EmbeddingModel;
+    textEmbeddingModel?: (model: string) => EmbeddingModel;
   };
 }
 
@@ -37,6 +37,10 @@ export const createTogetherEmbeddingProvider = (
       ? togetherai.embeddingModel(model)
       : togetherai.textEmbeddingModel?.(model);
 
+  if (!embeddingModel) {
+    throw new Error("Together.ai embedding model function not available");
+  }
+
   return {
     name: `together:${model}`,
     dimensions: undefined,

package/registry/embedding/vertex.ts

@@ -7,7 +7,7 @@ import { requireOptional } from "./_shared";
  */
 interface VertexModule {
   vertex: {
-    embeddingModel: (model: string) => EmbeddingModel<string>;
+    embeddingModel: (model: string) => EmbeddingModel;
   };
 }
 

package/registry/embedding/voyage.ts

@@ -7,8 +7,8 @@ import { requireOptional } from "./_shared";
  */
 interface VoyageModule {
   voyage: {
-    embeddingModel?: (model: string) => EmbeddingModel<string>;
-    textEmbeddingModel?: (model: string) => EmbeddingModel<string>;
+    embeddingModel?: (model: string) => EmbeddingModel;
+    textEmbeddingModel?: (model: string) => EmbeddingModel;
     multimodalEmbeddingModel?: (model: string) => EmbeddingModel<unknown>;
   };
 }

package/registry/rerank/cohere.ts

@@ -27,8 +27,8 @@ interface RerankFunction {
     query: string;
   }): Promise<{
     ranking: Array<{
-      index: number;
-      relevanceScore?: number;
+      originalIndex: number;
+      score: number;
     }>;
   }>;
 }
@@ -104,9 +104,9 @@ export const createCohereReranker = (config?: CohereRerankerConfig): Reranker =>
       const scores: number[] = [];
 
       for (const item of result.ranking) {
-        order.push(item.index);
-        if (item.relevanceScore !== undefined) {
-          scores.push(item.relevanceScore);
+        order.push(item.originalIndex);
+        if (item.score !== undefined) {
+          scores.push(item.score);
         }
       }