From 944c21cb6834338a93da55d5a63daefde407c5a9 Mon Sep 17 00:00:00 2001 From: Akshay Chintalapati <64036106+akshayvkt@users.noreply.github.com> Date: Thu, 30 May 2024 17:02:53 -0700 Subject: [PATCH 1/3] Update README.md Updated code to install requirements based on repo hierarchy changes --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index a0f4cbe..32b95f6 100644 --- a/README.md +++ b/README.md @@ -147,7 +147,7 @@ To run this script, you need to have the following: 2. Install the required Python packages by running the following command: ```bash -pip install -r requirements.txt +pip install -r flask_app/requirements.txt ``` 3. Replace the placeholder API key in the script with your actual Anthropic API key: From 8d53c41b5fa89cf6d60a527846a119539e2f5ff4 Mon Sep 17 00:00:00 2001 From: Akshay Chintalapati <64036106+akshayvkt@users.noreply.github.com> Date: Thu, 30 May 2024 17:14:38 -0700 Subject: [PATCH 2/3] Update requirements.txt --- flask_app/requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/flask_app/requirements.txt b/flask_app/requirements.txt index 2ead9a7..516b0b1 100644 --- a/flask_app/requirements.txt +++ b/flask_app/requirements.txt @@ -5,3 +5,4 @@ ollama groq openai Flask +google-generativeai \ No newline at end of file From 5170f74a02ee3e0291b26b91a1a09eebcbce18fc Mon Sep 17 00:00:00 2001 From: Akshay Chintalapati <64036106+akshayvkt@users.noreply.github.com> Date: Thu, 30 May 2024 21:32:18 -0700 Subject: [PATCH 3/3] Update maestro-gpt4o.py --- maestro-gpt4o.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/maestro-gpt4o.py b/maestro-gpt4o.py index 4adb01c..70d76fe 100644 --- a/maestro-gpt4o.py +++ b/maestro-gpt4o.py @@ -28,6 +28,7 @@ def calculate_subagent_cost(model, input_tokens, output_tokens): "claude-3-opus-20240229": {"input_cost_per_mtok": 15.00, "output_cost_per_mtok": 75.00}, "claude-3-haiku-20240307": {"input_cost_per_mtok": 0.25, "output_cost_per_mtok": 1.25}, "claude-3-sonnet-20240229": {"input_cost_per_mtok": 3.00, "output_cost_per_mtok": 15.00}, + "gpt-4o": {"input_cost_per_mtok": 5.00, "output_cost_per_mtok": 15.00} } # Calculate cost @@ -61,7 +62,8 @@ def gpt_orchestrator(objective, file_content=None, previous_results=None, use_se usage = gpt_response.usage console.print(Panel(response_text, title=f"[bold green]gpt Orchestrator[/bold green]", title_align="left", border_style="green", subtitle="Sending task to gpt 👇")) - console.print(f"Input Tokens: {usage.prompt_tokens}, Output Tokens: {usage.completion_tokens}, Total Tokens: {usage.total_tokens}") + total_cost = calculate_subagent_cost(ORCHESTRATOR_MODEL,usage.prompt_tokens,usage.completion_tokens) + console.print(f"Input Tokens: {usage.prompt_tokens}, Output Tokens: {usage.completion_tokens}, Total Tokens: {usage.total_tokens}, Total Cost: ${total_cost:.4f}") search_query = None if use_search: @@ -113,7 +115,8 @@ def gpt_sub_agent(prompt, search_query=None, previous_gpt_tasks=None, use_search usage = gpt_response.usage console.print(Panel(response_text, title="[bold blue]gpt Sub-agent Result[/bold blue]", title_align="left", border_style="blue", subtitle="Task completed, sending result to gpt 👇")) - console.print(f"Input Tokens: {usage.prompt_tokens}, Output Tokens: {usage.completion_tokens}, Total Tokens: {usage.total_tokens}") + total_cost = calculate_subagent_cost(SUB_AGENT_MODEL,usage.prompt_tokens,usage.completion_tokens) + console.print(f"Input Tokens: {usage.prompt_tokens}, Output Tokens: {usage.completion_tokens}, Total Tokens: {usage.total_tokens}, Total Cost: ${total_cost:.4f}") if usage.completion_tokens >= 4000: # Threshold set to 4000 as a precaution console.print("[bold yellow]Warning:[/bold yellow] Output may be truncated. Attempting to continue the response.")