ruslanmv commited on
Commit
a7710ce
Β·
verified Β·
1 Parent(s): 38b1c87

Fix file viewer + template syntax + requirements.txt generation

Browse files
agent_generator_pkg/src/agent_generator/application/__pycache__/build_service.cpython-311.pyc CHANGED
Binary files a/agent_generator_pkg/src/agent_generator/application/__pycache__/build_service.cpython-311.pyc and b/agent_generator_pkg/src/agent_generator/application/__pycache__/build_service.cpython-311.pyc differ
 
agent_generator_pkg/src/agent_generator/application/build_service.py CHANGED
@@ -77,6 +77,21 @@ def build(
77
  language="markdown",
78
  ))
79
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
80
  # Syntax validation
81
  errors: list[ValidationIssue] = []
82
  warnings: list[ValidationIssue] = []
 
77
  language="markdown",
78
  ))
79
 
80
+ # Generate requirements.txt for Python projects
81
+ if ext == "py":
82
+ deps = ["pydantic>=2.0"]
83
+ if framework_name in ("crewai", "crewai_flow"):
84
+ deps.extend(["crewai>=1.12", "crewai-tools>=0.14"])
85
+ elif framework_name == "langgraph":
86
+ deps.extend(["langgraph>=1.1", "langchain-core>=0.1"])
87
+ elif framework_name == "react":
88
+ deps.append("requests>=2.31")
89
+ generated_files.append(GeneratedFile(
90
+ path="requirements.txt",
91
+ content="\n".join(sorted(set(deps))) + "\n",
92
+ language="text",
93
+ ))
94
+
95
  # Syntax validation
96
  errors: list[ValidationIssue] = []
97
  warnings: list[ValidationIssue] = []
agent_generator_pkg/src/agent_generator/frameworks/crewai_flow/__pycache__/generator.cpython-311.pyc CHANGED
Binary files a/agent_generator_pkg/src/agent_generator/frameworks/crewai_flow/__pycache__/generator.cpython-311.pyc and b/agent_generator_pkg/src/agent_generator/frameworks/crewai_flow/__pycache__/generator.cpython-311.pyc differ
 
