add second example, patch openai tool init

Author: WillBeebe

examples/0-simple/src/main.py

@@ -2,8 +2,22 @@
 from agents.ada import Ada
 
 agent = Ada(client=AnthropicLLM())
-prompt = "Name five fruit that start with the letter a."
 
-print(prompt)
-response = agent.generate_text(prompt)
-print(response.content)
+def chat(prompt):
+  print(":USER:")
+  print(prompt)
+  print("\n"*3)
+  response = agent.generate_text(prompt)
+  print(":ADA:")
+  print(response.content)
+  print("\n"*3)
+
+print("\n"*2)
+chat("Name five fruit that start with the letter a.")
+# Ada has a few tools, this is one of them (todo: link)
+chat("Get the repo structure of the current directory.")
+# agents store history in-memory by default,
+# they can be configured to work with other storage backends like postgres, mongo, etc...
+chat("Summarize our conversation in 10 words.")
+
+

examples/1-two-agents/Makefile

@@ -0,0 +1,2 @@
+start:
+	poetry run python src/main.py

examples/1-two-agents/poetry.lock

@@ -0,0 +1,28 @@
+[tool.poetry]
+name = "ada-example-1-two-agents"
+version = "0.1.0"
+description = ""
+authors = ["Will Beebe"]
+packages = [
+  {include = "*", from="src"},
+]
+
+[tool.poetry.dependencies]
+python = "^3.11"
+# ada-python = "^0.2.0"
+ada-python = {path = "../../", develop = true}
+
+[tool.poetry.dev-dependencies]
+pytest = "^8.1.1"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+
+[tool.ruff]
+select = ["E", "F", "W", "I001"]
+ignore = ["E501", "F541"]
+fixable = ["W", "I001"]
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

examples/1-two-agents/pyproject.toml

@@ -0,0 +1,39 @@
+from abcs.anthropic import AnthropicLLM
+from abcs.openai import OpenAILLM
+from agents.aristotle import Aristotle
+from agents.plato import Plato
+
+aristotle = Aristotle(client=AnthropicLLM())
+# other agents don't have to be from other providers,
+# but they can be. (todo: link)
+plato = Plato(client=OpenAILLM())
+chat_rounds = 3
+
+def chat(prompt):
+  for i in range(chat_rounds):
+    print(f":ROUND {i} / {chat_rounds}:")
+    response = aristotle.generate_text(prompt)
+    prompt = response.content
+    print(":Aristotle:")
+    print(response.content)
+    print("\n"*1)
+
+    response = plato.generate_text(prompt)
+    prompt = response.content
+    print(":Plato:")
+    print(response.content)
+    print("\n"*1)
+
+
+prompt = """I'm studying for an exam about Plato.
+     I want to role play as Plato to test my knowledge.
+     Can you act as Aristotle and challenge my core beliefs?
+     """
+print("\n"*2)
+print(":USER:")
+print(prompt)
+print("\n"*2)
+chat(prompt)
+
+
+

examples/1-two-agents/src/main.py

@@ -1,6 +1,5 @@
 import logging
 import os
-from pprint import pprint
 from typing import Any, Dict, List, Optional
 
 import anthropic
@@ -109,7 +108,7 @@ def call_tool(
                     ],
                 }
         logger.debug("\n"*20)
-        logger.debug(pprint(past_messages + [tool_message]))
+        logger.debug(f"{past_messages + [tool_message]}")
         logger.debug("\n"*20)
         response = self.client.messages.create(
             model=self.model,

src/abcs/anthropic.py

@@ -62,6 +62,9 @@ def generate_text(
         combined_history = system_message + past_messages + [{"role": "user", "content": prompt}]
         logger.debug("Generating text with prompt: '%s'", prompt)
 
+        if tools is not None and len(tools) == 0:
+            tools = None
+
         try:
             logger.debug('\n'*10)
             logger.debug(combined_history)
@@ -119,7 +122,7 @@ def generate_text(
                     model=self.model,
                     messages=all_history,
                     # might be able to remove
-                    # tools=tools,
+                    tools=tools,
                 )
             return self._translate_response(response)
         except Exception as e:

src/abcs/openai.py

@@ -14,7 +14,7 @@
 class Agent(LLM):
     def __init__(self, client, tool_manager: ToolManager, system_prompt: str, tools=[], storage_manager: StorageManager = None):
         self.tools = tools
-        logger.info("Initializing Agent with tools: %s and system prompt: '%s'", tools, system_prompt)
+        logger.debug("Initializing Agent with tools: %s and system prompt: '%s'", tools, system_prompt)
         super().__init__(
             client=client,
             model=None,