File size: 3,341 Bytes
5374a2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
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() |