[BUFG][Swarms Integration]

Author: Kye

errors.txt

@@ -0,0 +1,115 @@
+/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/thop/profile.py:12: DeprecationWarning: distutils Version classes are deprecated. Use packaging.version instead.
+  if LooseVersion(torch.__version__) < LooseVersion("1.0.0"):
+/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/thop/profile.py:68: DeprecationWarning: distutils Version classes are deprecated. Use packaging.version instead.
+  if LooseVersion(torch.__version__) >= LooseVersion("1.1.0"):
+distutils Version classes are deprecated. Use packaging.version instead.
+distutils Version classes are deprecated. Use packaging.version instead.
+Traceback (most recent call last):
+  File "/Users/defalt/Desktop/Athena/research/tree-of-thoughts/example.py", line 62, in <module>
+    solution = tree_of_thoughts.solve(
+               ^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Users/defalt/Desktop/Athena/research/tree-of-thoughts/tree_of_thoughts/treeofthoughts.py", line 559, in solve
+    return self.monte_carlo_search(
+           ^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Users/defalt/Desktop/Athena/research/tree-of-thoughts/tree_of_thoughts/treeofthoughts.py", line 604, in monte_carlo_search
+    thoughts = self.model.generate_thoughts(
+               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Users/defalt/Desktop/Athena/research/tree-of-thoughts/tree_of_thoughts/tot_agent.py", line 106, in generate_thoughts
+    thoughts = self.run(prompt)
+               ^^^^^^^^^^^^^^^^
+  File "/Users/defalt/Desktop/Athena/research/tree-of-thoughts/tree_of_thoughts/tot_agent.py", line 70, in run
+    response = self.agent(task)
+               ^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/swarms/structs/agent.py", line 649, in __call__
+    self.run(task, img, *args, **kwargs)
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/swarms/structs/agent.py", line 570, in run
+    response = self.llm(
+               ^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/langchain/llms/base.py", line 876, in __call__
+    self.generate(
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/langchain/llms/base.py", line 656, in generate
+    output = self._generate_helper(
+             ^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/langchain/llms/base.py", line 544, in _generate_helper
+    raise e
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/langchain/llms/base.py", line 531, in _generate_helper
+    self._generate(
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/swarms/models/openai_models.py", line 1082, in _generate
+    full_response = completion_with_retry(
+                    ^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/swarms/models/openai_models.py", line 212, in completion_with_retry
+    return _completion_with_retry(**kwargs)
+           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/tenacity/__init__.py", line 289, in wrapped_f
+    return self(f, *args, **kw)
+           ^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/tenacity/__init__.py", line 379, in __call__
+    do = self.iter(retry_state=retry_state)
+         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/tenacity/__init__.py", line 314, in iter
+    return fut.result()
+           ^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/concurrent/futures/_base.py", line 449, in result
+    return self.__get_result()
+           ^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/concurrent/futures/_base.py", line 401, in __get_result
+    raise self._exception
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/tenacity/__init__.py", line 382, in __call__
+    result = fn(*args, **kwargs)
+             ^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/swarms/models/openai_models.py", line 210, in _completion_with_retry
+    return llm.client.create(**kwargs)
+           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/openai/api_resources/chat_completion.py", line 25, in create
+    return super().create(*args, **kwargs)
+           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/openai/api_resources/abstract/engine_api_resource.py", line 153, in create
+    response, _, api_key = requestor.request(
+                           ^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/openai/api_requestor.py", line 288, in request
+    result = self.request_raw(
+             ^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/openai/api_requestor.py", line 596, in request_raw
+    result = _thread_context.session.request(
+             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/requests/sessions.py", line 589, in request
+    resp = self.send(prep, **send_kwargs)
+           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/requests/sessions.py", line 703, in send
+    r = adapter.send(request, **kwargs)
+        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/requests/adapters.py", line 486, in send
+    resp = conn.urlopen(
+           ^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/urllib3/connectionpool.py", line 703, in urlopen
+    httplib_response = self._make_request(
+                       ^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/urllib3/connectionpool.py", line 449, in _make_request
+    six.raise_from(e, None)
+  File "<string>", line 3, in raise_from
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/site-packages/urllib3/connectionpool.py", line 444, in _make_request
+    httplib_response = conn.getresponse()
+                       ^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/http/client.py", line 1374, in getresponse
+    response.begin()
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/http/client.py", line 318, in begin
+    version, status, reason = self._read_status()
+                              ^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/http/client.py", line 279, in _read_status
+    line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
+               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/socket.py", line 705, in readinto
+    return self._sock.recv_into(b)
+           ^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/ssl.py", line 1278, in recv_into
+    return self.read(nbytes, buffer)
+           ^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/Library/Frameworks/Python.framework/Versions/3.11/lib/python3.11/ssl.py", line 1134, in read
+    return self._sslobj.read(len, buffer)
+           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+KeyboardInterrupt
+Implicitly cleaning up <TemporaryDirectory '/var/folders/zl/0vgkvtm52vjcs1skpnlqnssw0000gn/T/tmpk9wgadbgwandb'>
+Implicitly cleaning up <TemporaryDirectory '/var/folders/zl/0vgkvtm52vjcs1skpnlqnssw0000gn/T/tmpeqcgk3h9wandb-artifacts'>
+Implicitly cleaning up <TemporaryDirectory '/var/folders/zl/0vgkvtm52vjcs1skpnlqnssw0000gn/T/tmpoadrfcapwandb-media'>
+Implicitly cleaning up <TemporaryDirectory '/var/folders/zl/0vgkvtm52vjcs1skpnlqnssw0000gn/T/tmp431qofrrwandb-media'>
+Implicitly cleaning up <TemporaryDirectory '/var/folders/zl/0vgkvtm52vjcs1skpnlqnssw0000gn/T/tmpichwdx5i'>

example.py

@@ -3,6 +3,7 @@
 from dotenv import load_dotenv
 from swarms import Agent, OpenAIChat
 
+# Load the environment variables
 load_dotenv()
 
 # Get the API key from the environment
@@ -11,9 +12,12 @@
 # Initialize an agent from swarms
 agent = Agent(
     agent_name="tree_of_thoughts",
-    agent_description="This agent uses the tree_of_thoughts library to generate thoughts.",
+    agent_description=(
+        "This agent uses the tree_of_thoughts library to generate thoughts."
+    ),
     system_prompt=None,
-    llm = OpenAIChat(),   
+    llm=OpenAIChat(),
+    max_loops=1,
 )
 
 # Initialize the ToTAgent class with the API key

huggingface_example.py

@@ -1,7 +1,7 @@
 import os
 from tree_of_thoughts import ToTAgent, MonteCarloSearch
 from dotenv import load_dotenv
-from swarms import Agent, OpenAIChat, HuggingfaceLLM
+from swarms import Agent, HuggingfaceLLM
 
 load_dotenv()
 
@@ -11,11 +11,13 @@
 # Initialize an agent from swarms
 agent = Agent(
     agent_name="tree_of_thoughts",
-    agent_description="This agent uses the tree_of_thoughts library to generate thoughts.",
+    agent_description=(
+        "This agent uses the tree_of_thoughts library to generate thoughts."
+    ),
     system_prompt=None,
-    llm = HuggingfaceLLM(
+    llm=HuggingfaceLLM(
         "EleutherAI/gpt-neo-2.7B",
-    ),   
+    ),
 )
 
 # Initialize the ToTAgent class with the API key

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tree-of-thoughts"
-version = "0.4.2"
+version = "0.4.5"
 description = "Tree of Thoughts - Pytorch"
 authors = ["Kye Gomez <[email protected]>"]
 license = "MIT"

tree_of_thoughts/tot_agent.py

@@ -12,20 +12,20 @@ class ToTAgent:
     """
 
     OpenAI Language Model API Wrapper
-    
+
     Args:
         agent (Agent): Agent class from swarms
         strategy (str): Strategy to use for generating thoughts
         evaluation_strategy (str): Strategy to use for evaluating states
         enable_react (bool): Enable ReAct prompting
         k (int): Number of thoughts to generate
-        
+
     Methods:
         run(task: str) -> list: Generate text from prompt using OpenAI API
         generate_thoughts(state, k, initial_prompt, rejected_solutions=None) -> list: Generate thoughts from state using OpenAI API
         generate_solution(initial_prompt, state, rejected_solutions=None) -> str: Generate solution from state using OpenAI API
         evaluate_states(states, initial_prompt) -> dict: Evaluate states of reasoning using OpenAI API
-        
+
     Examples:
         >>> from tree_of_thoughts.tot_agent import ToTAgent
         >>> from swarms import Agent
@@ -103,7 +103,7 @@ def generate_thoughts(
         complete the {initial_prompt} without making the same mistakes you did with the evaluated rejected solutions. Be simple. Be direct. Provide intuitive solutions as soon as you think of them."""
 
         prompt += self.react_prompt
-        thoughts = self.generate_text(prompt)
+        thoughts = self.run(prompt)
         return thoughts
 
     def generate_solution(self, initial_prompt, state, rejected_solutions=None):
@@ -119,7 +119,7 @@ def generate_solution(self, initial_prompt, state, rejected_solutions=None):
             Devise the best possible solution for the task: {initial_prompt}, Here are evaluated solutions that were rejected: 
             ###{rejected_solutions}###, 
             complete the {initial_prompt} without making the same mistakes you did with the evaluated rejected solutions. Be simple. Be direct. Provide intuitive solutions as soon as you think of them."""
-            answer = self.generate_text(prompt, 1)
+            answer = self.run(prompt)
             print(f"Answerrrrrr {answer}")
             # print(thoughts)
             # print(f"General Solution : {answer}")

tree_of_thoughts/treeofthoughts.py

@@ -113,6 +113,8 @@ def adjust_pruning_threshold_moving_average(
 
 
 class BFS(TreeofThoughts):
+    """Class representing the Breadth-First Search algorithm for Tree of Thoughts."""
+
     def solve(
         self,
         initial_prompt,
@@ -122,6 +124,20 @@ def solve(
         value_threshold,
         pruning_threshold=0.5,
     ):
+        """
+        Solve the Tree of Thoughts problem using the Breadth-First Search algorithm.
+
+        Args:
+            initial_prompt (str): The initial prompt for generating thoughts.
+            num_thoughts (int): The number of thoughts to generate at each state.
+            max_steps (int): The maximum number of steps to take in the search.
+            max_states (int): The maximum number of states to keep track of.
+            value_threshold (float): The threshold value for selecting states.
+            pruning_threshold (float, optional): The threshold for dynamic pruning. Defaults to 0.5.
+
+        Returns:
+            str or None: The generated solution or the highest rated state.
+        """
         current_states = [initial_prompt]
         state_values = {}
         dynamic_pruning_threshold = pruning_threshold
@@ -177,18 +193,6 @@ def solve(
                                 self.logNewState(state, value)
                                 logger.debug(f"State Values: {state_values}")
 
-            # if state_values:
-            #     highest_rated_solution = max(state_values.items(), key=lambda x: x[1])
-            #     print(f"highest rated solution: {highest_rated_solution}")
-            #     highest_rated_state = highest_rated_solution[0]  # Use a different name to avoid confusion
-            #     print(f'highest rated state: {highest_rated_state}')
-            #     try:
-            #         solution = self.model.generate_solution(initial_prompt, highest_rated_state)
-            #     except Exception as e:
-            #         logger.error(f"Error in generating solution: {e}")
-            #         solution = None  # Set a fallback value for solution
-
-            #     return solution if solution is not None else highest_rated_state  # Return highest rated state if solution is None
             if state_values:
                 highest_rated_solution = max(
                     state_values.items(), key=lambda x: x[1]