kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

LinkedIn AI Writer features like carousel, video script, and comment response generator

Browse Source
This commit is contained in:
ajaysi
2025-04-14 22:34:10 +05:30
parent dd9a9e5f09
commit d4f1fc77a1
49 changed files with 7636 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/ai_web_researcher/gpt_online_researcher.py
View File

@@ -38,7 +38,7 @@ from lib.alwrity_ui.display_google_serp_results import (
)
from lib.alwrity_ui.google_trends_ui import display_google_trends_data, process_trends_data
from .tavily_ai_search import get_tavilyai_results
from .tavily_ai_search import do_tavily_ai_search
from .metaphor_basic_neural_web_search import metaphor_search_articles, streamlit_display_metaphor_results
from .google_serp_search import google_search
from .google_trends_researcher import do_google_trends_analysis
@@ -148,8 +148,8 @@ def gpt_web_researcher(search_keywords, search_mode, **kwargs):
include_domains = kwargs.pop('include_domains', None)
search_depth = kwargs.pop('search_depth', 'advanced')
# Pass the parameters to get_tavilyai_results
t_results = get_tavilyai_results(
# Pass the parameters to do_tavily_ai_search
t_results = do_tavily_ai_search(
keywords=search_keywords,
max_results=kwargs.get('num_results', 10),
include_domains=include_domains,
@@ -343,8 +343,8 @@ def do_tavily_ai_search(search_keywords, max_results=10, **kwargs):
'include_domains': kwargs.get('include_domains', [""]) if kwargs.get('include_domains') else [""]
}
# Pass the parameters to get_tavilyai_results
t_results = get_tavilyai_results(
# Pass the parameters to do_tavily_ai_search
t_results = do_tavily_ai_search(
keywords=search_keywords,
**tavily_params
)

4
lib/ai_web_researcher/tavily_ai_search.py
View File

@@ -16,7 +16,7 @@ Usage:
Modifications:
- To modify the script, update the environment variables in the .env file with the required API keys.
- Adjust the search parameters, such as keywords and search depth, in the `get_tavilyai_results` function as needed.
- Adjust the search parameters, such as keywords and search depth, in the `do_tavily_ai_search` function as needed.
- Customize logging configurations and table formatting according to preferences.
To-Do (TBD):
@@ -49,7 +49,7 @@ from tenacity import retry, stop_after_attempt, wait_random_exponential
@retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
def get_tavilyai_results(keywords, max_results=5, include_domains=None, search_depth="advanced", **kwargs):
def do_tavily_ai_search(keywords, max_results=5, include_domains=None, search_depth="advanced", **kwargs):
"""
Get Tavily AI search results based on specified keywords and options.
"""