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

[fix] SearchProvider query fixed

This commit is contained in:
kimheesu 2025-07-01 13:06:08 +09:00
parent 52019d2005
commit a9efcc74c4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tradingagents/dataflows/search_provider.py
View File

@ -16,7 +16,7 @@ class GoogleSearchProvider(SearchProvider):
self.client = genai.Client() self.client = genai.Client()
self.model = model self.model = model
def search(self, query: str, ticker: str, curr_date: str) -> str: def search(self, query: str) -> str:
google_search_tool = Tool( google_search_tool = Tool(
google_search=GoogleSearch() google_search=GoogleSearch()
) )
@ -44,7 +44,7 @@ class OpenAISearchProvider(SearchProvider):
self.client = OpenAI(base_url=backend_url) self.client = OpenAI(base_url=backend_url)
self.model = model self.model = model
def search(self, query: str, ticker: str, curr_date: str) -> str: def search(self, query: str) -> str:
response = self.client.responses.create( response = self.client.responses.create(
model=self.model, model=self.model,
input=[ input=[