code cleanup

using bedrock
added readme

Author: brnaba-aws

examples/movie-production/movie-production-demo.py

@@ -1,29 +1,24 @@
-from dotenv import load_dotenv
-import streamlit as st
-load_dotenv()
-import os
 import uuid
-import os
 import asyncio
+import streamlit as st
+import os
 from  search_web import tool_handler
-from tool import Tool, ToolResult
+from tool import Tool
 from multi_agent_orchestrator.orchestrator import MultiAgentOrchestrator, OrchestratorConfig
 from multi_agent_orchestrator.agents import (
-    AnthropicAgent,  AnthropicAgentOptions,
-    AgentResponse
+    AgentResponse,
+    BedrockLLMAgent,
+    BedrockLLMAgentOptions
 )
 from multi_agent_orchestrator.types import ConversationMessage
 from multi_agent_orchestrator.classifiers import ClassifierResult
-from supervisor import SupervisorMode, SupervisorModeOptions
+from supervisor_agent import SupervisorAgent, SupervisorAgentOptions
 
 # Set up the Streamlit app
 st.title("AI Movie Production Demo 🎬")
 st.caption("Bring your movie ideas to life with the teams of script writing and casting AI agents")
 
 
-# Get Anthropic API key from user
-anthropic_api_key = st.text_input("Enter Anthropic API Key to access Claude Sonnet 3.5", type="password", value=os.getenv('ANTHROPIC_API_KEY', None))
-
 search_web_tool = Tool(name='search_web',
                           description='Search Web for information',
                           properties={
@@ -34,8 +29,8 @@
                           },
                           required=['query'])
 
