jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: update Ollama components to use langchain-ollama integration (#4369)

Browse source 
* refactor: reorganize langchain dependencies in pyproject.toml

* feat: add langchain-ollama dependency to pyproject.toml

* fix: update imports to use langchain_ollama for OllamaEmbeddings and ChatOllama
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-11-04 10:00:15 -03:00 • committed by GitHub
commit f66139a800
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 427 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/components/embeddings/ollama.py
View file

@ -1,4 +1,4 @@
from langchain_community.embeddings import OllamaEmbeddings
from langchain_ollama import OllamaEmbeddings
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import Embeddings

2
src/backend/base/langflow/components/models/ollama.py
View file

@ -2,7 +2,7 @@ from typing import Any
from urllib.parse import urljoin
import httpx
from langchain_community.chat_models import ChatOllama
from langchain_ollama import ChatOllama
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel