michel/Philosophy-RAG-demo
1
0
Fork 0
forked from AI_team/Philosophy-RAG-demo
Code Pull Requests Activity

Add some newlines for read ability

Browse Source
This commit is contained in:
Nielson Janné 2025-03-15 13:42:28 +01:00
parent 976ccd155f
commit abc4ac947c
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
generic_rag/app.py
View File

@ -65,7 +65,9 @@ class State(TypedDict):
def retrieve(state: State): def retrieve(state: State):
vector_store = cl.user_session.get("vector_store") vector_store = cl.user_session.get("vector_store")
retrieved_docs = vector_store.similarity_search(state["question"]) retrieved_docs = vector_store.similarity_search(state["question"])
return {"context": retrieved_docs} return {"context": retrieved_docs}
@ -76,6 +78,7 @@ def generate(state: State):
docs_content = "\n\n".join(doc.page_content for doc in state["context"]) docs_content = "\n\n".join(doc.page_content for doc in state["context"])
messages = prompt.invoke({"question": state["question"], "context": docs_content}) messages = prompt.invoke({"question": state["question"], "context": docs_content})
response = llm.invoke(messages) response = llm.invoke(messages)
return {"answer": response.content} return {"answer": response.content}