Spaces:

mikail-bitik
/

CorrectiveRAGProject

Sleeping

CorrectiveRAGProject / graph /nodes /retrieve.py

First commit for deployment

7a4d005 10 months ago

310 Bytes

from ingestion import retriever
from graph.state import GraphState
from typing import Dict, Any

def retrieve(state: GraphState) -> Dict[str, Any]:
    print("----RETRIEVE----")

    question = state["question"]
    documents = retriever.invoke(question)
    return {"question":question, "documents":documents}

	from ingestion import retriever
	from graph.state import GraphState
	from typing import Dict, Any

	def retrieve(state: GraphState) -> Dict[str, Any]:
	print("----RETRIEVE----")

	question = state["question"]
	documents = retriever.invoke(question)
	return {"question":question, "documents":documents}