Change to OllamaEmbeddings, still doesn't work in compose

Author: jexp

bot.Dockerfile

@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y \
 
 COPY requirements.txt .
 
-RUN pip install -r requirements.txt
+RUN pip install --upgrade -r requirements.txt
 
 # COPY .env .
 COPY bot.py .

bot.py

@@ -3,7 +3,7 @@
 import streamlit as st
 from langchain.vectorstores.neo4j_vector import Neo4jVector
 from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.embeddings import GPT4AllEmbeddings
+from langchain.embeddings import OllamaEmbeddings
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationalRetrievalChain
 from langchain.memory import ConversationBufferMemory
@@ -20,7 +20,7 @@
 os.environ["NEO4J_URL"] = url
 
 # embeddings = OpenAIEmbeddings()
-embeddings = GPT4AllEmbeddings()
+embeddings = OllamaEmbeddings()
 
 neo4j_db = Neo4jVector.from_existing_index(
     embedding=embeddings,

docker-compose.yml

@@ -54,7 +54,6 @@ services:
       - loader
 
     ports:
-      - 8080:8080
       - 8501:8501
 
 networks:

loader.Dockerfile

@@ -2,7 +2,7 @@ FROM python:latest
 
 COPY requirements.txt .
 
-RUN pip install -r requirements.txt
+RUN pip install --upgrade -r requirements.txt
 
 # COPY .env .
 COPY loader.py .

loader.py

@@ -6,7 +6,7 @@
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.docstore.document import Document
 from dotenv import load_dotenv
-from langchain.embeddings import GPT4AllEmbeddings
+from langchain.embeddings import OllamaEmbeddings
 
 load_dotenv('.env')
 
@@ -18,7 +18,7 @@
 os.environ["NEO4J_URL"] = url
 
 # embeddings = OpenAIEmbeddings()
-embeddings = GPT4AllEmbeddings()
+embeddings = OllamaEmbeddings()
 
 
 # Read the wikipedia article

requirements.txt

@@ -4,5 +4,4 @@ python-dotenv
 wikipedia
 tiktoken
 neo4j
-gpt4all
 streamlit