File size: 500 Bytes
6861e2a
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain_core.prompts import ChatPromptTemplate
from app.models.model import LLM
from app.managers import vector_manager as vm
from app.utils.prompts import summary_prompts

def get_summary_chain(language: str = "Deutsch"):
    system_prompt = summary_prompts[language]
    prompt = ChatPromptTemplate.from_messages([
        ("system", system_prompt),
        ("human", "{input}"),
    ])
    
    return LLM, prompt