Spaces:

mikail-bitik
/

CorrectiveRAGProject

Sleeping

CorrectiveRAGProject / graph /nodes /generate.py

First commit for deployment

7a4d005 10 months ago

450 Bytes

from graph.chains.generation import generation_chain
from graph.state import GraphState
from typing import Dict, Any


def generate(state: GraphState) -> Dict[str, Any]:
    print("----GENERATE----")
    question = state["question"]
    documents = state["documents"]

    generation = generation_chain.invoke(
        {"context": documents, "question":question}
    )

    return {"question":question, "documents":documents, "generation":generation}

	from graph.chains.generation import generation_chain
	from graph.state import GraphState
	from typing import Dict, Any


	def generate(state: GraphState) -> Dict[str, Any]:
	print("----GENERATE----")
	question = state["question"]
	documents = state["documents"]

	generation = generation_chain.invoke(
	{"context": documents, "question":question}
	)

	return {"question":question, "documents":documents, "generation":generation}