|
10 | 10 | # Mapping of framework choices to their respective agent import and class names
|
11 | 11 | FRAMEWORKS = {
|
12 | 12 | 'openai': {
|
13 |
| - 'agent_import': 'from agent.example_openai_agent import ExampleAgent', |
| 13 | + 'agent_import': 'from agents.example_openai_agent import ExampleAgent', |
14 | 14 | 'agent_class': 'ExampleAgent',
|
15 | 15 | 'agent_template_filename': 'example_openai_agent.py.tpl'
|
16 | 16 | },
|
17 | 17 | 'langchain': {
|
18 |
| - 'agent_import': 'from agent.example_langchain_agent import ExampleAgent', |
| 18 | + 'agent_import': 'from agents.example_langchain_agent import ExampleAgent', |
19 | 19 | 'agent_class': 'ExampleAgent',
|
20 | 20 | 'agent_template_filename': 'example_langchain_agent.py.tpl'
|
21 | 21 | },
|
22 | 22 | 'llama': {
|
23 |
| - 'agent_import': 'from agent.example_llama_agent import ExampleAgent', |
| 23 | + 'agent_import': 'from agents.example_llama_agent import ExampleAgent', |
24 | 24 | 'agent_class': 'ExampleAgent',
|
25 | 25 | 'agent_template_filename': 'example_llamaindex_agent.py.tpl'
|
26 | 26 | },
|
27 | 27 | 'blank': {
|
28 |
| - 'agent_import': 'from agent.example_agent import ExampleAgent', |
| 28 | + 'agent_import': 'from agents.example_agent import ExampleAgent', |
29 | 29 | 'agent_class': 'ExampleAgent',
|
30 | 30 | 'agent_template_filename': 'example_agent.py.tpl'
|
31 | 31 | }
|
@@ -75,8 +75,8 @@ def init(project_name, framework):
|
75 | 75 |
|
76 | 76 | # Copy agent template to agents directory
|
77 | 77 | agent_template_filename = FRAMEWORKS[framework]['agent_template_filename']
|
78 |
| - agent_src_path = TEMPLATES_DIR / 'agent' / agent_template_filename |
79 |
| - agent_dst_path = project_path / 'agent' / agent_template_filename[:-4] # Remove '.tpl' extension |
| 78 | + agent_src_path = TEMPLATES_DIR / 'agents' / agent_template_filename |
| 79 | + agent_dst_path = project_path / 'agents' / agent_template_filename[:-4] # Remove '.tpl' extension |
80 | 80 | shutil.copyfile(agent_src_path, agent_dst_path)
|
81 | 81 |
|
82 | 82 | # Copy .env template
|
|
0 commit comments