-
Notifications
You must be signed in to change notification settings - Fork 42
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update agent #3
base: main
Are you sure you want to change the base?
Update agent #3
Conversation
rlancemartin
commented
Nov 26, 2024
- Tavily async client + scraping
- Consolidate research into a single step
…ch into a single step
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
looks good! will make a new repo for companies agent
search_docs = await asyncio.gather(*search_tasks) | ||
|
||
# Deduplicate and format sources | ||
source_str = deduplicate_and_format_sources(search_docs, max_tokens_per_source=1000, include_raw_content=True) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
probably not needed for v0, but also would be great to propagate sources for each extracted datapoint
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
perhaps can be just included in the extraction schema...
) | ||
|
||
# Execute all searches concurrently | ||
search_docs = await asyncio.gather(*search_tasks) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nice!
query_list = [query.search_query for query in results.queries] | ||
search_tasks = [] | ||
for query in query_list: | ||
search_tasks.append( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
the only downside is that it's harder to switch to a different search tool that doesn't return raw content, but this is fine for now