From ae7a2c202a701eb62409e3e79a4cbbf2634a92f1 Mon Sep 17 00:00:00 2001
From: Alex Yang <himself65@outlook.com>
Date: Wed, 7 Feb 2024 14:26:39 -0600
Subject: [PATCH] fix: add alias class `OllamaEmbedding` (#527)
---
packages/core/src/embeddings/OllamaEmbedding.ts | 7 +++++++
packages/core/src/embeddings/index.ts | 1 +
packages/core/src/llm/ollama.ts | 2 +-
3 files changed, 9 insertions(+), 1 deletion(-)
create mode 100644 packages/core/src/embeddings/OllamaEmbedding.ts
diff --git a/packages/core/src/embeddings/OllamaEmbedding.ts b/packages/core/src/embeddings/OllamaEmbedding.ts
new file mode 100644
index 000000000..896282c58
--- /dev/null
+++ b/packages/core/src/embeddings/OllamaEmbedding.ts
@@ -0,0 +1,7 @@
+import { Ollama } from "../llm/ollama";
+import { BaseEmbedding } from "./types";
+
+/**
+ * OllamaEmbedding is an alias for Ollama that implements the BaseEmbedding interface.
+ */
+export class OllamaEmbedding extends Ollama implements BaseEmbedding {}
diff --git a/packages/core/src/embeddings/index.ts b/packages/core/src/embeddings/index.ts
index 80a788f58..d106de31b 100644
--- a/packages/core/src/embeddings/index.ts
+++ b/packages/core/src/embeddings/index.ts
@@ -2,6 +2,7 @@ export * from "./ClipEmbedding";
export * from "./HuggingFaceEmbedding";
export * from "./MistralAIEmbedding";
export * from "./MultiModalEmbedding";
+export { OllamaEmbedding } from "./OllamaEmbedding";
export * from "./OpenAIEmbedding";
export { TogetherEmbedding } from "./together";
export * from "./types";
diff --git a/packages/core/src/llm/ollama.ts b/packages/core/src/llm/ollama.ts
index f1e9b3efc..4f2922908 100644
--- a/packages/core/src/llm/ollama.ts
+++ b/packages/core/src/llm/ollama.ts
@@ -1,5 +1,5 @@
import { CallbackManager, Event } from "../callbacks/CallbackManager";
-import { BaseEmbedding } from "../embeddings";
+import { BaseEmbedding } from "../embeddings/types";
import { ok } from "../env";
import {
ChatMessage,
--
GitLab