charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: make tool naming llm provider agnostic

This commit is contained in:
kevin-bruton 2025-09-27 14:39:53 +02:00
parent aae3b17f9c
commit 0134792d19
8 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cli/main.py
View File

@ -465,10 +465,10 @@ def get_user_selections():
)
selected_research_depth = select_research_depth()
# Step 5: OpenAI backend
# Step 5: LLM Provider
console.print(
create_question_box(
"Step 5: OpenAI backend", "Select which service to talk to"
"Step 5: LLM Provider", "Select which service to talk to"
)
)
selected_llm_provider, backend_url = select_llm_provider()

2
tradingagents/agents/analysts/fundamentals_analyst.py
View File

@ -10,7 +10,7 @@ def create_fundamentals_analyst(llm, toolkit):
company_name = state["company_of_interest"]
if toolkit.config["online_tools"]:
tools = [toolkit.get_fundamentals_openai]
tools = [toolkit.get_fundamentals_from_llm]
else:
tools = [
toolkit.get_finnhub_company_insider_sentiment,

2
tradingagents/agents/analysts/news_analyst.py
View File

@ -9,7 +9,7 @@ def create_news_analyst(llm, toolkit):
ticker = state["company_of_interest"]
if toolkit.config["online_tools"]:
tools = [toolkit.get_global_news_openai, toolkit.get_google_news]
tools = [toolkit.get_global_news_from_llm, toolkit.get_google_news]
else:
tools = [
toolkit.get_finnhub_news,

2
tradingagents/agents/analysts/social_media_analyst.py
View File

@ -10,7 +10,7 @@ def create_social_media_analyst(llm, toolkit):
company_name = state["company_of_interest"]
if toolkit.config["online_tools"]:
tools = [toolkit.get_stock_news_openai]
tools = [toolkit.get_stock_news_from_llm]
else:
tools = [
toolkit.get_reddit_stock_info,

12
tradingagents/agents/utils/agent_utils.py
View File

@ -363,7 +363,7 @@ class Toolkit:
@staticmethod
@tool
def get_stock_news_openai(
def get_stock_news_from_llm(
ticker: Annotated[str, "the company's ticker"],
curr_date: Annotated[str, "Current date in yyyy-mm-dd format"],
):
@ -377,7 +377,7 @@ class Toolkit:
"""
try:
openai_news_results = interface.get_stock_news_openai(ticker, curr_date)
openai_news_results = interface.get_stock_news_from_llm(ticker, curr_date)
return openai_news_results
except ValueError as e:
# Return the detailed error message to the agent
@ -385,7 +385,7 @@ class Toolkit:
@staticmethod
@tool
def get_global_news_openai(
def get_global_news_from_llm(
curr_date: Annotated[str, "Current date in yyyy-mm-dd format"],
):
"""
@ -397,7 +397,7 @@ class Toolkit:
"""
try:
openai_news_results = interface.get_global_news_openai(curr_date)
openai_news_results = interface.get_global_news_from_llm(curr_date)
return openai_news_results
except ValueError as e:
# Return the detailed error message to the agent
@ -405,7 +405,7 @@ class Toolkit:
@staticmethod
@tool
def get_fundamentals_openai(
def get_fundamentals_from_llm(
ticker: Annotated[str, "the company's ticker"],
curr_date: Annotated[str, "Current date in yyyy-mm-dd format"],
):
@ -419,7 +419,7 @@ class Toolkit:
"""
try:
openai_fundamentals_results = interface.get_fundamentals_openai(
openai_fundamentals_results = interface.get_fundamentals_from_llm(
ticker, curr_date
)
return openai_fundamentals_results

2
tradingagents/agents/utils/memory.py
View File

@ -14,7 +14,7 @@ class FinancialSituationMemory:
# Use a good general-purpose model for financial text
self.embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
self.embedding_type = "local"
print(f"✅ Using local embeddings with sentence-transformers")
print(f"✅ Using local embeddings with sentence-transformers for {name}")
except ImportError:
print("⚠️ sentence-transformers not found. Install with: pip install sentence-transformers")
print("Falling back to ChromaDB's default embeddings...")

6
tradingagents/dataflows/interface.py
View File

@ -932,19 +932,19 @@ def _call_llm_api(prompt, config):
raise ValueError(error_msg) from e
def get_stock_news_openai(ticker, curr_date):
def get_stock_news_from_llm(ticker, curr_date):
config = get_config()
prompt = f"Can you search Social Media for {ticker} from 7 days before {curr_date} to {curr_date}? Make sure you only get the data posted during that period."
return _call_llm_api(prompt, config)
def get_global_news_openai(curr_date):
def get_global_news_from_llm(curr_date):
config = get_config()
prompt = f"Can you search global or macroeconomics news from 7 days before {curr_date} to {curr_date} that would be informative for trading purposes? Make sure you only get the data posted during that period."
return _call_llm_api(prompt, config)
def get_fundamentals_openai(ticker, curr_date):
def get_fundamentals_from_llm(ticker, curr_date):
config = get_config()
prompt = f"Can you search for fundamental analysis discussions on {ticker} during the month before {curr_date} to the month of {curr_date}. Make sure you only get the data posted during that period. List as a table, with PE/PS/Cash flow/ etc"
return _call_llm_api(prompt, config)

6
tradingagents/graph/trading_graph.py
View File

@ -153,7 +153,7 @@ class TradingAgentsGraph:
"social": ToolNode(
[
# online tools
self.toolkit.get_stock_news_openai,
self.toolkit.get_stock_news_from_llm,
# offline tools
self.toolkit.get_reddit_stock_info,
]
@ -161,7 +161,7 @@ class TradingAgentsGraph:
"news": ToolNode(
[
# online tools
self.toolkit.get_global_news_openai,
self.toolkit.get_global_news_from_llm,
self.toolkit.get_google_news,
# offline tools
self.toolkit.get_finnhub_news,
@ -171,7 +171,7 @@ class TradingAgentsGraph:
"fundamentals": ToolNode(
[
# online tools
self.toolkit.get_fundamentals_openai,
self.toolkit.get_fundamentals_from_llm,
# offline tools
self.toolkit.get_finnhub_company_insider_sentiment,
self.toolkit.get_finnhub_company_insider_transactions,