Merge pull request #255 from PrefectHQ/ten

v0.9 foundation

.github/workflows/run-tests.yml

@@ -44,6 +44,9 @@ jobs:
       - name: Install ControlFlow
         run: uv pip install --system ".[tests]"
 
+      - name: Install Prefect @ b6d19c1 (remove after release)
+        run: uv pip install --system prefect@git+https://github.com/prefecthq/prefect@b6d19c1
+
       - name: Run tests
         run: pytest -vv
         if: ${{ !(github.event.pull_request.head.repo.fork) }}

.pre-commit-config.yaml

@@ -1,7 +1,7 @@
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
-    rev: v0.4.9
+    rev: v0.6.2
     hooks:
       # Run the linter.
       - id: ruff
@@ -10,7 +10,7 @@ repos:
       - id: ruff-format
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.10.0
+    rev: v1.11.2
     hooks:
       - id: mypy
         additional_dependencies:

examples/business_headline_sentiment.py

@@ -1,8 +1,9 @@
 # uv pip install langchain-community, duckduckgo-search
 
-import controlflow as cf
 from langchain_community.tools import DuckDuckGoSearchRun
 
+import controlflow as cf
+
 summarizer = cf.Agent(
     name="Headline Summarizer",
     description="An AI agent that fetches and summarizes current events",

examples/controlflow_docs.py

@@ -1,8 +1,9 @@
 from pathlib import Path
 
+from langchain_openai import OpenAIEmbeddings
+
 import controlflow as cf
 from controlflow.tools import tool
-from langchain_openai import OpenAIEmbeddings
 
 try:
     from langchain_community.document_loaders import DirectoryLoader

examples/engineer/engineer.py

@@ -1,9 +1,10 @@
 from pathlib import Path
 
+from pydantic import BaseModel
+
 import controlflow as cf
 import controlflow.tools.code
 import controlflow.tools.filesystem
-from pydantic import BaseModel
 
 # load the instructions
 instructions = open(Path(__file__).parent / "instructions.md").read()

examples/multi_agent_conversation.py

@@ -70,7 +70,7 @@ def demo(topic: str):
         context=dict(topic=topic),
         instructions="every agent should speak at least once. only one agent per turn. Keep responses 1-2 paragraphs max.",
     )
-    return task
+    task.run()
 
 
 if __name__ == "__main__":

examples/poem.py

@@ -1,6 +1,7 @@
-from controlflow import Agent, Task, flow, instructions, task
 from pydantic import BaseModel
 
+from controlflow import Agent, Task, flow, instructions, task
+
 
 class Name(BaseModel):
     first_name: str

examples/restaurant_recs.py

@@ -1,6 +1,7 @@
-from controlflow import Task, flow
 from pydantic import BaseModel
 
+from controlflow import Task, flow
+
 
 class Restaurant(BaseModel):
     name: str

src/controlflow/__init__.py

@@ -7,8 +7,9 @@
 from controlflow.defaults import defaults
 
 from .agents import Agent
-from .tasks import Task
+from .tasks import Task, run, run_async
 from .flows import Flow
+from .orchestration import turn_strategies
 
 
 from .instructions import instructions

src/controlflow/agents/__init__.py

@@ -1,3 +1,2 @@
 from . import memory
-from .agent import Agent, BaseAgent
-from .teams import RoundRobinTeam, Team
+from .agent import Agent