From dacc472e156c9afb0d5f997d165231cf7da2e0eb Mon Sep 17 00:00:00 2001
From: pliny <133052465+elder-plinius@users.noreply.github.com>
Date: Wed, 29 Nov 2023 13:47:29 -0800
Subject: [PATCH] Create logistics.py
---
playground/demos/logistics/logistics.py | 67 +++++++++++++++++++++++++
1 file changed, 67 insertions(+)
create mode 100644 playground/demos/logistics/logistics.py
diff --git a/playground/demos/logistics/logistics.py b/playground/demos/logistics/logistics.py
new file mode 100644
index 00000000..36d74589
--- /dev/null
+++ b/playground/demos/logistics/logistics.py
@@ -0,0 +1,67 @@
+from swarms.structs import Agent
+import os
+from dotenv import load_dotenv
+from swarms.models import GPT4VisionAPI
+from swarms.prompts.logistics import (
+ Health_Security_Agent_Prompt,
+ Quality_Control_Agent_Prompt,
+ Productivity_Agent_Prompt,
+ Safety_Agent_Prompt,
+ Security_Agent_Prompt,
+ Sustainability_Agent_Prompt,
+ Efficiency_Agent_Prompt
+)
+
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
+
+llm = GPT4VisionAPI(openai_api_key=api_key)
+
+# Image for analysis
+factory_image = "factory_image1.jpg"
+
+# Initialize agents with respective prompts
+health_security_agent = Agent(
+ llm=llm, sop=Health_Security_Agent_Prompt, max_loops=3, multi_modal=True
+)
+quality_control_agent = Agent(
+ llm=llm, sop=Quality_Control_Agent_Prompt, max_loops=3, multi_modal=True
+)
+productivity_agent = Agent(
+ llm=llm, sop=Productivity_Agent_Prompt, max_loops=3, multi_modal=True
+)
+safety_agent = Agent(
+ llm=llm, sop=Safety_Agent_Prompt, max_loops=3, multi_modal=True
+)
+security_agent = Agent(
+ llm=llm, sop=Security_Agent_Prompt, max_loops=3, multi_modal=True
+)
+sustainability_agent = Agent(
+ llm=llm, sop=Sustainability_Agent_Prompt, max_loops=3, multi_modal=True
+)
+efficiency_agent = Agent(
+ llm=llm, sop=Efficiency_Agent_Prompt, max_loops=3, multi_modal=True
+)
+
+# Run agents with respective tasks on the same image
+health_analysis = health_security_agent.run(
+ "Analyze the safety of this factory", factory_image
+)
+quality_analysis = quality_control_agent.run(
+ "Examine product quality in the factory", factory_image
+)
+productivity_analysis = productivity_agent.run(
+ "Evaluate factory productivity", factory_image
+)
+safety_analysis = safety_agent.run(
+ "Inspect the factory's adherence to safety standards", factory_image
+)
+security_analysis = security_agent.run(
+ "Assess the factory's security measures and systems", factory_image
+)
+sustainability_analysis = sustainability_agent.run(
+ "Examine the factory's sustainability practices", factory_image
+)
+efficiency_analysis = efficiency_agent.run(
+ "Analyze the efficiency of the factory's manufacturing process", factory_image
+)