86 lines
No EOL
3.3 KiB
Python
86 lines
No EOL
3.3 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
from evoagentx.models import OpenAILLMConfig, OpenAILLM, LiteLLMConfig, LiteLLM
|
|
from evoagentx.workflow import WorkFlowGenerator, WorkFlowGraph, WorkFlow
|
|
from evoagentx.agents import AgentManager
|
|
from evoagentx.actions.code_extraction import CodeExtraction
|
|
from evoagentx.actions.code_verification import CodeVerification
|
|
from evoagentx.core.module_utils import extract_code_blocks
|
|
|
|
load_dotenv() # Loads environment variables from .env file
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
|
|
|
|
def main():
|
|
|
|
# LLM configuration
|
|
openai_config = OpenAILLMConfig(model="gpt-4o-mini", openai_key=OPENAI_API_KEY, stream=True, output_response=True, max_tokens=16000)
|
|
# Initialize the language model
|
|
llm = OpenAILLM(config=openai_config)
|
|
|
|
goal = "Generate html code for the Tetris game that can be played in the browser."
|
|
target_directory = "examples/output/tetris_game"
|
|
|
|
wf_generator = WorkFlowGenerator(llm=llm)
|
|
workflow_graph: WorkFlowGraph = wf_generator.generate_workflow(goal=goal)
|
|
|
|
# [optional] display workflow
|
|
workflow_graph.display()
|
|
# [optional] save workflow
|
|
# workflow_graph.save_module(f"{target_directory}/workflow_demo_4o_mini.json")
|
|
#[optional] load saved workflow
|
|
# workflow_graph: WorkFlowGraph = WorkFlowGraph.from_file(f"{target_directory}/workflow_demo_4o_mini.json")
|
|
|
|
agent_manager = AgentManager()
|
|
agent_manager.add_agents_from_workflow(workflow_graph, llm_config=openai_config)
|
|
|
|
workflow = WorkFlow(graph=workflow_graph, agent_manager=agent_manager, llm=llm)
|
|
output = workflow.execute()
|
|
|
|
# verfiy the code
|
|
verification_llm_config = LiteLLMConfig(model="anthropic/claude-3-7-sonnet-20250219", anthropic_key=ANTHROPIC_API_KEY, stream=True, output_response=True, max_tokens=20000)
|
|
verification_llm = LiteLLM(config=verification_llm_config)
|
|
|
|
code_verifier = CodeVerification()
|
|
output = code_verifier.execute(
|
|
llm = verification_llm,
|
|
inputs={
|
|
"requirements": goal,
|
|
"code": output
|
|
}
|
|
).verified_code
|
|
|
|
# extract the code
|
|
os.makedirs(target_directory, exist_ok=True)
|
|
code_blocks = extract_code_blocks(output)
|
|
if len(code_blocks) == 1:
|
|
file_path = os.path.join(target_directory, "index.html")
|
|
with open(file_path, "w") as f:
|
|
f.write(code_blocks[0])
|
|
print(f"You can open this HTML file in a browser to play the Tetris game: {file_path}")
|
|
return
|
|
|
|
code_extractor = CodeExtraction()
|
|
results = code_extractor.execute(
|
|
llm=llm,
|
|
inputs={
|
|
"code_string": output,
|
|
"target_directory": target_directory,
|
|
}
|
|
)
|
|
|
|
print(f"Extracted {len(results.extracted_files)} files:")
|
|
for filename, path in results.extracted_files.items():
|
|
print(f" - {filename}: {path}")
|
|
|
|
if results.main_file:
|
|
print(f"\nMain file: {results.main_file}")
|
|
file_type = os.path.splitext(results.main_file)[1].lower()
|
|
if file_type == '.html':
|
|
print(f"You can open this HTML file in a browser to play the Tetris game")
|
|
else:
|
|
print(f"This is the main entry point for your application")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |