Update utils.py

utils.py

@@ -31,22 +31,23 @@ from pygments.formatters import HtmlFormatter
 
 from langchain.chains import LLMChain, RetrievalQA
 from langgraph.graph import END, StateGraph
-from langchain.chat_models import ChatOpenAI
-from langchain.document_loaders import PyPDFLoader, WebBaseLoader, UnstructuredWordDocumentLoader, DirectoryLoader
+from langchain_community.chat_models import ChatOpenAI
+from langchain_community.document_loaders import PyPDFLoader, , WebBaseLoader, UnstructuredWordDocumentLoader, DirectoryLoader
 from langchain.document_loaders.blob_loaders.youtube_audio import YoutubeAudioLoader
 from langchain.document_loaders.generic import GenericLoader
 from langchain.document_loaders.parsers import OpenAIWhisperParser
 from langchain.schema import AIMessage, HumanMessage
-from langchain.llms import HuggingFaceHub
-from langchain.llms import HuggingFaceTextGenInference
-from langchain.embeddings import HuggingFaceInstructEmbeddings, HuggingFaceEmbeddings, HuggingFaceBgeEmbeddings, HuggingFaceInferenceAPIEmbeddings
-from langchain.tools import DuckDuckGoSearchRun
+from langchain_community.llms import HuggingFaceHub
+from langchain_community.llms import HuggingFaceTextGenInference
+from langchain_community.embeddings import HuggingFaceInstructEmbeddings, HuggingFaceEmbeddings, HuggingFaceBgeEmbeddings, HuggingFaceInferenceAPIEmbeddings
+from langchain_community.tools import DuckDuckGoSearchRun
 from langchain.retrievers.tavily_search_api import TavilySearchAPIRetriever
 from typing import Dict, TypedDict
 from langchain_core.messages import BaseMessage
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.prompts import PromptTemplate
 
+
 from langchain import hub
 from langchain.output_parsers.openai_tools import PydanticToolsParser
 from langchain.prompts import PromptTemplate
@@ -59,7 +60,7 @@ from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.runnables import RunnablePassthrough
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.vectorstores import Chroma
+from langchain_community.vectorstores import Chroma
 from chromadb.errors import InvalidDimensionException
 import io
 from PIL import Image, ImageDraw, ImageOps, ImageFont
@@ -398,7 +399,7 @@ def rag_chain(llm, prompt,  retriever):
 
     if (len(filtered_docs)>0):
         llm_chain = LLMChain(llm = llm, prompt = RAG_CHAIN_PROMPT)
-        result = llm_chain.run({"context": documents, "question": neu_prompt})
+        result = llm_chain.run({"context": filtered_docs, "question": neu_prompt})
     else:
         #Normale Abfrage, da keine relevanten Dokumente gefunden
         llm_chain = LLMChain(llm = llm, prompt = LLM_CHAIN_PROMPT)