65 lines
3.4 KiB
Python
65 lines
3.4 KiB
Python
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
|
|
import time
|
|
import json
|
|
from tradingagents.agents.utils.agent_utils import get_news
|
|
from tradingagents.agents.utils.data_prefetch import prefetch_social_data, format_social_context
|
|
from tradingagents.dataflows.config import get_config
|
|
|
|
|
|
def create_social_media_analyst(llm):
|
|
def social_media_analyst_node(state):
|
|
current_date = state["trade_date"]
|
|
ticker = state["company_of_interest"]
|
|
company_name = state["company_of_interest"]
|
|
|
|
# Pre-fetch data so it's available even if the backend doesn't support tool calling
|
|
prefetched = prefetch_social_data(ticker, current_date)
|
|
data_context = format_social_context(prefetched)
|
|
|
|
tools = [
|
|
get_news,
|
|
]
|
|
|
|
system_message = (
|
|
"You are a social media and company specific news researcher/analyst tasked with analyzing social media posts, recent company news, and public sentiment for a specific company over the past week. You will be given a company's name your objective is to write a comprehensive long report detailing your analysis, insights, and implications for traders and investors on this company's current state after looking at social media and what people are saying about that company, analyzing sentiment data of what people feel each day about the company, and looking at recent company news. Use the provided news data below for your analysis. If you have access to callable tools (get_news), you may use them for additional data. Try to look at all sources possible from social media to sentiment to news. Do not simply state the trends are mixed, provide detailed and finegrained analysis and insights that may help traders make decisions."
|
|
+ """ Make sure to append a Markdown table at the end of the report to organize key points in the report, organized and easy to read.""",
|
|
)
|
|
|
|
prompt = ChatPromptTemplate.from_messages(
|
|
[
|
|
(
|
|
"system",
|
|
"You are a helpful AI assistant, collaborating with other assistants."
|
|
" Use the provided data below to write your analysis report."
|
|
" If you have access to callable tools, you may use them for additional data."
|
|
" If you or any other assistant has the FINAL TRANSACTION PROPOSAL: **BUY/HOLD/SELL** or deliverable,"
|
|
" prefix your response with FINAL TRANSACTION PROPOSAL: **BUY/HOLD/SELL** so the team knows to stop."
|
|
"\n{system_message}"
|
|
"\nFor your reference, the current date is {current_date}. The current company we want to analyze is {ticker}"
|
|
"\n\n--- PRE-FETCHED SOCIAL/NEWS DATA ---\n{data_context}\n--- END PRE-FETCHED DATA ---",
|
|
),
|
|
MessagesPlaceholder(variable_name="messages"),
|
|
]
|
|
)
|
|
|
|
prompt = prompt.partial(system_message=system_message)
|
|
prompt = prompt.partial(current_date=current_date)
|
|
prompt = prompt.partial(ticker=ticker)
|
|
prompt = prompt.partial(data_context=data_context)
|
|
|
|
chain = prompt | llm.bind_tools(tools)
|
|
|
|
result = chain.invoke(state["messages"])
|
|
|
|
report = ""
|
|
|
|
if len(result.tool_calls) == 0:
|
|
report = result.content
|
|
|
|
return {
|
|
"messages": [result],
|
|
"sentiment_report": report,
|
|
}
|
|
|
|
return social_media_analyst_node
|