Fix 3.9 typing

Author: jlowin

src/controlflow/agents/agent.py

@@ -83,7 +83,7 @@ class Agent(ControlFlowModel, abc.ABC):
         default=False,
         description="If True, the agent is given tools for interacting with a human user.",
     )
-    memories: list[Memory] | list[AsyncMemory] = Field(
+    memories: list[Union[Memory, AsyncMemory]] = Field(
         default=[],
         description="A list of memory modules for the agent to use.",
     )
@@ -345,7 +345,7 @@ def _run_model(
 
         create_markdown_artifact(
             markdown=f"""
-{response.content or '(No content)'}
+{response.content or "(No content)"}
 
 #### Payload
 ```json
@@ -409,7 +409,7 @@ async def _run_model_async(
 
         create_markdown_artifact(
             markdown=f"""
-{response.content or '(No content)'}
+{response.content or "(No content)"}
 
 #### Payload
 ```json

src/controlflow/defaults.py

@@ -40,9 +40,7 @@ class Defaults(ControlFlowModel):
     model: Optional[Any]
     history: History
     agent: Agent
-    memory_provider: (
-        Optional[Union[MemoryProvider, str]] | Optional[Union[AsyncMemoryProvider, str]]
-    )
+    memory_provider: Optional[Union[MemoryProvider, AsyncMemoryProvider, str]]
 
     # add more defaults here
     def __repr__(self) -> str:

src/controlflow/orchestration/orchestrator.py

@@ -188,7 +188,7 @@ def get_tools(self) -> list[Tool]:
         tools = as_tools(tools)
         return tools
 
-    def get_memories(self) -> list[Memory] | list[AsyncMemory]:
+    def get_memories(self) -> list[Union[Memory, AsyncMemory]]:
         memories = set()
 
         memories.update(self.agent.memories)
@@ -525,7 +525,7 @@ def compile_prompt(self) -> str:
         ]
 
         prompt = "\n\n".join([p for p in prompts if p])
-        logger.debug(f"{'='*10}\nCompiled prompt: {prompt}\n{'='*10}")
+        logger.debug(f"{'=' * 10}\nCompiled prompt: {prompt}\n{'=' * 10}")
         return prompt
 
     def compile_messages(self) -> list[BaseMessage]:

src/controlflow/orchestration/prompt_templates.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 from pydantic import model_validator
 
@@ -98,7 +98,7 @@ def should_render(self) -> bool:
 
 class MemoryTemplate(Template):
     template_path: str = "memories.jinja"
-    memories: list[Memory] | list[AsyncMemory]
+    memories: list[Union[Memory, AsyncMemory]]
 
     def should_render(self) -> bool:
         return bool(self.memories)