From 7f5d7a5320d4cb283c7eb759e0463a10f731559a Mon Sep 17 00:00:00 2001
From: Vitalii Mostipaka <mostipaka.vitalii@gmail.com>
Date: Tue, 29 Apr 2025 09:40:32 +0300
Subject: [PATCH] feat: add new models to OLLAMA_EMBEDDING_MODELS and
 OLLAMA_TOOL_MODELS_BASE (#7586)

Co-authored-by: Edwin Jose <edwin.jose@datastax.com>
---
 src/backend/base/langflow/base/models/ollama_constants.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/backend/base/langflow/base/models/ollama_constants.py b/src/backend/base/langflow/base/models/ollama_constants.py
index 8620c7dfe..596479633 100644
--- a/src/backend/base/langflow/base/models/ollama_constants.py
+++ b/src/backend/base/langflow/base/models/ollama_constants.py
@@ -5,6 +5,7 @@ OLLAMA_EMBEDDING_MODELS = [
     "snowflake-arctic-embed",
     "all-minilm",
     "bge-m3",
+    "bge-large",
     "paraphrase-multilingual",
     "granite-embedding",
     "jina-embeddings-v2-base-en",
@@ -36,6 +37,7 @@ OLLAMA_TOOL_MODELS_BASE = [
     "aya-expanse",
     "granite3-moe",
     "firefunction-v2",
+    "cogito",
 ]