File size: 1,171 Bytes
30b0af8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import logging
from llms import llm
# from llms.openai import llm
from prompts import QA_PROMPT
from prompts import SUMMARIZATION_PROMPT
logger = logging.getLogger(__name__)
def get_qa_chain():
logger.info("Creating QA chain")
print("> Creating QA chain")
try:
# qa_chain = QA_PROMPT
qa_chain = QA_PROMPT | llm
logger.info("QA chain created")
print("> QA chain created")
return qa_chain
except Exception as e:
msg = f"Error: {e}"
logger.exception(msg)
print(msg)
raise e
def get_summarization_chain():
logger.info("Creating summarization chain")
print("> Creating summarization chain")
try:
# summarization_chain = SUMMARIZATION_PROMPT
summarization_chain = SUMMARIZATION_PROMPT | llm
logger.info("Summarization chain created")
print("> Summarization chain created")
return summarization_chain
except Exception as e:
msg = f"Error: {e}"
logger.exception(msg)
print(msg)
raise e
qa_chain = get_qa_chain()
summarization_chain = get_summarization_chain()
|