You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

20 lines
679 B

from .llm_handler import call_llm
import logging
logger = logging.getLogger(__name__)
def get_relevant_links(website_name, links, model="gpt-4", provider="openai"):
"""Uses the specified LLM model to decide which links are relevant for a brochure."""
system_prompt = "You are an AI assistant that selects the most relevant links for a company brochure."
user_prompt = f"""
Here are links found on {website_name}'s website. Identify the relevant ones:
{links}
"""
messages = [
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_prompt}
]
return call_llm(messages, model=model, provider=provider)