14 lines
511 B
Python
14 lines
511 B
Python
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)
|