diff --git a/README.md b/README.md index 064a713a..9df4e877 100644 --- a/README.md +++ b/README.md @@ -103,7 +103,7 @@ llm = OpenAIChat( agent = Agent(llm=llm, max_loops=1, autosave=True, dashboard=True) # Run the workflow on a task -agent.run("Generate a 10,000 word blog on health and wellness.") +agent.run("Generate a 10,000 word article on health and wellness for men under 50.") ``` @@ -114,7 +114,7 @@ agent.run("Generate a 10,000 word blog on health and wellness.") import os from dotenv import load_dotenv from swarms import Agent, OpenAIChat -from playground.memory.chromadb_example import ChromaDB +from swarms.playground.memory.chromadb_example import ChromaDB import logging import os import uuid diff --git a/playground/__init__.py b/playground/__init__.py new file mode 100644 index 00000000..c1359528 --- /dev/null +++ b/playground/__init__.py @@ -0,0 +1,3 @@ + +from playground.memory import * # noqa: E402, F403 + diff --git a/pyproject.toml b/pyproject.toml index 9d4ca4b0..a5be57f1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -53,7 +53,8 @@ docstring_parser = "0.16" fastapi = "*" openai = ">=1.30.1,<2.0" termcolor = "*" - +chromadb = "*" +anthropic = "*" [tool.poetry.group.lint.dependencies] black = ">=23.1,<25.0" diff --git a/requirements.txt b/requirements.txt index 09a4fc1d..df29590a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,4 +27,6 @@ mypy-protobuf>=3.0.0 pytest>=8.1.1 termcolor>=2.4.0 pandas>=2.2.2 -fastapi>=0.110.1 \ No newline at end of file +fastapi>=0.110.1 +chromadb +anthropic \ No newline at end of file