mirror of
https://github.com/User0332/rewards-farmer.git
synced 2026-09-16 01:31:36 +00:00
Ethan Stoner
1a224919b2
The character check allowed "." and "..", which are made entirely of allowed characters and still walk out of the directory, so the traversal guard only stopped the cases containing a separator. Reject both by name and check the resolved path against the profile root as well, since the character set constrains the characters rather than where they point. Also drop the MouseUtils and KeyboardUtils built in run_account and never used, RewardsTaskUtils builds its own, along with the two imports that leaves unused, and say "lowercased" in the no-source warning, which is what the code returns.
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
"""Where search queries come from.
|
|
|
|
Two backends. `llm` is the default and is unchanged, so nothing about an
|
|
existing setup moves. `trends` uses public feeds and needs no account, no
|
|
model and no key, which is the difference between running this in five
|
|
minutes and installing Ollama first.
|
|
|
|
QUERY_SOURCE=trends python src/main.py
|
|
|
|
The LLM's whole job in this project is producing short strings to type into
|
|
Bing, and Bing's own autosuggest answers that question directly.
|
|
"""
|
|
|
|
import logging
|
|
import os
|
|
|
|
import query_sources
|
|
|
|
# llm_utils is imported inside the llm branch rather than here. It imports
|
|
# ollama at module scope, so importing it eagerly would make the ollama package
|
|
# a hard requirement even for a run that never touches a model, which is the
|
|
# opposite of the point. A trends-only install, the Docker image for instance,
|
|
# does not ship it.
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
LLM = "llm"
|
|
TRENDS = "trends"
|
|
|
|
DEFAULT_SOURCE = LLM
|
|
|
|
ENV_VAR = "QUERY_SOURCE"
|
|
|
|
|
|
def selected_source() -> str:
|
|
"""Read on each call so a test can change it without reimporting."""
|
|
choice = os.environ.get(ENV_VAR, DEFAULT_SOURCE).strip().lower()
|
|
|
|
return choice if choice in (LLM, TRENDS) else DEFAULT_SOURCE
|
|
|
|
|
|
def search_query_for_task(task_description: str) -> str:
|
|
"""A query for one "Search on Bing for X" card."""
|
|
if selected_source() == TRENDS:
|
|
query = query_sources.query_from_task_description(task_description)
|
|
|
|
if query:
|
|
return query
|
|
|
|
# Every feed was unreachable. The description still contains the topic,
|
|
# so a trimmed version beats skipping the card entirely.
|
|
logger.warning("No query source reachable, using the lowercased task description.")
|
|
|
|
return task_description.lower()
|
|
|
|
import llm_utils
|
|
|
|
return llm_utils.get_search_query_from_task_description(task_description)
|
|
|
|
|
|
def related_queries(count: int):
|
|
"""`count` queries for the daily search quota."""
|
|
if selected_source() == TRENDS:
|
|
queries = query_sources.related_queries(count)
|
|
|
|
if queries:
|
|
return queries
|
|
|
|
logger.warning("No query source reachable, falling back to the wordlist.")
|
|
|
|
# nouns.txt is already in the repo for exactly this kind of seed.
|
|
return query_sources.wordlist_queries(count)
|
|
|
|
import llm_utils
|
|
|
|
return llm_utils.get_related_search_queries(llm_utils.get_random_noun(), num_queries=count)
|