add embedding engine to telem

Mintplex-Labs · Dec 7, 2023 · 33de34f · 33de34f
1 parent ef79649
commit 33de34f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 0 deletions.
diff --git a/server/endpoints/api/workspace/index.js b/server/endpoints/api/workspace/index.js
@@ -65,6 +65,7 @@ function apiWorkspaceEndpoints(app) {
       await Telemetry.sendTelemetry("workspace_created", {
         multiUserMode: multiUserMode(response),
         LLMSelection: process.env.LLM_PROVIDER || "openai",
+        Embedder: process.env.EMBEDDING_ENGINE || "inherit",
         VectorDbSelection: process.env.VECTOR_DB || "pinecone",
       });
       response.status(200).json({ workspace, message });
@@ -488,6 +489,7 @@ function apiWorkspaceEndpoints(app) {
         const result = await chatWithWorkspace(workspace, message, mode);
         await Telemetry.sendTelemetry("sent_chat", {
           LLMSelection: process.env.LLM_PROVIDER || "openai",
+          Embedder: process.env.EMBEDDING_ENGINE || "inherit",
           VectorDbSelection: process.env.VECTOR_DB || "pinecone",
         });
         response.status(200).json({ ...result });

diff --git a/server/endpoints/chat.js b/server/endpoints/chat.js
@@ -77,6 +77,7 @@ function chatEndpoints(app) {
         await Telemetry.sendTelemetry("sent_chat", {
           multiUserMode: multiUserMode(response),
           LLMSelection: process.env.LLM_PROVIDER || "openai",
+          Embedder: process.env.EMBEDDING_ENGINE || "inherit",
           VectorDbSelection: process.env.VECTOR_DB || "pinecone",
         });
         response.end();
@@ -154,6 +155,7 @@ function chatEndpoints(app) {
           {
             multiUserMode: multiUserMode(response),
             LLMSelection: process.env.LLM_PROVIDER || "openai",
+            Embedder: process.env.EMBEDDING_ENGINE || "inherit",
             VectorDbSelection: process.env.VECTOR_DB || "pinecone",
           },
           user?.id

diff --git a/server/endpoints/workspaces.js b/server/endpoints/workspaces.js
@@ -32,6 +32,7 @@ function workspaceEndpoints(app) {
           {
             multiUserMode: multiUserMode(response),
             LLMSelection: process.env.LLM_PROVIDER || "openai",
+            Embedder: process.env.EMBEDDING_ENGINE || "inherit",
             VectorDbSelection: process.env.VECTOR_DB || "pinecone",
           },
           user?.id

diff --git a/server/models/documents.js b/server/models/documents.js
@@ -74,6 +74,7 @@ const Document = {
 
     await Telemetry.sendTelemetry("documents_embedded_in_workspace", {
       LLMSelection: process.env.LLM_PROVIDER || "openai",
+      Embedder: process.env.EMBEDDING_ENGINE || "inherit",
       VectorDbSelection: process.env.VECTOR_DB || "pinecone",
     });
     return { failed: failedToEmbed, embedded };
@@ -105,6 +106,7 @@ const Document = {
 
     await Telemetry.sendTelemetry("documents_removed_in_workspace", {
       LLMSelection: process.env.LLM_PROVIDER || "openai",
+      Embedder: process.env.EMBEDDING_ENGINE || "inherit",
       VectorDbSelection: process.env.VECTOR_DB || "pinecone",
     });
     return true;