appease test gods

run-llama · Jan 17, 2025 · 13d9455 · 13d9455
1 parent 25121f4
commit 13d9455
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/llama-index-core/llama_index/core/agent/workflow/base_agent.py b/llama-index-core/llama_index/core/agent/workflow/base_agent.py
@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Callable, List, Sequence, Optional
+from typing import Callable, List, Sequence, Optional, Union
 
 from llama_index.core.agent.workflow.workflow_events import (
     AgentOutput,
@@ -52,7 +52,7 @@ class BaseWorkflowAgent(BaseModel, PromptMixin, ABC):
 
     @field_validator("tools", mode="before")
     def validate_tools(
-        cls, v: Optional[Sequence[BaseTool | Callable]]
+        cls, v: Optional[Sequence[Union[BaseTool, Callable]]]
     ) -> Optional[Sequence[BaseTool]]:
         """Validate tools.
 

diff --git a/llama-index-core/llama_index/core/agent/workflow/multi_agent_workflow.py b/llama-index-core/llama_index/core/agent/workflow/multi_agent_workflow.py
@@ -451,7 +451,7 @@ def run(
     @classmethod
     def from_tools_or_functions(
         cls,
-        tools_or_functions: list[BaseTool | Callable],
+        tools_or_functions: List[Union[BaseTool, Callable]],
         llm: Optional[LLM] = None,
         system_prompt: Optional[str] = None,
         state_prompt: Optional[Union[str, BasePromptTemplate]] = None,