import sys

from program.config import AgentWRetreiverConfig
from program.program import AgentWRetreiver
from program.retriever import ExampleRetriever

if __name__ == "__main__":
    username = sys.argv[1]  # ← first arg after script name (username)
    config = AgentWRetreiverConfig(num_fetch=1)
    retriever = ExampleRetriever(config, needed_param="hi")
    agent = AgentWRetreiver(config, retriever=retriever)
    repo_path = f"{username}/nested_repo_2"
    agent.push_to_hub(repo_path, with_code=True)