From f73d46bf10c426eac785110d65409553b715dfd5 Mon Sep 17 00:00:00 2001
From: Huu Le <39040748+leehuwuj@users.noreply.github.com>
Date: Fri, 17 Jan 2025 16:59:05 +0700
Subject: [PATCH] fix missing multiagent code (#483)
---
.changeset/fair-months-camp.md | 5 +++++
helpers/python.ts | 7 +++++++
.../agents/python/blog/app/workflows/planner.py | 2 +-
.../app/workflows/financial_report.py | 12 ++++++------
.../engines/python/agent/tools/interpreter.py | 2 +-
5 files changed, 20 insertions(+), 8 deletions(-)
create mode 100644 .changeset/fair-months-camp.md
diff --git a/.changeset/fair-months-camp.md b/.changeset/fair-months-camp.md
new file mode 100644
index 00000000..33caa38d
--- /dev/null
+++ b/.changeset/fair-months-camp.md
@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+Fix missing copy of the multiagent code
diff --git a/helpers/python.ts b/helpers/python.ts
index 5201c1b5..3569a81d 100644
--- a/helpers/python.ts
+++ b/helpers/python.ts
@@ -480,6 +480,13 @@ export const installPythonTemplate = async ({
await copyRouterCode(root, tools ?? []);
}
+ // Copy multiagents overrides
+ if (template === "multiagent") {
+ await copy("**", path.join(root), {
+ cwd: path.join(compPath, "multiagent", "python"),
+ });
+ }
+
if (template === "multiagent" || template === "reflex") {
if (useCase) {
const sourcePath =
diff --git a/templates/components/agents/python/blog/app/workflows/planner.py b/templates/components/agents/python/blog/app/workflows/planner.py
index 339bb739..f2e010fa 100644
--- a/templates/components/agents/python/blog/app/workflows/planner.py
+++ b/templates/components/agents/python/blog/app/workflows/planner.py
@@ -317,7 +317,7 @@ class Planner:
# gather completed sub-tasks and response pairs
completed_outputs_str = ""
for sub_task_name, task_output in completed_sub_task.items():
- task_str = f"{sub_task_name}:\n" f"\t{task_output!s}\n"
+ task_str = f"{sub_task_name}:\n\t{task_output!s}\n"
completed_outputs_str += task_str
# get a string for the remaining sub-tasks
diff --git a/templates/components/agents/python/financial_report/app/workflows/financial_report.py b/templates/components/agents/python/financial_report/app/workflows/financial_report.py
index 4dffeebc..adfdb27d 100644
--- a/templates/components/agents/python/financial_report/app/workflows/financial_report.py
+++ b/templates/components/agents/python/financial_report/app/workflows/financial_report.py
@@ -108,13 +108,13 @@ class FinancialReportWorkflow(Workflow):
self.query_engine_tool = query_engine_tool
self.code_interpreter_tool = code_interpreter_tool
self.document_generator_tool = document_generator_tool
- assert (
- query_engine_tool is not None
- ), "Query engine tool is not found. Try run generation script or upload a document file first."
+ assert query_engine_tool is not None, (
+ "Query engine tool is not found. Try run generation script or upload a document file first."
+ )
assert code_interpreter_tool is not None, "Code interpreter tool is required"
- assert (
- document_generator_tool is not None
- ), "Document generator tool is required"
+ assert document_generator_tool is not None, (
+ "Document generator tool is required"
+ )
self.tools = [
self.query_engine_tool,
self.code_interpreter_tool,
diff --git a/templates/components/engines/python/agent/tools/interpreter.py b/templates/components/engines/python/agent/tools/interpreter.py
index 20f93b85..133bbcaf 100644
--- a/templates/components/engines/python/agent/tools/interpreter.py
+++ b/templates/components/engines/python/agent/tools/interpreter.py
@@ -161,7 +161,7 @@ class E2BCodeInterpreter:
if self.interpreter:
logger.info(
- f"\n{'='*50}\n> Running following AI-generated code:\n{code}\n{'='*50}"
+ f"\n{'=' * 50}\n> Running following AI-generated code:\n{code}\n{'=' * 50}"
)
exec = self.interpreter.run_code(code)
--
GitLab