eaglelandsonce commited on
Commit
dfb41c0
1 Parent(s): a3dac62

Upload 2 files

Browse files
Files changed (1) hide show
  1. crewai/agents/cache/cache_hit.py +7 -3
crewai/agents/cache/cache_hit.py CHANGED
@@ -1,5 +1,6 @@
1
- from langchain_core.agents import AgentAction
2
- from pydantic.v1 import BaseModel, Field
 
3
 
4
  from .cache_handler import CacheHandler
5
 
@@ -10,5 +11,8 @@ class CacheHit(BaseModel):
10
  class Config:
11
  arbitrary_types_allowed = True
12
 
13
- action: AgentAction = Field(description="Action taken")
 
 
 
14
  cache: CacheHandler = Field(description="Cache Handler for the tool")
 
1
+ from typing import Any
2
+
3
+ from pydantic import BaseModel, Field
4
 
5
  from .cache_handler import CacheHandler
6
 
 
11
  class Config:
12
  arbitrary_types_allowed = True
13
 
14
+ # Making it Any instead of AgentAction to avoind
15
+ # pydantic v1 vs v2 incompatibility, langchain should
16
+ # soon be updated to pydantic v2
17
+ action: Any = Field(description="Action taken")
18
  cache: CacheHandler = Field(description="Cache Handler for the tool")