From c23da4d79d29a0450043d9ff5ad3d46a220d596a Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Mon, 20 Nov 2023 10:05:46 -0800
Subject: [PATCH] KAIZU FIXES: AccountSwarm + AI Research team

---
 README.md                                              |  2 +-
 .../demos/accountant_team/account_team2.py             |  0
 playground/demos/ai_research_team/main.py              |  2 +-
 .../swarms_example.ipynb                               |  0
 swarms/__init__.py                                     | 10 ++++++++++
 5 files changed, 12 insertions(+), 2 deletions(-)
 rename account_team2.py => playground/demos/accountant_team/account_team2.py (100%)
 rename swarms_example.ipynb => playground/swarms_example.ipynb (100%)

diff --git a/README.md b/README.md
index 20d6d573..7b80c435 100644
--- a/README.md
+++ b/README.md
@@ -40,7 +40,7 @@ We have a small gallery of examples to run here, [for more check out the docs to
 
 ### Example in Colab:
 
-<a target="_blank" href="https://colab.research.google.com/github/kyegomez/swarms/blob/master/swarms_example.ipynb">
+<a target="_blank" href="https://colab.research.google.com/github/kyegomez/swarms/blob/master/playground/swarms_example.ipynb">
 <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
 </a> Run example in Colab, using your OpenAI API key. 
 
diff --git a/account_team2.py b/playground/demos/accountant_team/account_team2.py
similarity index 100%
rename from account_team2.py
rename to playground/demos/accountant_team/account_team2.py
diff --git a/playground/demos/ai_research_team/main.py b/playground/demos/ai_research_team/main.py
index c986e123..45ef38a8 100644
--- a/playground/demos/ai_research_team/main.py
+++ b/playground/demos/ai_research_team/main.py
@@ -16,7 +16,7 @@ load_dotenv()
 anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
 openai_api_key = os.getenv("OPENAI_API_KEY")
 
-PDF_PATH = "shallowfeedforward.pdf"
+PDF_PATH = "videocon.pdf"
 
 
 # Base llms
diff --git a/swarms_example.ipynb b/playground/swarms_example.ipynb
similarity index 100%
rename from swarms_example.ipynb
rename to playground/swarms_example.ipynb
diff --git a/swarms/__init__.py b/swarms/__init__.py
index 0fd05d72..ef00c1dc 100644
--- a/swarms/__init__.py
+++ b/swarms/__init__.py
@@ -1,3 +1,4 @@
+import logging
 import os
 import warnings
 
@@ -5,6 +6,15 @@ warnings.filterwarnings("ignore", category=UserWarning)
 
 # disable tensorflow warnings
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
+
+try:
+
+    log = logging.getLogger("pytorch")
+    log.propagate = False
+    log.setLevel(logging.ERROR)
+except Exception as error:
+    print(f"Pytorch logging not disabled: {error}")
+
 from swarms.agents import *  # noqa: E402, F403
 from swarms.swarms import *  # noqa: E402, F403
 from swarms.structs import *  # noqa: E402, F403