from .agentbase import AgentBase
from .prompts import RECOMMENDATION_AGENT_PROMPT
from modals.inputs import LLMConfig
from .tools import recommend_talent_tool


llm_config = LLMConfig(
    api_key="AIzaSyCOIgSPUdsfLcrQTwIN61W5Me5lv1krOr8",
    base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
    model="gemini-2.0-flash",
    )

def recommend_talent_agent(brand_request):
  messages = [{'role': 'system', 'content': RECOMMENDATION_AGENT_PROMPT}, {'role': 'user', 'content': f'Brand Request: {brand_request}'}]
  agent = AgentBase(llm_config=llm_config)
  parsed_response = None
  if not isinstance(parsed_response, dict):
    response = agent.generate_response(messages=messages)
    parsed_response = agent.parse_json(response)
  return parsed_response


if __name__ == "__main__":
    brand_request = "Find me young male fitness bloggers with a medium-sized following"
    search_tool_args = recommend_talent_agent(brand_request)
    print(search_tool_args)
    print(recommend_talent_tool(**search_tool_args))