16 lines
651 B
Python
16 lines
651 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
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"
|
|
extra = str(Path(__file__).resolve().parent.parent.parent / "extra_files" / "extra.yaml")
|
|
agent.push_to_hub(repo_path, with_code=True, extra_files=[extra])
|