2024-01-02 20:32:16 +00:00
from langchain_community . chat_models import ChatOpenAI
2024-01-02 20:09:45 +00:00
from langchain_core . output_parsers import StrOutputParser
2024-01-03 21:28:05 +00:00
from langchain_core . prompts import ChatPromptTemplate
2024-01-02 20:09:45 +00:00
from langchain_core . runnables import ConfigurableField
2023-12-18 22:00:18 +00:00
WRITER_SYSTEM_PROMPT = " You are an AI critical thinker research assistant. Your sole purpose is to write well written, critically acclaimed, objective and structured reports on given text. " # noqa: E501
# Report prompts from https://github.com/assafelovic/gpt-researcher/blob/master/gpt_researcher/master/prompts.py
RESEARCH_REPORT_TEMPLATE = """ Information:
- - - - - - - -
{ research_summary }
- - - - - - - -
Using the above information , answer the following question or topic : " {question} " in a detailed report - - \
The report should focus on the answer to the question , should be well structured , informative , \
in depth , with facts and numbers if available and a minimum of 1 , 200 words .
You should strive to write the report as long as you can using all relevant and necessary information provided .
You must write the report with markdown syntax .
You MUST determine your own concrete and valid opinion based on the given information . Do NOT deter to general and meaningless conclusions .
Write all used source urls at the end of the report , and make sure to not add duplicated sources , but only one reference for each .
You must write the report in apa format .
Please do your best , this is very important to my career . """ # noqa: E501
RESOURCE_REPORT_TEMPLATE = """ Information:
- - - - - - - -
{ research_summary }
- - - - - - - -
Based on the above information , generate a bibliography recommendation report for the following question or topic : " {question} " . \
The report should provide a detailed analysis of each recommended resource , explaining how each source can contribute to finding answers to the research question . \
Focus on the relevance , reliability , and significance of each source . \
Ensure that the report is well - structured , informative , in - depth , and follows Markdown syntax . \
Include relevant facts , figures , and numbers whenever available . \
The report should have a minimum length of 1 , 200 words .
Please do your best , this is very important to my career . """ # noqa: E501
OUTLINE_REPORT_TEMPLATE = """ Information:
- - - - - - - -
{ research_summary }
- - - - - - - -
Using the above information , generate an outline for a research report in Markdown syntax for the following question or topic : " {question} " . \
The outline should provide a well - structured framework for the research report , including the main sections , subsections , and key points to be covered . \
The research report should be detailed , informative , in - depth , and a minimum of 1 , 200 words . \
Use appropriate Markdown syntax to format the outline and ensure readability .
Please do your best , this is very important to my career . """ # noqa: E501
model = ChatOpenAI ( temperature = 0 )
prompt = ChatPromptTemplate . from_messages (
[
( " system " , WRITER_SYSTEM_PROMPT ) ,
( " user " , RESEARCH_REPORT_TEMPLATE ) ,
]
) . configurable_alternatives (
ConfigurableField ( " report_type " ) ,
default_key = " research_report " ,
resource_report = ChatPromptTemplate . from_messages (
[
( " system " , WRITER_SYSTEM_PROMPT ) ,
( " user " , RESOURCE_REPORT_TEMPLATE ) ,
]
) ,
outline_report = ChatPromptTemplate . from_messages (
[
( " system " , WRITER_SYSTEM_PROMPT ) ,
( " user " , OUTLINE_REPORT_TEMPLATE ) ,
]
) ,
)
chain = prompt | model | StrOutputParser ( )