-script_writer_agent = AnthropicAgent(AnthropicAgentOptions(
-    api_key=os.getenv('ANTHROPIC_API_KEY', None),
+script_writer_agent = BedrockLLMAgent(BedrockLLMAgentOptions(
+    model_id='us.anthropic.claude-3-sonnet-20240229-v1:0',
     name="ScriptWriterAgent",
     description="""\
 You are an expert screenplay writer. Given a movie idea and genre,
@@ -47,8 +42,8 @@
 3. Ensure the script aligns with the specified genre and target audience
 """))
 
-casting_director_agent = AnthropicAgent(AnthropicAgentOptions(
-    api_key=os.getenv('ANTHROPIC_API_KEY', None),
+casting_director_agent = BedrockLLMAgent(BedrockLLMAgentOptions(
+    model_id='anthropic.claude-3-haiku-20240307-v1:0',
     name="CastingDirectorAgent",
     description="""\
 You are a talented casting director. Given a script outline and character descriptions,\
@@ -63,15 +58,15 @@
 """,
 
 tool_config={
-    'tool': [search_web_tool.to_claude_format()],
+    'tool': [search_web_tool.to_bedrock_format()],
     'toolMaxRecursions': 20,
     'useToolHandler': tool_handler
     },
     save_chat=False
 ))
 
-movie_producer_supervisor = AnthropicAgent(AnthropicAgentOptions(
-    api_key=os.getenv('ANTHROPIC_API_KEY', None),
+movie_producer_supervisor = BedrockLLMAgent(BedrockLLMAgentOptions(
+    model_id='us.anthropic.claude-3-5-sonnet-20241022-v2:0',
     name='MovieProducerAgent',
     description="""
 Experienced movie producer overseeing script and casting.
@@ -85,7 +80,7 @@
 """,
 ))
 
-supervisor = SupervisorMode(SupervisorModeOptions(
+supervisor = SupervisorAgent(SupervisorAgentOptions(
     supervisor=movie_producer_supervisor,
     team=[script_writer_agent, casting_director_agent],
     trace=True

examples/movie-production/movie-production-result.png

@@ -0,0 +1,37 @@
+## 🎬 AI Movie Production Agent
+This Streamlit app is an AI-powered movie production assistant that helps bring your movie ideas to life using Claude 3 on Amazon BedrocK. It automates the process of script writing and casting, allowing you to create compelling movie concepts with ease.
+
+![image](./movie-production.png)
+![image](./movie-production-result.png)
+
+### Features
+- Generates script outlines based on your movie idea, genre, and target audience
+- Suggests suitable actors for main roles, considering their past performances and current availability
+- Provides a concise movie concept overview
+
+### How to get Started?
+
+1. Clone the GitHub repository
+
+```bash
+git clone https://github.com/awslabs/multi-agent-orchestrator.git
+```
+2. Install the required dependencies:
+
+```bash
+cd examples/movie-production
+pip install -r requirements.txt
+```
+3. Get your AWS Credentials
+
+4. Run the Streamlit App
+```bash
+streamlit run movie-production-demo.py
+```
+
+### How it Works?
+
+The AI Movie Production Agent utilizes three main components:
+- **ScriptWriterAgent**: Develops a compelling script outline with character descriptions and key plot points based on the given movie idea and genre.
+- **CastingDirectorAgent**: Suggests suitable actors for the main roles, considering their past performances and current availability by making web search using a tool.
+- **MovieProducerAgent**: Oversees the entire process, coordinating between the ScriptWriter and CastingDirector, and providing a concise movie concept overview.

examples/movie-production/movie-production.png

@@ -1,2 +1,3 @@
-serpapi
-google-search-results
+multi-agent-orchestrator
+streamlit
+duckduckgo-search

examples/movie-production/readme.md

@@ -1,8 +1,7 @@
-import json
 from typing import Any
 from tool import ToolResult
-from multi_agent_orchestrator.types import ParticipantRole
-
+from multi_agent_orchestrator.types import ParticipantRole, ConversationMessage
+from multi_agent_orchestrator.utils.logger import Logger
 from duckduckgo_search import DDGS
 
 async def tool_handler(response: Any, conversation: list[dict[str, Any]],) -> Any:
@@ -14,18 +13,18 @@ async def tool_handler(response: Any, conversation: list[dict[str, Any]],) -> An
 
     for block in content_blocks:
         # Determine if it's a tool use block based on platform
-        tool_use_block =  block if block.type == "tool_use" else None
+        tool_use_block =  block.get('toolUse') if "toolUse" in block else None
         if not tool_use_block:
             continue
 
-        tool_name = (tool_use_block.name)
+        tool_name = (tool_use_block.get('name'))
 
         tool_id = (
-            tool_use_block.id
+            tool_use_block.get('toolUseId')
         )
 
         # Get input based on platform
-        input_data = (tool_use_block.input)
+        input_data = (tool_use_block.get('input'))
 
         # Process the tool use
         if (tool_name == "search_web"):
@@ -37,15 +36,12 @@ async def tool_handler(response: Any, conversation: list[dict[str, Any]],) -> An
         tool_result = ToolResult(tool_id, result)
 
         # Format according to platform
-        formatted_result = (tool_result.to_anthropic_format())
+        formatted_result = (tool_result.to_bedrock_format())
 
         tool_results.append(formatted_result)
 
-        # Create and return appropriate message format
-        return {
-            'role': ParticipantRole.USER.value,
-            'content': tool_results
-        }
+    # Create and return appropriate message format
+    return ConversationMessage(role=ParticipantRole.USER.value, content=tool_results)
 
 def search_web(query: str, num_results: int = 2) -> str:
     """
@@ -56,23 +52,17 @@ def search_web(query: str, num_results: int = 2) -> str:
         num_results(int): The number of results to return.
 
     Returns:
-        str: The search results from Google.
-            Keys:
-                - 'search_results': List of organic search results.
-                - 'recipes_results': List of recipes search results.
-                - 'shopping_results': List of shopping search results.
-                - 'knowledge_graph': The knowledge graph.
-                - 'related_questions': List of related questions.
+        str: The search results from DDG.
     """
 
     try:
 
-        print(f"Searching DDG for: {query}")
+        Logger.info(f"Searching DDG for: {query}")
 
         search = DDGS().text(query, max_results=num_results)
         return ('\n'.join(result.get('body','') for result in search))
 
 
     except Exception as e:
-        print(f"Error searching for the query {query}: {e}")
+        Logger.error(f"Error searching for the query {query}: {e}")
         return f"Error searching for the query {query}: {e}"