import requests
from langchain_core.messages import SystemMessage,HumanMessage
from ..utils.schemas import BotAndDiversityFormatter
from ..utils.prompts import bot_and_diversity_prompt
from ..utils.utils import process_query


def get_bot_and_diversity(state,llm_gpt,url):
    messages = [SystemMessage(content=bot_and_diversity_prompt), 
    HumanMessage(content=str(state['messages']))]
    parameters=llm_gpt.with_structured_output(BotAndDiversityFormatter , method='function_calling').invoke(messages)
    print(parameters)
    response = requests.get(url, params={'top_n': parameters.top_n, 'start_date': None , 'end_date':None,'influencer_username':parameters.influencer_name})
    return response