agent_generator_pkg/src/agent_generator/frameworks/crewai_flow/generator.py CHANGED
@@ -50,7 +50,7 @@ _FLOW_TEMPLATE = Template(
50
  def begin(self):
51
  """Initialize the flow and run the first task."""
52
  self.state.current_step = "begin"
53
- {%- if tasks %}
54
  crew = Crew(
55
  agents=[{{ tasks[0].agent_id }}],
56
  tasks=[
@@ -64,9 +64,9 @@ _FLOW_TEMPLATE = Template(
64
  )
65
  result = crew.kickoff()
66
  self.state.results["{{ tasks[0].id }}"] = str(result)
67
- {%- endif %}
68
  return self.state
69
- {%- for task in tasks[1:] %}
70
 
71
  @listen(begin{% if not loop.first %}_step_{{ loop.index }}{% endif %})
72
  def step_{{ loop.index + 1 }}(self):
@@ -86,7 +86,7 @@ _FLOW_TEMPLATE = Template(
86
  result = crew.kickoff()
87
  self.state.results["{{ task.id }}"] = str(result)
88
  return self.state
89
- {%- endfor %}
90
 
91
 
92
  def main() -> Any:
 
50
  def begin(self):
51
  """Initialize the flow and run the first task."""
52
  self.state.current_step = "begin"
53
+ {% if tasks %}
54
  crew = Crew(
55
  agents=[{{ tasks[0].agent_id }}],
56
  tasks=[
 
64
  )
65
  result = crew.kickoff()
66
  self.state.results["{{ tasks[0].id }}"] = str(result)
67
+ {% endif %}
68
  return self.state
69
+ {% for task in tasks[1:] %}
70
 
71
  @listen(begin{% if not loop.first %}_step_{{ loop.index }}{% endif %})
72
  def step_{{ loop.index + 1 }}(self):
 
86
  result = crew.kickoff()
87
  self.state.results["{{ task.id }}"] = str(result)
88
  return self.state
89
+ {% endfor %}
90
 
91
 
92
  def main() -> Any:
agent_generator_pkg/src/agent_generator/frameworks/langgraph/__pycache__/generator.cpython-311.pyc CHANGED
Binary files a/agent_generator_pkg/src/agent_generator/frameworks/langgraph/__pycache__/generator.cpython-311.pyc and b/agent_generator_pkg/src/agent_generator/frameworks/langgraph/__pycache__/generator.cpython-311.pyc differ
 
agent_generator_pkg/src/agent_generator/frameworks/langgraph/generator.py CHANGED
@@ -26,9 +26,9 @@ _LANG_TEMPLATE = Template(
26
  class WorkflowState(TypedDict):
27
  """Shared state passed between nodes."""
28
  input: str
29
- {%- for task in tasks %}
30
  {{ task.id }}_output: str
31
- {%- endfor %}
32
 
33
  # ─────────────────────────────────────────────────────────
34
  # Node functions
@@ -38,14 +38,14 @@ _LANG_TEMPLATE = Template(
38
  """{{ task.goal }}"""
39
  # Process input and produce output for this step
40
  context = state.get("input", "")
41
- {%- if task.inputs %}
42
  # Also consider upstream outputs
43
- {%- for inp_name in task.inputs %}
44
  prev = state.get("{{ inp_name }}", "")
45
  if prev:
46
  context = f"{context}\\n{prev}"
47
- {%- endfor %}
48
- {%- endif %}
49
  result = f"[{{ task.id }}] Processed: {context[:200]}"
50
  return {"{{ task.id }}_output": result}
51
 
@@ -58,19 +58,19 @@ _LANG_TEMPLATE = Template(
58
  def build_graph() -> StateGraph:
59
  """Build and return the compiled LangGraph workflow."""
60
  graph = StateGraph(WorkflowState)
61
- {%- for task in tasks %}
62
  graph.add_node("{{ task.id }}", {{ task.id }})
63
- {%- endfor %}
64
 
65
  # Edges
66
- {%- if edges %}
67
  graph.add_edge(START, "{{ edges[0].source }}")
68
- {%- for edge in edges %}
69
  graph.add_edge("{{ edge.source }}", "{{ edge.target }}")
70
- {%- endfor %}
71
- {%- else %}
72
  graph.add_edge(START, "{{ tasks[0].id }}")
73
- {%- endif %}
74
 
75
  return graph
76
 
 
26
  class WorkflowState(TypedDict):
27
  """Shared state passed between nodes."""
28
  input: str
29
+ {% for task in tasks %}
30
  {{ task.id }}_output: str
31
+ {% endfor %}
32
 
33
  # ─────────────────────────────────────────────────────────
34
  # Node functions
 
38
  """{{ task.goal }}"""
39
  # Process input and produce output for this step
40
  context = state.get("input", "")
41
+ {% if task.inputs %}
42
  # Also consider upstream outputs
43
+ {% for inp_name in task.inputs %}
44
  prev = state.get("{{ inp_name }}", "")
45
  if prev:
46
  context = f"{context}\\n{prev}"
47
+ {% endfor %}
48
+ {% endif %}
49
  result = f"[{{ task.id }}] Processed: {context[:200]}"
50
  return {"{{ task.id }}_output": result}
51
 
 
58
  def build_graph() -> StateGraph:
59
  """Build and return the compiled LangGraph workflow."""
60
  graph = StateGraph(WorkflowState)
61
+ {% for task in tasks %}
62
  graph.add_node("{{ task.id }}", {{ task.id }})
63
+ {% endfor %}
64
 
65
  # Edges
66
+ {% if edges %}
67
  graph.add_edge(START, "{{ edges[0].source }}")
68
+ {% for edge in edges %}
69
  graph.add_edge("{{ edge.source }}", "{{ edge.target }}")
70
+ {% endfor %}
71
+ {% else %}
72
  graph.add_edge(START, "{{ tasks[0].id }}")
73
+ {% endif %}
74
 
75
  return graph
76
 
agent_generator_pkg/src/agent_generator/frameworks/react/__pycache__/generator.cpython-311.pyc CHANGED
Binary files a/agent_generator_pkg/src/agent_generator/frameworks/react/__pycache__/generator.cpython-311.pyc and b/agent_generator_pkg/src/agent_generator/frameworks/react/__pycache__/generator.cpython-311.pyc differ
 
agent_generator_pkg/src/agent_generator/frameworks/react/generator.py CHANGED
@@ -153,9 +153,9 @@ _REACT_TEMPLATE = Template(
153
  def main() -> Any:
154
  """Entry-point for MCP wrapper or direct execution."""
155
  results = {}
156
- {%- for task in tasks %}
157
  results["{{ task.id }}"] = run_{{ task.id }}()
158
- {%- endfor %}
159
  return results
160
 
161
 
 
153
  def main() -> Any:
154
  """Entry-point for MCP wrapper or direct execution."""
155
  results = {}
156
+ {% for task in tasks %}
157
  results["{{ task.id }}"] = run_{{ task.id }}()
158
+ {% endfor %}
159
  return results
160
 
161