alexkueck commited on
Commit
9b64feb
1 Parent(s): 85430d3

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -2
utils.py CHANGED
@@ -28,7 +28,8 @@ from pygments import highlight
28
  from pygments.lexers import guess_lexer,get_lexer_by_name
29
  from pygments.formatters import HtmlFormatter
30
 
31
- from langchain.chains import LLMChain, RetrievalQA, RunnableSequence
 
32
  from langchain.prompts import PromptTemplate
33
  from langchain_community.document_loaders import PyPDFLoader, UnstructuredWordDocumentLoader, DirectoryLoader
34
  #from langchain.document_loaders.blob_loaders.youtube_audio import YoutubeAudioLoader
@@ -316,7 +317,7 @@ def rag_chain(llm, prompt, retriever):
316
  # Erstelle eine RunnableSequence
317
  chain = RunnableSequence(steps=[RAG_CHAIN_PROMPT, llm])
318
  # Verwende die Kette
319
- result = chain.invoke({"context": relevant_docs, "question": prompt})
320
  else:
321
  # keine relevanten Dokumente gefunden
322
  result = "Keine relevanten Dokumente gefunden"
 
28
  from pygments.lexers import guess_lexer,get_lexer_by_name
29
  from pygments.formatters import HtmlFormatter
30
 
31
+ from langchain.chains import LLMChain, RetrievalQA
32
+ from langchain import RunnableSequence
33
  from langchain.prompts import PromptTemplate
34
  from langchain_community.document_loaders import PyPDFLoader, UnstructuredWordDocumentLoader, DirectoryLoader
35
  #from langchain.document_loaders.blob_loaders.youtube_audio import YoutubeAudioLoader
 
317
  # Erstelle eine RunnableSequence
318
  chain = RunnableSequence(steps=[RAG_CHAIN_PROMPT, llm])
319
  # Verwende die Kette
320
+ result = chain.invoke({"context": relevant_docs, "question": prompt})
321
  else:
322
  # keine relevanten Dokumente gefunden
323
  result = "Keine relevanten Dokumente gefunden"