get rid of cohere completly

Author: yassinsws

genAi/llm.py

@@ -5,13 +5,7 @@
 
 from dotenv import load_dotenv
 import os
-
-from pydantic import BaseModel
-
 from chains import FlashcardChain, QuizChain
-from response_models import FlashcardResponse
-from request_models import SummaryRequest
-
 from rag import RAGHelper
 
 load_dotenv()

genAi/rag.py

@@ -3,7 +3,6 @@
 from langchain_community.document_loaders import PyMuPDFLoader, TextLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain_weaviate.vectorstores import WeaviateVectorStore
-from langchain_cohere import CohereEmbeddings
 from langchain_core.documents import Document
 from dotenv import load_dotenv
 import os
@@ -14,7 +13,6 @@
 # Setup shared embeddings model
 load_dotenv()
 embeddings_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
-# embeddings_model_cohere = CohereEmbeddings(model="embed-english-light-v3.0", cohere_api_key=os.getenv("COHERE_API_KEY"))
 
 # Disable Huggingface's tokenizer parallelism (avoid deadlocks caused by process forking in langchain)
 os.environ["TOKENIZERS_PARALLELISM"] = "false"

genAi/requirements.txt

@@ -10,8 +10,6 @@ langchain-openai
 langchain-weaviate
 langchain-community
 langchain-text-splitters
-cohere==5.12.0
-langchain-cohere==0.4.3
 pymupdf # for the PDF loader
 langchain_huggingface
 sentence-transformers